Merge pull request #301 from zanonena/feature/interface-companion

Generate companion objects for union interfaces in Kotlin
Netflix · Dec 3, 2021 · c651a77 · c651a77
2 parents ece4e0d + 6d7dadb
commit c651a77
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/...main/kotlin/com/netflix/graphql/dgs/codegen/generators/kotlin/KotlinUnionTypeGenerator.kt b/...main/kotlin/com/netflix/graphql/dgs/codegen/generators/kotlin/KotlinUnionTypeGenerator.kt
@@ -48,6 +48,9 @@ class KotlinUnionTypeGenerator(private val config: CodeGenConfig) {
             interfaceBuilder.addAnnotation(jsonTypeInfoAnnotation())
             interfaceBuilder.addAnnotation(jsonSubTypesAnnotation(memberTypes))
         }
+
+        interfaceBuilder.addType(TypeSpec.companionObjectBuilder().build())
+
         val fileSpec = FileSpec.get(packageName, interfaceBuilder.build())
         return CodeGenResult(kotlinInterfaces = listOf(fileSpec))
     }

diff --git a/...hql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/KotlinCodeGenTest.kt b/...hql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/KotlinCodeGenTest.kt
@@ -1581,7 +1581,9 @@ class KotlinCodeGenTest {
                 |  JsonSubTypes.Type(value = Actor::class, name = "Actor"),
                 |  JsonSubTypes.Type(value = Rating::class, name = "Rating")
                 |])
-                |public interface SearchResult
+                |public interface SearchResult {
+                |  public companion object
+                |}
 
         """.trimMargin()
         )