Merge pull request #2452 from digma-ai/fix/recheck

Added recheck response

Author: opoliarush

src/main/java/org/digma/intellij/plugin/insights/AbstractInsightsMessageRouterHandler.java

@@ -46,7 +46,7 @@ public boolean doOnQuery(@NotNull Project project, @NotNull CefBrowser browser,
 
             case "INSIGHTS/OPEN_HISTOGRAM" -> openHistogram(requestJsonNode);
 
-            case "INSIGHTS/RECALCULATE" -> recalculate(requestJsonNode);
+            case "INSIGHTS/RECALCULATE" -> recalculate(browser, requestJsonNode);
 
             case "INSIGHTS/GO_TO_TRACE" -> goToTrace(requestJsonNode);
 
@@ -281,10 +281,12 @@ private void openLiveView(JsonNode jsonNode) throws JsonProcessingException {
         InsightsService.getInstance(getProject()).openLiveView(codeObjectId);
     }
 
-
-    private void recalculate(JsonNode jsonNode) throws JsonProcessingException {
+    private void recalculate(@NotNull CefBrowser browser, JsonNode jsonNode) throws JsonProcessingException {
         var insightId = getObjectMapper().readTree(jsonNode.get("payload").toString()).get("id").asText();
         InsightsService.getInstance(getProject()).recalculate(insightId);
+
+        var message = new SetInsightRecalculatedMessage(new SetInsightRecalculated(insightId));
+        serializeAndExecuteWindowPostMessageJavaScript(browser, message);
     }
 
     private void goToTrace(JsonNode jsonNode) {
@@ -310,7 +312,5 @@ private void goToTraceComparison(JsonNode jsonNode) {
             InsightsService.getInstance(getProject()).goToTraceComparison(traceId1, traceName1, traceId2, traceName2, insightType);
         }
     }
-
-
 }
 

src/main/java/org/digma/intellij/plugin/insights/InsightsServiceImpl.java

@@ -92,14 +92,11 @@ public void recalculate(@NotNull String insightId) {
         }
     }
 
-
     public void goToTrace(@NotNull String traceId, @NotNull String traceName, @NotNull String insightType, @Nullable String spanCodeObjectId) {
         JaegerUtilKt.openJaegerFromInsight(project, traceId, traceName, insightType, spanCodeObjectId);
     }
 
     public void goToTraceComparison(@NotNull String traceId1, @NotNull String traceName1, @NotNull String traceId2, @NotNull String traceName2, @NotNull String insightType) {
         JaegerUtilKt.openJaegerComparisonFromInsight(project, traceId1, traceName1, traceId2, traceName2, insightType);
     }
-
-
 }

src/main/kotlin/org/digma/intellij/plugin/ui/insights/model/SetInsightRecalculated.kt

@@ -0,0 +1,20 @@
+package org.digma.intellij.plugin.ui.insights.model
+
+import com.fasterxml.jackson.annotation.JsonCreator
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties
+import org.digma.intellij.plugin.ui.jcef.JCEFGlobalConstants
+import java.beans.ConstructorProperties
+
+
+@JsonIgnoreProperties(ignoreUnknown = true)
+data class SetInsightRecalculated @JsonCreator(mode = JsonCreator.Mode.PROPERTIES)
+@ConstructorProperties("insightId")
+constructor(
+    val insightId: String,
+)
+
+data class SetInsightRecalculatedMessage(val payload: SetInsightRecalculated) {
+    val type = JCEFGlobalConstants.REQUEST_MESSAGE_TYPE
+    val action = "INSIGHTS/SET_RECALCULATED"
+}
+