Node.js多线程编程：使用worker_threads处理CPU密集型任务

你知道吗？尽管 Node.js 以其单线程、事件驱动的非阻塞 I/O 模型而闻名，但它也提供了处理 CPU 密集型任务的方案，以避免阻塞主事件循环。这就要用到 worker_threads 模块。它允许我们创建真正的并行线程，从而有效提升计算密集型操作的性能。

简单来说，主线程可以创建一个或多个工作线程（Worker），将繁重的计算任务分配出去。两者之间通过消息传递进行通信，工作线程完成任务后将结果返回给主线程，这样主线程就能保持响应，轻松处理更多请求。

正确的实现方式：分离关注点

一个清晰、可维护的做法是将工作线程的逻辑放在独立的文件中。这样做不仅代码结构更清晰，也便于复用和测试。

下面是主线程（main.js）的代码示例：

// 主线程代码
const { Worker } = require('worker_threads');

// 创建一个新的工作线程，指定独立的工作文件
const worker = new Worker('./worker.js');

// 监听工作线程发送回来的消息
worker.on('message', (result) => {
  console.log(`Worker result: ${result}`);
});

// 向工作线程发送任务数据
worker.postMessage({ num1: 5, num2: 3 });

接着，工作线程的逻辑写在独立的 worker.js 文件中：

// 工作线程代码
const { parentPort } = require('worker_threads');

// 监听主线程发送过来的消息
parentPort.on('message', ({ num1, num2 }) => {
  const result = num1 + num2;
  parentPort.postMessage(result); // 将计算结果发送回主线程
});

在这个示例中，worker_threads 模块的核心用法清晰明了。主线程与工作线程完全解耦，通过消息机制协作。这种方式极大地提升了代码的可维护性和可复用性，是构建健壮Node.js应用的良好实践。

不推荐的实现方式：内联代码

相反，有一种不那么优雅的做法，是将工作线程的代码直接以内联字符串的形式写在主线程中。虽然功能上可以实现，但会带来维护上的噩梦。

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

// 不推荐：将工作线程逻辑以内联方式创建
const worker = new Worker(`
  const { parentPort } = require('worker_threads');
  parentPort.on('message', ({ num1, num2 }) => {
    const result = num1 + num2;
    parentPort.postMessage(result);
  });
`, { eval: true });

worker.on('message', (result) => {
  console.log(`Worker result: ${result}`);
});
worker.postMessage({ num1: 5, num2: 3 });

你能看出问题吗？这种写法把逻辑硬编码在字符串里，导致代码难以阅读、调试和复用。任何逻辑修改都需要改动这个字符串，并且失去了语法高亮和代码提示的支持，非常容易出错。

核心要点与适用场景

使用 worker_threads 模块的关键在于理解其适用场景。它并非用于处理高 I/O 并发（那本就是 Node.js 的强项），而是专为 CPU 密集型任务 设计的。

哪些算 CPU 密集型任务呢？比如：

• 复杂的数据处理与转换
• 图像或视频的压缩/解压
• 加密解密运算
• 复杂的数学计算或模拟

通过将这些耗时计算任务分流到工作线程，主线程的事件循环得以保持轻盈，能够快速响应新的网络请求或 I/O 事件，从而整体提升应用程序的吞吐量和响应速度。

掌握多线程编程，是迈向高性能 Node.js 后端开发的重要一步。如果你对这类提升应用性能的架构话题感兴趣，欢迎来云栈社区与其他开发者一起交流探讨。