send telemetry when scaling

Signed-off-by: Andre Dietisheim <[email protected]>
redhat-developer · Aug 25, 2023 · df31833 · df31833
1 parent 7919bf1
commit df31833
Showing 1 changed file with 44 additions and 12 deletions.
diff --git a/src/main/kotlin/com/redhat/devtools/intellij/kubernetes/actions/ScaleReplicaAction.kt b/src/main/kotlin/com/redhat/devtools/intellij/kubernetes/actions/ScaleReplicaAction.kt
@@ -13,8 +13,12 @@ package com.redhat.devtools.intellij.kubernetes.actions
 import com.intellij.openapi.actionSystem.AnActionEvent
 import com.redhat.devtools.intellij.common.actions.StructureTreeAction
 import com.redhat.devtools.intellij.kubernetes.dialogs.ScaleReplicaDialog
+import com.redhat.devtools.intellij.kubernetes.model.IResourceModel
 import com.redhat.devtools.intellij.kubernetes.model.Notification
+import com.redhat.devtools.intellij.kubernetes.model.resource.kubernetes.KubernetesReplicas
 import com.redhat.devtools.intellij.kubernetes.model.util.toMessage
+import com.redhat.devtools.intellij.kubernetes.telemetry.TelemetryService
+import com.redhat.devtools.intellij.telemetry.core.service.TelemetryMessageBuilder
 import io.fabric8.kubernetes.api.model.HasMetadata
 import io.fabric8.kubernetes.api.model.Pod
 import io.fabric8.kubernetes.api.model.ReplicationController
@@ -32,33 +36,46 @@ class ScaleReplicaAction: StructureTreeAction() {
     }
 
     override fun actionPerformed(event: AnActionEvent?, path: Array<out TreePath>?, selected: Array<out Any>?) {
-        val project = getEventProject(event) ?: return
-        val toScale = selected?.firstOrNull()?.getElement<HasMetadata>() ?: return
-        val model = getResourceModel() ?: return
+        val telemetry = TelemetryService.instance.action("scale")
+        val project = getEventProject(event) ?: return aborting("no project", telemetry)
+        val toScale = selected?.firstOrNull()?.getElement<HasMetadata>() ?: return aborting("no resource selected", telemetry)
+        val model = getResourceModel() ?: return aborting("resource model not found", telemetry)
         try {
             val replicator = model.getReplicas(toScale)
             val replicas = replicator?.replicas
             if (replicator == null
                 || replicas == null) {
-                Notification().error(
-                    "Error Scaling",
-                    "Could not scale ${toScale.kind} '${toScale.metadata.name}: unsupported resource"
-                )
+                error("replicator not found/unsupported kind", toScale, telemetry)
                 return
             }
             val resourceLabel = "${replicator.resource.kind} ${replicator.resource.metadata.name}"
             ScaleReplicaDialog(
                 project,
                 resourceLabel,
                 replicas,
-                { replicas: Int -> model.setReplicas(replicas, replicator)},
+                setReplicas(replicator, model, telemetry),
                 (event?.inputEvent as? MouseEvent)?.locationOnScreen
             ).show()
         } catch (e: RuntimeException) {
-            Notification().error(
-                "Error Scaling",
-                "Could not scale ${toScale.kind} '${toScale.metadata.name}': ${toMessage(e)}"
-            )
+            error(toMessage(e), toScale, telemetry)
+        }
+    }
+
+    private fun setReplicas(
+        replicator: KubernetesReplicas.Replicator,
+        model: IResourceModel,
+        telemetry: TelemetryMessageBuilder.ActionMessage
+    ): (Int) -> Unit {
+        return { replicas: Int ->
+            try {
+                model.setReplicas(replicas, replicator)
+                telemetry
+                    .property(TelemetryService.PROP_RESOURCE_KIND, replicator.resource.kind)
+                    .success()
+                    .send()
+            } catch (e: RuntimeException) {
+                error(toMessage(e), replicator.resource, telemetry)
+            }
         }
     }
 
@@ -76,4 +93,19 @@ class ScaleReplicaAction: StructureTreeAction() {
                 || element is StatefulSet
                 || element is Pod
     }
+
+    private fun error(cause: String, resource: HasMetadata, telemetry: TelemetryMessageBuilder.ActionMessage) {
+        Notification().error(
+            "Error Scaling",
+            "Could not scale ${resource.kind} '${resource.metadata.name}: $cause"
+        )
+        telemetry
+            .property(TelemetryService.PROP_RESOURCE_KIND, resource.kind)
+            .error("Could not scale ${resource.kind}: $cause")
+            .send()
+    }
+
+    private fun aborting(cause: String, telemetry: TelemetryMessageBuilder.ActionMessage) {
+        telemetry.error("aborting: $cause").send()
+    }
 }