Add data fetcher references in schema provider for @DgsEntityFetcher …

…methods
Netflix · Jan 24, 2025 · 0465e5e · 0465e5e
1 parent d3ef3a1
commit 0465e5e
Showing 1 changed file with 7 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,17 @@ 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
                         }