JS `Node.js Worker Threads`：CPU 密集型任务的多线程解决方案

各位观众老爷，大家好！今天咱们来聊聊 Node.js 里一个相当给力的东西：Worker Threads，也就是工作线程。这玩意儿能让你原本单线程的 Node.js 应用，摇身一变，玩起多线程，专门用来对付那些 CPU 密集型的任务。准备好了吗？咱们这就开讲！

一、Node.js 的“辛酸史”：单线程的爱与痛

Node.js，这名字听起来就一股青春活力，对吧？它最大的特点就是单线程，而且是基于事件循环（Event Loop）的。这么设计的好处有很多，比如：

轻量级： 单线程嘛，内存占用少，启动速度快。
非阻塞 I/O： 遇到 I/O 操作，不会傻等，而是先去处理其他事情，等 I/O 完成了再回来处理结果。这让 Node.js 在处理高并发 I/O 密集型任务时，效率杠杠的。

但是！人生不如意事十之八九，单线程也有它的阿喀琉斯之踵。那就是：

CPU 密集型任务的噩梦： 啥叫 CPU 密集型任务？就是那些需要大量计算的任务，比如图片处理、音视频编码、复杂的数学运算等等。这些任务会长时间占用 CPU，导致事件循环被阻塞，整个 Node.js 应用卡住，就像便秘一样难受。

举个例子，假设你要计算圆周率 π 的小数点后 1 亿位。这绝对是个 CPU 密集型任务。如果你直接在 Node.js 的主线程里算，你的服务器可能就直接失去响应，用户只能对着菊花转圈圈。

// 阻塞主线程的计算圆周率的代码 (千万别在生产环境里这么干！)
function calculatePi(digits) {
  // 省略复杂的计算过程... (这里只是示意)
  let pi = 3.141592653589793; // 假装算了好久
  return pi;
}

const startTime = Date.now();
const pi = calculatePi(100000000); // 计算 1 亿位圆周率
const endTime = Date.now();

console.log(`π ≈ ${pi}`);
console.log(`计算耗时：${endTime - startTime} ms`);

// 在计算过程中，主线程会被阻塞，无法处理其他请求

二、Worker Threads：救星驾到！

为了解决 CPU 密集型任务带来的问题，Node.js 引入了 Worker Threads。简单来说，就是允许你创建多个线程，让它们并行执行任务，而不会阻塞主线程。

你可以把 Worker Threads 想象成你的分身，每个分身都可以帮你处理不同的任务，这样你就不会被 CPU 密集型任务给困住了。

三、Worker Threads 的基本用法

要使用 Worker Threads，你需要用到 worker_threads 模块。这个模块提供了创建和管理线程的 API。

1. 创建 Worker 线程

首先，你要创建一个 Worker 线程。这可以通过 new Worker(filename) 来实现，其中 filename 是一个 JavaScript 文件的路径，这个文件将在新的线程中执行。

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

const worker = new Worker('./worker.js'); // 创建一个 Worker 线程，执行 worker.js 文件

2. 主线程和 Worker 线程之间的通信

主线程和 Worker 线程之间需要进行通信，才能传递数据和控制线程的执行。worker_threads 模块提供了 postMessage 和 on('message') 方法来实现通信。

主线程向 Worker 线程发送消息： 使用 worker.postMessage(data)。
Worker 线程接收主线程的消息： 使用 parentPort.on('message', (data) => { ... })。其中 parentPort 是 worker_threads 模块提供的一个全局对象，代表 Worker 线程的父线程（也就是主线程）。
Worker 线程向主线程发送消息： 使用 parentPort.postMessage(data)。
主线程接收 Worker 线程的消息： 使用 worker.on('message', (data) => { ... })。

3. 一个简单的例子：计算斐波那契数列

咱们来看一个简单的例子，用 Worker Threads 来计算斐波那契数列。斐波那契数列的计算是一个典型的 CPU 密集型任务。

主线程 (main.js):

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

function runService(workerData) {
  return new Promise((resolve, reject) => {
    const worker = new Worker('./worker.js', { workerData });
    worker.on('message', resolve);
    worker.on('error', reject);
    worker.on('exit', (code) => {
      if (code !== 0)
        reject(new Error(`Worker stopped with exit code ${code}`));
    })
  })
}

async function run() {
  const result = await runService({ value: 40 });
  console.log(result);
}

run().catch(err => console.error(err));

Worker 线程 (worker.js):

const { parentPort, workerData } = require('worker_threads');

function fibonacci(n) {
  return n < 2 ? n : fibonacci(n - 1) + fibonacci(n - 2);
}

const data = workerData.value;
parentPort.postMessage(fibonacci(data));

在这个例子中，主线程创建了一个 Worker 线程，并将要计算的斐波那契数列的项数（40）通过 workerData 传递给 Worker 线程。Worker 线程接收到数据后，计算斐波那契数列，并将结果通过 parentPort.postMessage 发送回主线程。主线程接收到结果后，打印到控制台。

四、Worker Threads 的高级用法

除了基本的创建和通信，Worker Threads 还有一些高级用法，可以让你更灵活地控制线程的执行。

1. 共享内存 (SharedArrayBuffer)

Worker Threads 之间可以共享内存，这样可以避免大量的数据拷贝，提高性能。要使用共享内存，你需要用到 SharedArrayBuffer 对象。

SharedArrayBuffer 对象代表一段共享的内存区域，可以在多个线程之间访问。你可以使用 Int32Array、Float64Array 等类型化数组来操作共享内存中的数据。

2. Atomics 操作

由于多个线程可以同时访问共享内存，因此需要使用原子操作来保证数据的完整性。Atomics 对象提供了一组原子操作，可以用来读取、写入和修改共享内存中的数据，而不会发生数据竞争。

3. 使用 MessageChannel 进行更复杂的通信

MessageChannel 提供了一种创建双向通信通道的方式，可以在主线程和 Worker 线程之间，或者在多个 Worker 线程之间进行更复杂的通信。

五、Worker Threads 的注意事项

在使用 Worker Threads 时，需要注意以下几点：

线程安全： Worker Threads 之间共享数据时，要特别注意线程安全问题，避免数据竞争。可以使用锁、原子操作等机制来保证数据的完整性。
内存占用： 每个 Worker Thread 都会占用一定的内存，因此要合理控制线程的数量，避免内存溢出。
调试： 多线程程序的调试比较复杂，可以使用 Node.js 的调试工具或者专门的线程调试工具来辅助调试。
序列化： 通过 postMessage 传递的数据需要进行序列化和反序列化，这会带来一定的性能开销。尽量避免传递大量的数据，或者使用共享内存来减少数据拷贝。
并非银弹： Worker Threads 并不是解决所有 CPU 密集型任务的银弹。对于一些简单的任务，使用 Worker Threads 可能会带来额外的开销，反而降低性能。需要根据实际情况进行评估和选择。

六、Worker Threads 的应用场景

Worker Threads 可以应用于各种需要处理 CPU 密集型任务的场景，例如：

图片处理： 对图片进行缩放、裁剪、滤镜等操作。
音视频编码： 对音视频文件进行编码、解码、转码等操作。
科学计算： 进行复杂的数学运算、统计分析等操作。
密码学： 进行加密、解密、哈希等操作。
游戏开发： 进行游戏逻辑的计算、物理引擎的模拟等操作。

七、Worker Threads 和 Cluster 的区别

Node.js 还有一个 Cluster 模块，也可以用来实现多进程并发。那么 Worker Threads 和 Cluster 有什么区别呢？

特性	Worker Threads	Cluster
进程/线程	线程 (共享内存空间)	进程 (独立的内存空间)
通信方式	`postMessage`, `SharedArrayBuffer`	IPC (进程间通信)
资源消耗	相对较低	相对较高
适用场景	CPU 密集型任务，需要共享数据	I/O 密集型任务，需要提高应用的可用性和容错性
复杂性	相对较高 (需要处理线程安全问题)	相对较低

简单来说，Worker Threads 更适合于处理 CPU 密集型任务，并且需要共享数据的场景。而 Cluster 更适合于处理 I/O 密集型任务，并且需要提高应用的可用性和容错性的场景。

八、总结

Worker Threads 是 Node.js 中一个非常强大的工具，可以让你充分利用多核 CPU 的性能，解决 CPU 密集型任务带来的问题。但是，使用 Worker Threads 也需要注意一些细节，例如线程安全、内存占用、调试等等。希望今天的讲座能帮助大家更好地理解和使用 Worker Threads。

好了，今天的讲座就到这里。感谢大家的观看！下次有机会再跟大家分享其他的技术知识。拜拜！

发表回复 取消回复

发表回复取消回复