Refresh status of attached cluster

packages/databricks-vscode/src/cluster/ClusterListDataProvider.ts

@@ -148,10 +148,7 @@ export class ClusterListDataProvider
             }
         );
 
-        if (
-            element.stateMessage &&
-            !["RUNNING", "TERMINATED"].includes(element.state)
-        ) {
+        if (element.stateMessage && element.state !== "RUNNING") {
             children.push({
                 label: "State message:",
                 description: element.stateMessage,

packages/databricks-vscode/src/cluster/ClusterManager.ts

@@ -1,19 +1,42 @@
-import {cluster, Cluster} from "@databricks/databricks-sdk";
+import {cluster, Cluster, Time, TimeUnits} from "@databricks/databricks-sdk";
 import {CancellationToken, CancellationTokenSource, Disposable} from "vscode";
 
 export class ClusterManager implements Disposable {
     private cancellationTokenSource?: CancellationTokenSource;
+    private refreshTimer?: NodeJS.Timer;
 
-    constructor(readonly cluster: Cluster) {}
+    constructor(
+        readonly cluster: Cluster,
+        readonly onChange: (state: cluster.ClusterInfoState) => void = (
+            state
+        ) => {},
+        readonly refreshTimeout: Time = new Time(10, TimeUnits.seconds)
+    ) {
+        this.setInterval();
+    }
+
+    private setInterval() {
+        this.refreshTimer = setInterval(async () => {
+            await this.cluster.refresh();
+            this.onChange(this.cluster.state);
+        }, this.refreshTimeout.toMillSeconds().value);
+    }
+
+    private clearInterval() {
+        clearInterval(this.refreshTimer);
+        this.refreshTimer = undefined;
+    }
 
     dispose() {
         this.cancellationTokenSource?.cancel();
         this.cancellationTokenSource?.dispose();
+        this.clearInterval();
     }
 
     async start(
         onProgress: (state: cluster.ClusterInfoState) => void = (state) => {}
     ) {
+        this.clearInterval();
         this.cancellationTokenSource?.cancel();
         this.cancellationTokenSource = new CancellationTokenSource();
 
@@ -23,11 +46,13 @@ export class ClusterManager implements Disposable {
         );
 
         onProgress(this.cluster.state);
+        this.setInterval();
     }
 
     async stop(
         onProgress: (state?: cluster.ClusterInfoState) => void = (state) => {}
     ) {
+        this.clearInterval();
         this.cancellationTokenSource?.cancel();
         this.cancellationTokenSource = new CancellationTokenSource();
 
@@ -38,5 +63,6 @@ export class ClusterManager implements Disposable {
                 onProgress(clusterInfo.state)
         );
         onProgress(this.cluster.state);
+        this.setInterval();
     }
 }

packages/databricks-vscode/src/configuration/ConnectionManager.ts

@@ -363,7 +363,9 @@ export class ConnectionManager {
         if (this.cluster !== newCluster) {
             this._clusterManager?.dispose();
             this._clusterManager = newCluster
-                ? new ClusterManager(newCluster)
+                ? new ClusterManager(newCluster, (state) => {
+                      this.onDidChangeClusterEmitter.fire(this.cluster);
+                  })
                 : undefined;
             this.onDidChangeClusterEmitter.fire(this.cluster);
         }