Update usage of func

gitpod-io · Nov 8, 2023 · 75b1576 · 75b1576
1 parent 10f8b46
commit 75b1576
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 14 deletions.
diff --git a/components/dashboard/src/service/json-rpc-workspace-client.ts b/components/dashboard/src/service/json-rpc-workspace-client.ts
@@ -38,13 +38,19 @@ export class JsonRpcWorkspaceClient implements PromiseClient<typeof WorkspaceSer
             throw new ConnectError("signal is required", Code.InvalidArgument);
         }
         const it = generateAsyncGenerator<WorkspaceInstance>(
-            (sink) => {
-                const dispose = getGitpodService().registerClient({
-                    onInstanceUpdate: (instance) => {
-                        sink.next(instance);
-                    },
-                });
-                return dispose.dispose;
+            (queue) => {
+                try {
+                    const dispose = getGitpodService().registerClient({
+                        onInstanceUpdate: (instance) => {
+                            queue.push(instance);
+                        },
+                    });
+                    return () => {
+                        dispose.dispose();
+                    };
+                } catch (e) {
+                    queue.fail(e);
+                }
             },
             { signal: options.signal },
         );

diff --git a/components/server/src/workspace/workspace-service.ts b/components/server/src/workspace/workspace-service.ts
@@ -736,13 +736,17 @@ export class WorkspaceService {
 
     public watchWorkspaceStatus(userId: string, opts: { signal: AbortSignal }) {
         return generateAsyncGenerator<WorkspaceInstance>((sink) => {
-            const dispose = this.subscriber.listenForWorkspaceInstanceUpdates(userId, (_ctx, instance) => {
-                sink.next(instance);
-            });
-            return () => {
-                console.log("=============dispose");
-                dispose.dispose();
-            };
+            try {
+                const dispose = this.subscriber.listenForWorkspaceInstanceUpdates(userId, (_ctx, instance) => {
+                    sink.push(instance);
+                });
+                return () => {
+                    console.log("=============dispose");
+                    dispose.dispose();
+                };
+            } catch (e) {
+                sink.fail(e);
+            }
         }, opts);
     }