Added scaling highlights

Author: opoliarush

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

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

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

@@ -335,6 +335,11 @@ public String getHighlightsTopInsightsV2(HighlightsRequest request) {
         return execute(() -> client.analyticsProvider.getHighlightsTopInsightsV2(request));
     }
 
+    @Override
+    public String getHighlightsScaling(HighlightsRequest request) {
+        return execute(() -> client.analyticsProvider.getHighlightsScaling(request));
+    }
+
     @Override
     public HttpResponse lowLevelCall(HttpRequest request) {
 
@@ -954,5 +959,11 @@ Call<ResponseBody> setInsightCustomStartTime(
         @POST("/authentication/register")
         Call<String> register(@Body Map<String, Object> request);
 
+        @Headers({
+                "Accept: application/+json",
+                "Content-Type:application/json"
+        })
+        @POST("/highlights/scaling")
+        Call<String> getHighlightsScaling(@Body HighlightsRequest request);
     }
 }

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

@@ -403,6 +403,11 @@ public String getHighlightsTopInsightsV2(HighlightsRequest request) throws Analy
         return executeCatching(() -> analyticsProviderProxy.getHighlightsTopInsightsV2(request));
     }
 
+    public String getHighlightsScaling(HighlightsRequest request) throws AnalyticsServiceException {
+        return executeCatching(() -> analyticsProviderProxy.getHighlightsScaling(request));
+    }
+
+
     public String getHighlightsImpact(HighlightsRequest request) throws AnalyticsServiceException {
         return executeCatching(() -> analyticsProviderProxy.getHighlightsImpact(request));
     }

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

@@ -7,13 +7,9 @@ import com.fasterxml.jackson.databind.node.ObjectNode
 import com.intellij.openapi.project.Project
 import org.cef.browser.CefBrowser
 import org.digma.intellij.plugin.analytics.getVersion
-import org.digma.intellij.plugin.analytics.isCentralized
 import org.digma.intellij.plugin.log.Log
 import org.digma.intellij.plugin.model.rest.highlights.HighlightsRequest
-import org.digma.intellij.plugin.ui.common.updateObservabilityValue
-import org.digma.intellij.plugin.ui.highlights.model.SetHighlightsImpactMessage
-import org.digma.intellij.plugin.ui.highlights.model.SetHighlightsPerformanceMessage
-import org.digma.intellij.plugin.ui.highlights.model.SetHighlightsTopInsightsMessage
+import org.digma.intellij.plugin.ui.highlights.model.*
 import org.digma.intellij.plugin.ui.jcef.BaseCommonMessageRouterHandler
 import org.digma.intellij.plugin.ui.jcef.getQueryMapFromPayload
 import org.digma.intellij.plugin.ui.jcef.serializeAndExecuteWindowPostMessageJavaScript
@@ -23,12 +19,12 @@ class HighlightsMessageRouterHandler(project: Project) : BaseCommonMessageRouter
     override fun doOnQuery(project: Project, browser: CefBrowser, requestJsonNode: JsonNode, rawRequest: String, action: String): Boolean {
 
         var version = getVersion();
-        if (version == "unknown" || version >= "0.3.7")
-        {
+        if (version == "unknown" || compareVersions("0.3.7", version) <= 0) {
             when (action) {
                 "MAIN/GET_HIGHLIGHTS_PERFORMANCE_DATA" -> getHighlightsPerformanceV2(browser, requestJsonNode)
                 "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)
 
                 else -> {
                     return false
@@ -62,6 +58,21 @@ class HighlightsMessageRouterHandler(project: Project) : BaseCommonMessageRouter
         serializeAndExecuteWindowPostMessageJavaScript(browser, message)
     }
 
+    private fun compareVersions(left: String, right: String): Int {
+        val leftParts = left.split(".").map { it.toInt() }
+        val rightParts = right.split(".").map { it.toInt() }
+
+        for (i in 0 until minOf(leftParts.size, rightParts.size)) {
+            if (leftParts[i] < rightParts[i]) {
+                return -1
+            } else if (leftParts[i] > rightParts[i]) {
+                return 1
+            }
+        }
+
+        return leftParts.size.compareTo(rightParts.size)
+    }
+
     private fun getHighlightsTopInsights(browser: CefBrowser, requestJsonNode: JsonNode) {
         Log.log(logger::trace, project, "getHighlightsTopInsights called")
 
@@ -108,6 +119,26 @@ class HighlightsMessageRouterHandler(project: Project) : BaseCommonMessageRouter
         }
     }
 
+
+    @Synchronized
+    @Throws(JsonProcessingException::class)
+    private fun getHighlightsScaling(browser: CefBrowser, requestJsonNode: JsonNode) {
+
+        Log.log(logger::trace, project, "getHighlightsScaling called")
+
+        val payloadQuery = getPayloadQuery(requestJsonNode)
+        if (payloadQuery is ObjectNode) {
+
+            var request = createHighlightsRequest(payloadQuery, "scopedSpanCodeObjectId");
+            val payload = HighlightsService.getInstance(project).getHighlightsScaling(request)
+
+            val message = SetHighlightsScalingMessage(payload)
+            Log.log(logger::trace, project, "sending MAIN/SET_HIGHLIGHTS_PERFORMANCE_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

@@ -83,4 +83,16 @@ class HighlightsService(val project: Project) : InsightsServiceImpl(project) {
             null
         }
     }
+
+    fun getHighlightsScaling(request: HighlightsRequest): String? {
+        EDT.assertNonDispatchThread()
+
+        return try {
+            val highlightsPerformance = AnalyticsService.getInstance(project).getHighlightsScaling(request)
+            highlightsPerformance
+        } 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/SetHighlightsScalingMessage.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 SetHighlightsScalingMessage(@JsonRawValue val payload: String?) {
+    val type = JCEFGlobalConstants.REQUEST_MESSAGE_TYPE
+    val action = "MAIN/SET_HIGHLIGHTS_SCALING_DATA"
+}