feat(jetbrains): show workspace resources

Author: Andrea Falzetti

components/ide/jetbrains/backend-plugin/src/main/kotlin/io/gitpod/jetbrains/remote/GitpodManager.kt

@@ -365,4 +365,31 @@ class GitpodManager : Disposable {
             }
         }
     }
+
+    var resourceStatus: Status.ResourcesStatusResponse? = null
+
+    private val metricsJob = GlobalScope.launch {
+        if (application.isHeadlessEnvironment) {
+            return@launch
+        }
+        val status = StatusServiceGrpc.newFutureStub(supervisorChannel)
+        while (isActive) {
+            try {
+                val f = status.resourcesStatus(Status.ResourcesStatuRequest.getDefaultInstance())
+                resourceStatus = f.asDeferred().await()
+            } catch (t: Throwable) {
+                if (t is CancellationException) {
+                    throw t
+                }
+                thisLogger().error("gitpod: failed to retrieve resource status: ", t)
+            }
+            delay(1000L)
+        }
+    }
+    init {
+        lifetime.onTerminationOrNow {
+            metricsJob.cancel()
+        }
+    }
+
 }

components/ide/jetbrains/backend-plugin/src/main/kotlin/io/gitpod/jetbrains/remote/latest/GitpodMetricControlProvider.kt

@@ -0,0 +1,49 @@
+// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License-AGPL.txt in the project root for license information.
+
+package io.gitpod.jetbrains.remote.latest
+
+import com.jetbrains.ide.model.uiautomation.BeControl
+import com.jetbrains.ide.model.uiautomation.DefiniteProgress
+import com.jetbrains.rd.platform.codeWithMe.unattendedHost.metrics.providers.IdeMetricsProvider
+import com.jetbrains.rd.ui.bedsl.dsl.VerticalGridBuilder
+import com.jetbrains.rd.ui.bedsl.dsl.verticalGrid
+import com.jetbrains.rd.util.lifetime.Lifetime
+import com.jetbrains.rd.util.reactive.Property
+import com.jetbrains.rdserver.diagnostics.BackendDiagnosticsService
+import com.jetbrains.rdserver.unattendedHost.customization.controlCenter.performance.MetricControlProvider
+import com.jetbrains.rdserver.unattendedHost.customization.controlCenter.performance.createMetricProgressWithLabels
+import com.jetbrains.rdserver.unattendedHost.customization.controlCenter.performance.createProgressBar
+import com.jetbrains.rdserver.unattendedHost.customization.controlCenter.performance.createProgressRow
+
+class GitpodMetricControlProvider : MetricControlProvider {
+    override val id: String = "gitpodControl"
+    override fun getControl(lifetime: Lifetime): BeControl {
+        return verticalGrid {
+            val backendDiagnosticsService = BackendDiagnosticsService.Companion.getInstance()
+           createCpuControl(this, backendDiagnosticsService, lifetime)
+            createMemoryControl(this, backendDiagnosticsService, lifetime)
+        }
+    }
+
+    private fun createCpuControl(ctx: VerticalGridBuilder, backendDiagnosticsService: BackendDiagnosticsService, lifetime: Lifetime) {
+        val data = createMetricProgressWithLabels(backendDiagnosticsService, lifetime, "gitpod_workspace_cpu_total", "gitpod_workspace_cpu_used", "gitpod_workspace_cpu_percentage")
+
+        val label = "Workspace CPU"
+        // TODO: make status dynamic based on a threshold
+        val status = Property(com.jetbrains.rd.platform.codeWithMe.unattendedHost.metrics.MetricsStatus.DANGER)
+
+        createProgressRow(ctx, lifetime, label, status, data.first, data.second, data.third)
+    }
+
+    private fun createMemoryControl(ctx: VerticalGridBuilder, backendDiagnosticsService: BackendDiagnosticsService, lifetime: Lifetime) {
+        val data = createMetricProgressWithLabels(backendDiagnosticsService, lifetime, "gitpod_workspace_memory_total", "gitpod_workspace_memory_used", "gitpod_workspace_memory_percentage")
+
+        val label = "Workspace Memory"
+        // TODO: make status dynamic based on a threshold
+        val status = Property(com.jetbrains.rd.platform.codeWithMe.unattendedHost.metrics.MetricsStatus.DANGER)
+
+        createProgressRow(ctx, lifetime, label, status, data.first, data.second, data.third)
+    }
+}

components/ide/jetbrains/backend-plugin/src/main/kotlin/io/gitpod/jetbrains/remote/latest/GitpodMetricProvider.kt

@@ -0,0 +1,40 @@
+// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License-AGPL.txt in the project root for license information.
+
+package io.gitpod.jetbrains.remote.latest
+
+import com.intellij.openapi.components.service
+import com.jetbrains.rd.platform.codeWithMe.unattendedHost.metrics.Metric
+import com.jetbrains.rd.platform.codeWithMe.unattendedHost.metrics.MetricType
+import com.jetbrains.rd.platform.codeWithMe.unattendedHost.metrics.MetricsStatus
+import com.jetbrains.rd.platform.codeWithMe.unattendedHost.metrics.providers.MetricProvider
+import io.gitpod.jetbrains.remote.GitpodManager
+
+
+class GitpodMetricProvider: MetricProvider {
+    private val manager = service<GitpodManager>()
+
+    override val id: String = "gitpodMetrics"
+    override fun getMetrics(): Map<String, Metric> {
+        // TODO: make values dynamic listening from supervisor
+        val resourceStatus = manager.resourceStatus
+
+        val cpuUsed = resourceStatus?.cpu?.used?.toFloat() ?: 0f
+        val cpuTotal = resourceStatus?.cpu?.limit?.toFloat() ?: 0f
+        val cpuPercentage = (cpuUsed / cpuTotal) * 100
+
+        val memoryUsed = resourceStatus?.memory?.used?.toFloat() ?: 0f
+        val memoryTotal = resourceStatus?.memory?.limit?.toFloat() ?: 0f
+        val memoryPercentage = (memoryUsed / memoryTotal) * 100
+
+        return mapOf(
+                "gitpod_workspace_cpu_used" to Metric(MetricType.OTHER, MetricsStatus.DANGER, cpuUsed),
+                "gitpod_workspace_cpu_total" to Metric(MetricType.OTHER, MetricsStatus.DANGER, cpuTotal),
+                "gitpod_workspace_cpu_percentage" to Metric(MetricType.OTHER, MetricsStatus.DANGER, cpuPercentage.toDouble()),
+                "gitpod_workspace_memory_used" to Metric(MetricType.PERFORMANCE, MetricsStatus.NORMAL, memoryUsed),
+                "gitpod_workspace_memory_total" to Metric(MetricType.PERFORMANCE, MetricsStatus.NORMAL, memoryTotal),
+                "gitpod_workspace_memory_percentage" to Metric(MetricType.PERFORMANCE, MetricsStatus.NORMAL, memoryPercentage.toDouble())
+        )
+    }
+}

components/ide/jetbrains/backend-plugin/src/main/resources-latest/META-INF/extensions.xml

@@ -7,5 +7,7 @@
     <extensions defaultExtensionNs="com.intellij">
         <projectService serviceImplementation="io.gitpod.jetbrains.remote.latest.GitpodTerminalService" client="guest" preload="true"/>
         <projectService serviceImplementation="io.gitpod.jetbrains.remote.latest.GitpodPortForwardingService" preload="true"/>
+        <gateway.customization.performance id="gitpodControl" order="before cpuControl" implementation="io.gitpod.jetbrains.remote.latest.GitpodMetricControlProvider"/>
+        <gateway.customization.metrics id="gitpodMetrics" implementation="io.gitpod.jetbrains.remote.latest.GitpodMetricProvider" />
     </extensions>
 </idea-plugin>