如何使用Node.js worker threads加快代码执行，解决Electron UI阻塞问题？

问题根因：所有耗时操作都运行在Node.js主线程，事件循环被阻塞后，Electron UI的渲染和交互响应都会被挂起。

Worker Threads 可以将CPU密集型任务放到独立线程执行，不会阻塞主线程事件循环，你可以按照以下步骤改造现有代码：

1. 新建独立的worker线程文件，比如task.worker.js，把耗时函数迁移到该文件中：

const { parentPort } = require('worker_threads');
const someModules = require("some-nodejs-modules");

async function first_CpuIntensive_Func() {
  // 原有CPU密集逻辑
  // 原有正则处理逻辑
  return someData;
}
async function second_CpuIntensive_Func() {
  // 原有CPU密集逻辑
  // 原有正则处理逻辑
  return someData;
}

// 监听主线程的调用指令
parentPort.on('message', async (taskType) => {
  let result;
  if (taskType === 'first') {
    result = await first_CpuIntensive_Func();
  } else if (taskType === 'second') {
    result = await second_CpuIntensive_Func();
  }
  parentPort.postMessage(result);
});

2. 修改原有对外导出的工具文件，封装worker调用逻辑，对外暴露的接口保持不变，原有调用代码无需修改：

const { Worker } = require('worker_threads');
const path = require('path');

function runWorkerTask(taskType) {
  return new Promise((resolve, reject) => {
    // 加载worker文件
    const worker = new Worker(path.join(__dirname, './task.worker.js'));
    // 发送任务类型到worker
    worker.postMessage(taskType);
    // 监听worker返回的结果
    worker.on('message', (result) => {
      worker.terminate();
      resolve(result);
    });
    // 监听worker报错
    worker.on('error', (err) => {
      worker.terminate();
      reject(err);
    });
    // 监听worker退出
    worker.on('exit', (code) => {
      if (code !== 0) reject(new Error(`Worker stopped with exit code ${code}`));
    });
  });
}

// 对外接口和原有保持完全一致
module.exports = {
  first_CpuIntensive_Func: () => runWorkerTask('first'),
  second_CpuIntensive_Func: () => runWorkerTask('second')
};

使用注意事项：

• 不要在worker和主线程之间传递体积过大的对象，序列化/反序列化过程会产生额外性能开销
• 如果需要高频调用这两个函数，可以复用worker实例，不需要每次调用都新建和销毁worker，减少额外开销
• Electron环境默认支持Worker Threads，主进程和开启了nodeIntegration的渲染进程都可以直接使用

• 分批分片处理任务：如果是重命名文件这类IO密集型操作，可以把数千个任务拆分为小批次处理，每批执行完成后用setImmediate让出事件循环，给UI渲染预留执行时间，示例如下：

async function batchRenameTask(files) {
  const batchSize = 15; // 可根据实际性能调整批次大小
  for (let i = 0; i < files.length; i += batchSize) {
    const currentBatch = files.slice(i, i + batchSize);
    await Promise.all(currentBatch.map(file => singleRenameLogic(file)));
    // 让出事件循环，避免阻塞UI
    await new Promise(resolve => setImmediate(resolve));
  }
}

• 使用Electron Utility Process：Electron 14及以上版本提供的专用API，用于在独立进程中执行耗时任务，相比Worker Threads支持更完整的Node.js API，不会和主进程共享事件循环，更适合执行较重的后台任务
• 优化现有耗时逻辑：优先排查现有逻辑的性能瓶颈，比如优化低效正则、减少不必要的循环遍历、替换为性能更高的第三方依赖，很多时候优化完逻辑后耗时会大幅下降，不需要额外引入多进程/多线程方案

内容的提问来源于stack exchange，提问作者Muhammad Taif Khan