Skip to content

Commit 3fb5220

Browse files
committed
use grackle's isOneOf def
1 parent c244ab5 commit 3fb5220

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

gen/rules/src/main/scala/clue/gen/SchemaGen.scala

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,9 +104,11 @@ trait SchemaGen extends Generator {
104104
) ++ parentBody
105105
) ++
106106
schema.types
107-
.collect[Class] { case InputObjectType(name, _, fields, directives) =>
108-
if (directives.exists(_.name == "oneOf")) buildInputSumType(name, fields)
109-
else buildInputCaseClass(name, fields)
107+
.collect {
108+
case iot @ InputObjectType(name, _, fields, _) if iot.isOneOf =>
109+
buildInputSumType(name, fields)
110+
case InputObjectType(name, _, fields, _) =>
111+
buildInputCaseClass(name, fields)
110112
}
111113
.map(
112114
_.addToParentBody(

0 commit comments

Comments
 (0)