JS `Process` `Signals` (`Node.js`)：处理操作系统信号与优雅退出

各位观众老爷们，大家好！今天咱们聊点硬核的，关于Node.js里的进程信号和优雅退出。这玩意儿听起来高大上，其实说白了，就是你的Node.js程序在跟操作系统“眉来眼去”的时候，怎么才能体面地分手，而不是一拍两散。

咱们先从信号说起。

一、什么是信号（Signals）？

想象一下，你正在家里舒舒服服地写代码，突然有人敲门，告诉你“着火了！”。这个“着火了！”就是信号。只不过，在操作系统里，发出信号的是操作系统或者其他进程，接收信号的是你的Node.js进程。

信号就是操作系统用来通知进程发生了某些事情的一种机制。这些事情可能很紧急，比如程序崩溃了，或者只是一个友好的请求，比如“请你关掉吧”。

常见的信号（Signals）

Node.js程序可以监听并处理很多种信号，但最常见的几个是：

SIGINT（中断信号）： 通常是用户按下Ctrl+C时发送的。
SIGTERM（终止信号）： 这是告诉进程“我要关闭你了，请做好准备”的信号。通常由kill命令或者进程管理工具发送。
SIGHUP（挂断信号）： 最初是用来通知进程终端已经断开连接的，现在通常用于重启服务。
SIGKILL（杀死信号）： 这是最狠的信号，直接杀死进程，不给任何机会。Node.js程序无法捕获或处理这个信号。

二、Node.js如何处理信号？

Node.js提供了process对象来处理信号。process对象有一个on()方法，可以用来监听特定的信号。

// 监听SIGINT信号 (Ctrl+C)
process.on('SIGINT', () => {
  console.log('接收到SIGINT信号，准备退出...');
  // 在这里执行清理工作，例如关闭数据库连接，保存数据等
  // ...
  process.exit(0); // 正常退出
});

// 监听SIGTERM信号
process.on('SIGTERM', () => {
  console.log('接收到SIGTERM信号，准备退出...');
  // 在这里执行清理工作
  // ...
  process.exit(0); // 正常退出
});

上面的代码片段展示了如何监听SIGINT和SIGTERM信号。当接收到这些信号时，会执行相应的回调函数。在回调函数中，你可以执行一些清理工作，例如关闭数据库连接，保存数据等。最后，调用process.exit(0)来正常退出程序。

注意： process.exit(0)表示正常退出，process.exit(1)表示异常退出。

三、优雅退出（Graceful Shutdown）

优雅退出是指在程序退出之前，完成所有必要的清理工作，以避免数据丢失或其他问题。这就像一个绅士，在离开之前会整理好房间，而不是直接拍屁股走人。

为什么需要优雅退出？

数据完整性： 如果程序正在写入数据库，突然被强制关闭，可能会导致数据损坏。
资源释放： 如果程序打开了文件、网络连接等资源，没有及时释放，可能会导致资源泄漏。
用户体验： 如果程序正在处理用户请求，突然被强制关闭，可能会导致用户请求失败。

如何实现优雅退出？

实现优雅退出的关键在于，在接收到退出信号之后，执行以下操作：

停止接收新的请求： 告诉负载均衡器或者其他客户端，不要再向这个进程发送新的请求。
完成正在处理的请求： 等待当前正在处理的请求完成，不要强制中断它们。
释放资源： 关闭数据库连接、文件、网络连接等资源。
记录日志： 记录退出日志，方便排查问题。

代码示例：一个简单的HTTP服务器的优雅退出

const http = require('http');

let server;
let connections = []; // 记录所有连接

const app = http.createServer((req, res) => {
  // 模拟处理请求
  setTimeout(() => {
    res.writeHead(200, { 'Content-Type': 'text/plain' });
    res.end('Hello, World!n');
    // 请求完成后，移除连接
    connections = connections.filter(c => c !== req.socket);
  }, 5000); // 模拟耗时操作

  connections.push(req.socket); // 保存连接
});

// 监听SIGINT信号
process.on('SIGINT', shutdown);
process.on('SIGTERM', shutdown);

function shutdown() {
  console.log('服务器正在关闭...');

  // 1. 停止接收新的连接
  server.close((err) => {
    if (err) {
      console.error('关闭服务器失败:', err);
      process.exit(1);
    }

    console.log('服务器已停止接收新的连接。');

    // 2. 关闭所有已有的连接
    Promise.all(connections.map(closeConnection)).then(() => {
      console.log('所有连接已关闭。');
      // 3. 其他清理工作，例如关闭数据库连接
      // ...
      console.log('清理工作已完成。');
      process.exit(0);
    }).catch(err => {
      console.error('关闭连接时发生错误:', err);
      process.exit(1);
    });
  });
}

function closeConnection(socket) {
  return new Promise((resolve, reject) => {
    socket.end(() => {
      console.log('连接已关闭。');
      resolve();
    });

    setTimeout(() => {
      console.error('强制关闭连接。');
      socket.destroy(); // 超时后强制关闭连接
      reject(new Error('连接关闭超时'));
    }, 5000); // 设置超时时间
  });
}

// 启动服务器
server = app.listen(3000, () => {
  console.log('服务器已启动，监听端口 3000');
});

代码解释：

connections数组： 用于记录所有客户端连接。
app.createServer()： 创建一个简单的HTTP服务器。
setTimeout()： 模拟一个耗时操作，模拟请求处理时间。
connections.push(req.socket)： 将每个新的连接添加到connections数组中。
connections = connections.filter(c => c !== req.socket)： 请求完成后，从connections数组中移除连接。
shutdown()函数： 处理退出逻辑。
- server.close()： 停止接收新的连接。
- Promise.all(connections.map(closeConnection))： 遍历所有连接，调用closeConnection()函数来关闭连接。使用Promise.all是为了确保所有连接都关闭之后，再执行后续的清理工作。
- closeConnection()函数： 关闭单个连接。使用socket.end()来正常关闭连接，并设置超时时间，如果超过超时时间，则强制关闭连接。
process.on('SIGINT', shutdown)和process.on('SIGTERM', shutdown)： 监听SIGINT和SIGTERM信号，并在接收到信号时调用shutdown()函数。

四、使用进程管理工具实现优雅重启

在生产环境中，通常使用进程管理工具（例如PM2、Docker等）来管理Node.js进程。这些工具可以帮助你实现优雅重启。

以PM2为例：

PM2提供了一个reload命令，可以实现零停机重启。

pm2 reload <app_name>

pm2 reload命令的原理是：

启动一个新的进程。
等待新的进程启动完成。
停止旧的进程。

在新的进程启动完成之前，PM2会将所有的请求转发到旧的进程。这样就可以保证服务不中断。

五、总结

处理操作系统信号和实现优雅退出是编写健壮的Node.js应用程序的关键。通过监听信号，我们可以捕获退出事件，并在程序退出之前执行必要的清理工作，从而避免数据丢失和其他问题。使用进程管理工具可以进一步简化优雅重启的流程。

一些建议：

不要阻塞信号处理函数： 信号处理函数应该尽快执行完成，不要执行耗时操作。如果需要执行耗时操作，可以使用setImmediate()或者process.nextTick()将任务放到事件循环的下一个阶段执行。
使用日志记录退出过程： 在程序退出之前，记录日志，方便排查问题。
测试优雅退出： 在开发和测试环境中，模拟退出信号，测试程序的优雅退出是否正常工作。
考虑使用现成的库： 有一些现成的库可以帮助你实现优雅退出，例如terminus。

表格总结：

信号	描述	默认行为	是否可捕获
SIGINT	中断信号 (Ctrl+C)	终止进程	是
SIGTERM	终止信号 (kill 命令)	终止进程	是
SIGHUP	挂断信号 (终端断开连接)	终止进程	是
SIGKILL	杀死信号 (kill -9 命令)	立即终止进程	否
SIGUSR1	用户自定义信号 1	终止进程	是
SIGUSR2	用户自定义信号 2	终止进程	是

记住，优雅退出是一种美德，它能让你的Node.js程序走得更远，活得更久。

好了，今天的讲座就到这里。希望大家有所收获！散会！

发表回复 取消回复

发表回复取消回复