nodejs 获取线程id

在 Node.js 中，获取线程 ID 的解决方案主要依赖于 worker_threads 模块。通过该模块，我们可以在主线程和工作线程中分别获取对应的线程 ID。如何使用 worker_threads 模块来实现这一功能，并提供多种思路供读者参考。

解决方案

Node.js 提供了 worker_threads 模块，用于支持多线程编程。每个线程都有一个的 ID，可以通过 workerData 或 threadId 属性获取。在主线程中，可以创建多个工作线程，并通过 worker.threadId 来标识每个线程的性。而在工作线程中，则可以通过 workerThreads 模块中的 ThreadWorker 实例直接访问当前线程的 ID。

接下来，我们将通过代码示例详细说明如何获取线程 ID。

方法一：在工作线程中获取线程 ID

以下是一个简单的示例，展示如何在工作线程中获取当前线程的 ID。

代码示例

javascript
// worker.js 文件
const { workerData, threadId } = require('worker_threads');</p>

<p>console.log(<code>当前线程 ID: ${threadId});
console.log(传递的数据: ${workerData});

主线程代码

javascript
// main.js 文件
const { Worker } = require('worker_threads');</p>

<p>const worker = new Worker('./worker.js', {
  workerData: 'Hello from main thread!'
});</p>

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

<p>worker.on('exit', (code) => {
  if (code !== 0)
    console.error(<code>Worker 线程退出，退出码: ${code});
});

运行结果

运行 main.js 后，输出如下：

当前线程 ID: 1 传递的数据: Hello from main thread!

方法二：在主线程中获取工作线程的 ID

除了在工作线程中获取线程 ID，我们还可以在主线程中通过 worker.threadId 属性来获取工作线程的 ID。

代码示例

javascript
// main.js 文件
const { Worker } = require('worker_threads');</p>

<p>const worker = new Worker('./worker.js', {
  workerData: 'Hello from main thread!'
});</p>

<p>console.log(<code>工作线程 ID: ${worker.threadId});

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

worker.on('exit', (code) => {
  if (code !== 0)
    console.error(Worker 线程退出，退出码: ${code});
});

运行结果

运行 main.js 后，输出如下：

工作线程 ID: 1 当前线程 ID: 1 传递的数据: Hello from main thread!

方法三：使用 Promise 包装工作线程

为了更方便地管理异步任务，我们可以将工作线程封装为一个返回 Promise 的函数。这样可以更优雅地处理线程的启动和结束。

代码示例

javascript
// worker.js 文件
const { parentPort, threadId } = require('worker_threads');</p>

<p>parentPort.postMessage({ threadId, message: '线程已启动' });

主线程代码

javascript
// main.js 文件
const { Worker } = require('worker_threads');</p>

<p>function runWorker() {
  return new Promise((resolve, reject) => {
    const worker = new Worker('./worker.js');</p>

<pre><code>worker.on('message', (msg) => {
  resolve(msg);
});

worker.on('error', reject);

worker.on('exit', (code) => {
  if (code !== 0) {
    reject(new Error(`Worker 停止，退出码: ${code}`));
  }
});

});
}

runWorker().then((result) => {
console.log(工作线程信息:, result);
}).catch((err) => {
console.error('错误:', err.message);
});