feat: Remove executor metrics on stop

near · Feb 13, 2024 · bd8aeeb · bd8aeeb
1 parent 439d760
commit bd8aeeb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/runner/src/metrics.ts b/runner/src/metrics.ts
@@ -73,6 +73,10 @@ export const registerWorkerMetrics = (workerId: number, metrics: string): void =
   workerMetrics[workerId] = metrics;
 };
 
+export const deregisterWorkerMetrics = (workerId: number): void => {
+  workerMetrics[workerId] = undefined;
+};
+
 export const startServer = async (): Promise<void> => {
   const app = express();
 

diff --git a/runner/src/stream-handler/stream-handler.ts b/runner/src/stream-handler/stream-handler.ts
@@ -60,6 +60,8 @@ export default class StreamHandler {
   }
 
   async stop (): Promise<void> {
+    deregisterWorkerMetrics(this.worker.threadId);
+
     await this.worker.terminate();
   }