Merge pull request #2157 from digma-ai/feature/span-info

Added span info endpoint

Author: opoliarush

analytics-provider/src/main/java/org/digma/intellij/plugin/analytics/AnalyticsProvider.java

@@ -133,5 +133,7 @@ public interface AnalyticsProvider extends Closeable {
 
     String getHighlightsScaling(HighlightsRequest request);
 
+    String getSpanInfo(String spanCodeObjectId);
+
     String getHighlightsImpact(HighlightsRequest request);
 }

analytics-provider/src/main/java/org/digma/intellij/plugin/analytics/RestAnalyticsProvider.java

@@ -340,6 +340,11 @@ public String getHighlightsScaling(HighlightsRequest request) {
         return execute(() -> client.analyticsProvider.getHighlightsScaling(request));
     }
 
+    @Override
+    public String getSpanInfo(String spanCodeObjectId) {
+        return execute(() -> client.analyticsProvider.getSpanInfo(spanCodeObjectId));
+    }
+
     @Override
     public HttpResponse lowLevelCall(HttpRequest request) {
 
@@ -965,5 +970,12 @@ Call<ResponseBody> setInsightCustomStartTime(
         })
         @POST("/highlights/scaling")
         Call<String> getHighlightsScaling(@Body HighlightsRequest request);
+
+        @Headers({
+                "Accept: application/+json",
+                "Content-Type:application/json"
+        })
+        @GET("/spans/info")
+        Call<String> getSpanInfo(@Query("SpanCodeObjectId") String spanCodeObjectId);
     }
 }

ide-common/src/main/java/org/digma/intellij/plugin/analytics/AnalyticsService.java

@@ -405,6 +405,9 @@ public String getHighlightsScaling(HighlightsRequest request) throws AnalyticsSe
         return executeCatching(() -> analyticsProviderProxy.getHighlightsScaling(request));
     }
 
+    public String getSpanInfo(String spanCodeObjectId) throws AnalyticsServiceException {
+        return executeCatching(() -> analyticsProviderProxy.getSpanInfo(spanCodeObjectId));
+    }
 
     public String getHighlightsImpact(HighlightsRequest request) throws AnalyticsServiceException {
         return executeCatching(() -> analyticsProviderProxy.getHighlightsImpact(request));

src/main/kotlin/org/digma/intellij/plugin/ui/highlights/HighlightsMessageRouterHandler.kt

@@ -25,6 +25,7 @@ class HighlightsMessageRouterHandler(project: Project) : BaseCommonMessageRouter
                 "MAIN/GET_HIGHLIGHTS_TOP_ISSUES_DATA" -> getHighlightsTopInsightsV2(browser, requestJsonNode)
                 "MAIN/GET_HIGHLIGHTS_IMPACT_DATA" -> getHighlightsImpact(browser, requestJsonNode)
                 "MAIN/GET_HIGHLIGHTS_SCALING_DATA" -> getHighlightsScaling(browser, requestJsonNode)
+                "MAIN/GET_SPAN_INFO_DATA" -> getSpanInfo(browser, requestJsonNode)
 
                 else -> {
                     return false
@@ -138,6 +139,24 @@ class HighlightsMessageRouterHandler(project: Project) : BaseCommonMessageRouter
         }
     }
 
+    @Synchronized
+    @Throws(JsonProcessingException::class)
+    private fun getSpanInfo(browser: CefBrowser, requestJsonNode: JsonNode) {
+
+        Log.log(logger::trace, project, "getSpanInfo called")
+
+        val payloadQuery = getPayloadQuery(requestJsonNode)
+        if (payloadQuery is ObjectNode) {
+
+            val request = payloadQuery.get("spanCodeObjectId").asText()
+            val payload = HighlightsService.getInstance(project).getSpanInfo(request)
+
+            val message = SetSpanInfoMessage(payload)
+            Log.log(logger::trace, project, "sending MAIN/SET_SPAN_INFO_DATA message")
+            serializeAndExecuteWindowPostMessageJavaScript(browser, message)
+        }
+    }
+
 
     private fun getHighlightsTopInsightsV2(browser: CefBrowser, requestJsonNode: JsonNode) {
         Log.log(logger::trace, project, "getHighlightsTopInsights called")

src/main/kotlin/org/digma/intellij/plugin/ui/highlights/HighlightsService.kt

@@ -95,4 +95,16 @@ class HighlightsService(val project: Project) : InsightsServiceImpl(project) {
             null
         }
     }
+
+    fun getSpanInfo(spanCodeObjectId: String): String? {
+        EDT.assertNonDispatchThread()
+
+        return try {
+            val spanInfo = AnalyticsService.getInstance(project).getSpanInfo(spanCodeObjectId)
+            spanInfo
+        } catch (e: AnalyticsServiceException) {
+            Log.warnWithException(logger, project, e, "Error loading highlights top insights {}", e.message)
+            null
+        }
+    }
 }

src/main/kotlin/org/digma/intellij/plugin/ui/highlights/model/SetSpanInfoMessage.kt

@@ -0,0 +1,9 @@
+package org.digma.intellij.plugin.ui.highlights.model
+
+import com.fasterxml.jackson.annotation.JsonRawValue
+import org.digma.intellij.plugin.ui.jcef.JCEFGlobalConstants
+
+data class SetSpanInfoMessage(@JsonRawValue val payload: String?) {
+    val type = JCEFGlobalConstants.REQUEST_MESSAGE_TYPE
+    val action = "MAIN/SET_SPAN_INFO_DATA"
+}