fix: ensure proper cleanup of message listeners on stop

purocean · Jan 22, 2025 · 554a569 · 554a569
1 parent 8185864
commit 554a569
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/server/watch-worker.ts b/src/main/server/watch-worker.ts
@@ -98,16 +98,18 @@ function init (id: number, filePath: string | string[], options: WatchOpts) {
     console.log(`watch process ${id} >`, filePath, 'stop')
     promiseQueue.length = 0
     watcher.close()
+    process.off('message', onMessage)
+    process.off('exit', stop)
   }
 
   function onMessage (message: Message) {
     if (message.id === id && message.type === 'stop') {
       stop()
-      process.off('message', onMessage)
     }
   }
 
   process.on('message', onMessage)
+  process.on('exit', stop)
 }
 
 process.on('message', (message: Message) => {