[server] notify only browser clients on instance/prebuild updates

Author: AlexTugarev

components/server/src/workspace/gitpod-server-impl.ts

@@ -285,6 +285,8 @@ export class GitpodServerImpl implements GitpodServerWithTracing, Disposable {
 
     private readonly disposables = new DisposableCollection();
 
+    private shouldNotifyClients = false;
+
     dispose(): void {
         this.disposables.dispose();
     }
@@ -309,12 +311,20 @@ export class GitpodServerImpl implements GitpodServerWithTracing, Disposable {
         log.debug({ userId: this.userID }, `clientRegion: ${clientHeaderFields.clientRegion}`);
         log.debug({ userId: this.userID }, "initializeClient");
 
-        this.listenForWorkspaceInstanceUpdates();
+        if (this.client && clientMetadata.type === "browser") {
+            this.shouldNotifyClients = true;
+        }
 
-        this.listenForPrebuildUpdates().catch((err) => log.error("error registering for prebuild updates", err));
+        if (this.shouldNotifyClients) {
+            this.listenForWorkspaceInstanceUpdates();
+            this.listenForPrebuildUpdates().catch((err) => log.error("error registering for prebuild updates", err));
+        }
     }
 
     private async listenForPrebuildUpdates() {
+        if (!this.shouldNotifyClients) {
+            return;
+        }
         // 'registering for prebuild updates for all projects this user has access to
         const projects = await this.getAccessibleProjects();
 
@@ -331,7 +341,7 @@ export class GitpodServerImpl implements GitpodServerWithTracing, Disposable {
             );
 
         for (const projectId of projects) {
-            this.disposables.pushAll([this.subscriber.listenForPrebuildUpdates(projectId, handler)]);
+            this.disposables.push(this.subscriber.listenForPrebuildUpdates(projectId, handler));
         }
 
         // TODO(at) we need to keep the list of accessible project up to date
@@ -453,7 +463,8 @@ export class GitpodServerImpl implements GitpodServerWithTracing, Disposable {
                 // Once we have those, we should remove this.
                 //
                 const ws = await this.workspaceDb.trace(ctx).findById(workspaceID);
-                if (!!ws && !!wsi && ws.ownerId !== this.userID) {
+                const relatedPrebuildFound = !!ws && !!wsi && ws.ownerId !== this.userID;
+                if (this.client && relatedPrebuildFound) {
                     const resetListenerFromRedis = this.subscriber.listenForWorkspaceInstanceUpdates(
                         ws.ownerId,
                         (ctx, instance) => {
@@ -529,7 +540,7 @@ export class GitpodServerImpl implements GitpodServerWithTracing, Disposable {
     }
 
     private listenForWorkspaceInstanceUpdates(): void {
-        if (!this.userID || !this.client) {
+        if (!this.shouldNotifyClients || !this.userID) {
             return;
         }
 
@@ -3046,19 +3057,23 @@ export class GitpodServerImpl implements GitpodServerWithTracing, Disposable {
         }
 
         const project = await this.projectsService.createProject(params, user);
+
         // update client registration for the logged in user
-        const prebuildUpdateHandler = (ctx: TraceContext, update: PrebuildWithStatus) =>
-            TraceContext.withSpan(
-                "forwardPrebuildUpdateToClient",
-                (ctx) => {
-                    traceClientMetadata(ctx, this.clientMetadata);
-                    TraceContext.setJsonRPCMetadata(ctx, "onPrebuildUpdate");
+        if (this.shouldNotifyClients) {
+            const prebuildUpdateHandler = (ctx: TraceContext, update: PrebuildWithStatus) =>
+                TraceContext.withSpan(
+                    "forwardPrebuildUpdateToClient",
+                    (ctx) => {
+                        traceClientMetadata(ctx, this.clientMetadata);
+                        TraceContext.setJsonRPCMetadata(ctx, "onPrebuildUpdate");
+
+                        this.client?.onPrebuildUpdate(update);
+                    },
+                    ctx,
+                );
 
-                    this.client?.onPrebuildUpdate(update);
-                },
-                ctx,
-            );
-        this.disposables.pushAll([this.subscriber.listenForPrebuildUpdates(project.id, prebuildUpdateHandler)]);
+            this.disposables.pushAll([this.subscriber.listenForPrebuildUpdates(project.id, prebuildUpdateHandler)]);
+        }
 
         return project;
     }