entgql: support SkipMutationCreateInput (and update) on entity (#598)

ent · Jul 24, 2024 · 7fc697f · 7fc697f
1 parent fcf98f0
commit 7fc697f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/entgql/schema.go b/entgql/schema.go
@@ -546,6 +546,13 @@ func (e *schemaGenerator) buildMutationInputs(t *gen.Type, ant *Annotation, gqlT
 	var defs []*ast.Definition
 
 	for _, i := range ant.MutationInputs {
+		if i.IsCreate && ant.Skip.Is(SkipMutationCreateInput) {
+			continue
+		}
+		if !i.IsCreate && ant.Skip.Is(SkipMutationUpdateInput) {
+			continue
+		}
+
 		desc := MutationDescriptor{Type: t, IsCreate: i.IsCreate}
 		name, err := desc.Input()
 		if err != nil {