Using FinalizationRegistry to always terminate workers

Author: constantinius

src/pool.js

@@ -75,6 +75,10 @@ class WorkerWrapper {
   }
 }
 
+const finalizationRegistry = new FinalizationRegistry((worker) => {
+  worker.terminate();
+});
+
 /**
  * Pool for workers to decode chunks of the images.
  */
@@ -114,7 +118,10 @@ class Pool {
       this.workerWrappers = (async () => {
         const workerWrappers = [];
         for (let i = 0; i < size; i++) {
-          workerWrappers.push(new WorkerWrapper(createWorker()));
+          const worker = createWorker();
+          const wrapper = new WorkerWrapper(worker);
+          workerWrappers.push(wrapper);
+          finalizationRegistry.register(wrapper, worker, wrapper);
         }
         return workerWrappers;
       })();

src/worker/create.js

@@ -0,0 +1,9 @@
+// The contents of this file may be overridded for some build processes
+
+import Worker from 'web-worker';
+
+export default function create() {
+  return new Worker(new URL('./decoder.js', import.meta.url), {
+    type: 'module',
+  });
+}