add parameters to enum constructor

Netflix · Mar 20, 2024 · 0ccf925 · 0ccf925
1 parent 3bae6d6
commit 0ccf925
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...core/src/main/kotlin/com/netflix/graphql/dgs/codegen/generators/java/DataTypeGenerator.kt b/...core/src/main/kotlin/com/netflix/graphql/dgs/codegen/generators/java/DataTypeGenerator.kt
@@ -382,7 +382,7 @@ abstract class BaseDataTypeGenerator(
         val enumBuilder = TypeSpec
             .enumBuilder("Field")
             .addModifiers(Modifier.PUBLIC)
-            .addField(FieldSpec.builder(com.squareup.javapoet.TypeName.INT, "ordinal").initializer("-1").build())
+            .addField(FieldSpec.builder(ClassName.INT, "ordinal").initializer("-1").build())
             .addMethod(MethodSpec
                 .methodBuilder("getOrdinal")
                 .addModifiers(Modifier.PUBLIC)
@@ -396,6 +396,7 @@ abstract class BaseDataTypeGenerator(
             )
             .addMethod(MethodSpec
                 .constructorBuilder()
+                .addParameter(ClassName.INT, "ordinal")
                 .addCode(
                     """
                         |this.ordinal = ordinal;