node.js之---子线程（child_process）模块

为什么需要子线程（child_process）模块

Worker Threads 的基本概念

如何使用 Worker Threads

Worker Threads 的性能

Worker 线程的优势和限制

进阶用法：共享内存

为什么需要子线程（child_process）模块

在 Node.js 中，Worker Threads 模块 （worker_threads）提供了一种在 Node.js 单线程中使用多线程的方式，从而能够更高效地处理计算密集型任务，避免阻塞主线程（事件循环）。这是 Node.js 中引入的一种并发处理机制，旨在提高性能，尤其是在需要大量计算的情况下。

Node.js 默认是单线程的，它通过事件循环来处理异步操作。虽然 Node.js 可以在后台异步执行 I/O 操作（如文件读取、数据库查询等），但它的事件循环在处理计算密集型任务时会被阻塞。这意味着长时间运行的 CPU 密集型任务（如大型数据处理或算法计算）可能会阻塞事件循环，从而影响整个应用的响应能力。

为了克服这个问题，Node.js 引入了 Worker Threads 模块，它允许你在单个进程中创建多个线程，每个线程都拥有自己的执行上下文，并可以并行地处理任务。

Worker Threads 的基本概念

主线程（Main thread）：主线程是 Node.js 应用的默认执行环境。所有的 I/O 操作和事件循环都在主线程中进行。
Worker 线程（Worker threads）：每个 Worker 线程拥有自己的事件循环和内存空间。它们与主线程并行运行，能够处理独立的任务。

如何使用 Worker Threads

要使用 Worker Threads，首先需要引入 worker_threads 模块。每个 Worker 线程都可以通过 Worker 类来创建，主线程和 Worker 线程之间的通信是通过 消息传递 实现的。主线程可以向 Worker 线程发送消息，Worker 线程也可以向主线程发送结果。

Worker Threads 的核心 API

worker_threads.Worker: 用于创建一个新的 Worker 线程。
worker_threads.isMainThread: 一个布尔值，用来判断当前代码是否在主线程中执行。
worker_threads.parentPort: 主线程和 Worker 线程之间的通信通道。
worker_threads.workerData: 允许向 Worker 线程传递数据。

javascript 复制代码

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

if (isMainThread) {
  // 主线程代码
  console.log('主线程正在运行');

  // 创建 Worker 线程
  const worker = new Worker(__filename, {
    workerData: { start: 1, end: 5 }
  });

  // 监听 Worker 线程返回的消息
  worker.on('message', (result) => {
    console.log(`主线程收到结果：${result}`);
  });

  worker.on('error', (err) => {
    console.error('Worker 线程发生错误:', err);
  });

  worker.on('exit', (code) => {
    if (code !== 0) {
      console.error(`Worker 线程退出时的错误代码: ${code}`);
    }
  });
} else {
  // Worker 线程代码
  console.log('Worker 线程正在运行');

  const { start, end } = workerData;

  // 执行任务并将结果返回给主线程
  let result = 0;
  for (let i = start; i <= end; i++) {
    result += i;
  }

  parentPort.postMessage(result);  // 向主线程发送结果
}

Worker Threads 的性能

线程池大小 ：默认情况下，Worker Threads 模块使用系统的线程池。每个 Worker 线程在独立的 CPU 核心上运行，理论上可以并行执行多个计算任务。
内存隔离：每个 Worker 线程拥有独立的内存空间和执行上下文，不会与其他线程共享数据，因此可以避免传统多线程编程中的竞态条件问题。
开销：每个 Worker 线程都需要一定的资源开销（内存、启动时间等），所以要谨慎创建过多的 Worker 线程。

Worker 线程的优势和限制

优点：

避免阻塞：Worker 线程可以并行处理计算密集型任务，不会阻塞主线程的事件循环。
内存隔离：每个 Worker 线程有独立的内存空间，避免了共享内存带来的问题。
更好的多核利用：可以利用多核 CPU 来并行处理任务，充分发挥硬件性能。

限制：

消息传递开销：线程间的通信是基于消息传递的，这可能会引入一定的延迟，尤其是在需要频繁交互的场景下。
内存限制：每个 Worker 线程都需要占用一定的内存和资源，创建大量的 Worker 线程可能导致内存消耗过高。
无法共享内存 ：Worker 线程之间没有共享内存空间，虽然可以通过 SharedArrayBuffer 实现共享内存，但这需要额外的管理和同步机制。

进阶用法：共享内存

虽然 Worker 线程之间没有直接的内存共享，但可以通过 SharedArrayBuffer 实现内存共享。SharedArrayBuffer 是一种允许在多个线程之间共享内存的结构，适用于需要高效交换大量数据的场景。

javascript 复制代码

const { Worker, isMainThread, parentPort, workerData } = require('worker_threads');
const { SharedArrayBuffer, Int32Array } = require('buffer');

if (isMainThread) {
  const sharedBuffer = new SharedArrayBuffer(4 * Int32Array.BYTES_PER_ELEMENT);
  const sharedArray = new Int32Array(sharedBuffer);

  // 将共享内存传递给 Worker 线程
  const worker = new Worker(__filename, { workerData: sharedBuffer });

  worker.on('message', () => {
    console.log(`Main Thread: Shared memory content: ${sharedArray[0]}`);
  });
} else {
  const sharedArray = new Int32Array(workerData);

  // 修改共享内存
  sharedArray[0] = 42;

  parentPort.postMessage('done');
}

在这个例子中，SharedArrayBuffer 允许主线程和 Worker 线程之间共享内存。通过 Int32Array 视图访问和修改这块内存。