Merge pull request #2109 from Netflix/feature/_entities-datafetchers

Add data fetcher references in schema provider for @DgsEntityFetcher methods
Netflix · Jan 24, 2025 · b0aab8a · b0aab8a
2 parents d3ef3a1 + c2e10ab
commit b0aab8a
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/graphql-dgs/src/main/kotlin/com/netflix/graphql/dgs/internal/DgsSchemaProvider.kt b/graphql-dgs/src/main/kotlin/com/netflix/graphql/dgs/internal/DgsSchemaProvider.kt
@@ -611,12 +611,19 @@ class DgsSchemaProvider
                     .forEach { method ->
                         val dgsEntityFetcherAnnotation = method.getAnnotation(DgsEntityFetcher::class.java)
 
+                        val entityFetcherTypeName = dgsEntityFetcherAnnotation.name
+                        val coordinateName = "_entities.$entityFetcherTypeName"
+
+                        val mergedAnnotations =
+                            MergedAnnotations
+                                .from(method, MergedAnnotations.SearchStrategy.TYPE_HIERARCHY)
+                        dataFetcherInfo.dataFetchers +=
+                            DataFetcherReference(dgsComponent.instance, method, mergedAnnotations, "Query", coordinateName)
+
                         val enableInstrumentation =
                             method.getAnnotation(DgsEnableDataFetcherInstrumentation::class.java)?.value
                                 ?: false
-                        val entityFetcherTypeName = dgsEntityFetcherAnnotation.name
                         if (enableInstrumentation) {
-                            val coordinateName = "__entities.$entityFetcherTypeName"
                             dataFetcherInfo.tracingEnabled += coordinateName
                             dataFetcherInfo.metricsEnabled += coordinateName
                         }