Switch to workerpool lib

Author: davidtaylorhq

packages/babel/package.json

@@ -67,7 +67,8 @@
   },
   "dependencies": {
     "@babel/helper-module-imports": "^7.18.6",
-    "@rollup/pluginutils": "^5.0.1"
+    "@rollup/pluginutils": "^5.0.1",
+    "workerpool": "^9.0.0"
   },
   "devDependencies": {
     "@babel/core": "^7.19.1",

packages/babel/src/index.js

@@ -3,11 +3,11 @@ import { fileURLToPath } from 'url';
 
 import * as babel from '@babel/core';
 import { createFilter } from '@rollup/pluginutils';
+import workerpool from 'workerpool';
 
 import { BUNDLED, HELPERS } from './constants.js';
 import transformCode from './transformCode.js';
 import { escapeRegExpCharacters, warnOnce } from './utils.js';
-import WorkerPool from './workerPool.js';
 
 const unpackOptions = ({
   extensions = babel.DEFAULT_EXTENSIONS,
@@ -144,7 +144,10 @@ function createParallelWorkerPool(parallel, overrides) {
 
   // Default limits to 4 workers. Benefits diminish after this point, because of the setup cost.
   const workerCount = typeof parallel === 'number' ? parallel : Math.min(cpus().length, 4);
-  return new WorkerPool(WORKER_PATH, workerCount);
+  return workerpool.pool(WORKER_PATH, {
+    maxWorkers: workerCount,
+    workerType: 'thread'
+  });
 }
 
 function executeWithWorkerPool(workerPool, context, taskOpts, babelOptions) {
@@ -157,7 +160,7 @@ function executeWithWorkerPool(workerPool, context, taskOpts, babelOptions) {
     );
   }
 
-  return workerPool.runTask(taskOpts).catch((err) => context.error(err.message));
+  return workerPool.exec('transform', [taskOpts]).catch((err) => context.error(err.message));
 }
 
 function createBabelInputPluginFactory(customCallback = returnObject) {

packages/babel/src/worker.js

@@ -1,25 +1,16 @@
-import { parentPort } from 'worker_threads';
+import workerpool from 'workerpool';
 
 import transformCode from './transformCode.js';
 
-parentPort.on('message', async (opts) => {
-  try {
-    const result = await transformCode({
-      ...opts,
-      error: (msg) => {
-        throw new Error(msg);
-      }
-    });
-    parentPort.postMessage({
-      result
-    });
-  } catch (error) {
-    parentPort.postMessage({
-      error: {
-        message: error.message,
-        stack: error.stack,
-        name: error.name
-      }
-    });
-  }
+async function transform(opts) {
+  return transformCode({
+    ...opts,
+    error: (msg) => {
+      throw new Error(msg);
+    }
+  });
+}
+
+workerpool.worker({
+  transform
 });