Try upstream services when getting latencies

apollographql · Sep 10, 2024 · b8cf0d3 · b8cf0d3
1 parent 4f0be26
commit b8cf0d3
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/intellij-plugin/src/main/kotlin/com/apollographql/ijplugin/gradle/ApolloKotlinService.kt b/intellij-plugin/src/main/kotlin/com/apollographql/ijplugin/gradle/ApolloKotlinService.kt
@@ -43,6 +43,9 @@ data class ApolloKotlinService(
 
     @XCollection
     val endpointHeaders: Map<String, String>? = null,
+
+    @XCollection
+    val upstreamServiceIds: List<Id> = emptyList(),
 ) {
   data class Id(
       @Attribute
@@ -67,5 +70,4 @@ data class ApolloKotlinService(
   val id: Id
     @Transient
     get() = Id(gradleProjectPath, serviceName)
-
 }
diff --git a/...lij-plugin/src/main/kotlin/com/apollographql/ijplugin/gradle/GradleToolingModelService.kt b/...lij-plugin/src/main/kotlin/com/apollographql/ijplugin/gradle/GradleToolingModelService.kt
@@ -257,6 +257,7 @@ class GradleToolingModelService(
                 .distinct(),
             endpointUrl = serviceInfo.endpointUrlCompat(toolingModel),
             endpointHeaders = serviceInfo.endpointHeadersCompat(toolingModel),
+            upstreamServiceIds = upstreamApolloKotlinServices.map { it.id }
         )
       }
     }

diff --git a/...n/src/main/kotlin/com/apollographql/ijplugin/studio/fieldinsights/FieldInsightsService.kt b/...n/src/main/kotlin/com/apollographql/ijplugin/studio/fieldinsights/FieldInsightsService.kt
@@ -126,8 +126,20 @@ class FieldInsightsServiceImpl(private val project: Project) : FieldInsightsServ
     return fieldLatenciesByService.isNotEmpty()
   }
 
+  private fun getFieldLatenciesForService(serviceId: ApolloKotlinService.Id): FieldInsights.FieldLatencies? {
+    if (fieldLatenciesByService.containsKey(serviceId)) {
+      return fieldLatenciesByService[serviceId]
+    }
+    // Try upstream services
+    val apolloKotlinService = GradleToolingModelService.getApolloKotlinServices(project).firstOrNull { it.id == serviceId } ?: return null
+    for (upstreamServiceId in apolloKotlinService.upstreamServiceIds) {
+      return getFieldLatenciesForService(upstreamServiceId) ?: continue
+    }
+    return null
+  }
+
   override fun getLatency(serviceId: ApolloKotlinService.Id, typeName: String, fieldName: String): Double? {
-    return fieldLatenciesByService[serviceId]?.getLatency(parentType = typeName, fieldName = fieldName)
+    return getFieldLatenciesForService(serviceId)?.getLatency(parentType = typeName, fieldName = fieldName)
   }
 
   private fun refreshInspections() {