333
schangxiang@126.com
2025-09-19 18966e02fb573c7e2bb0c6426ed792b38b910940
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
'use strict';
 
const cluster = require('cluster');
const getExitFunction = require('./exit');
 
const init = Symbol('graceful-process-init');
 
module.exports = (options = {}) => {
  const logger = options.logger || console;
  let logLevel = (options.logLevel || 'info').toLowerCase();
  if (logger !== console) {
    // don't handle custom logger level
    logLevel = 'info';
  }
  const printLogLevels = {
    info: true,
    warn: true,
    error: true,
  };
  if (logLevel === 'warn') {
    printLogLevels.info = false;
  } else if (logLevel === 'error') {
    printLogLevels.info = false;
    printLogLevels.warn = false;
  }
  const label = options.label || `graceful-process#${process.pid}`;
 
  if (process[init]) {
    printLogLevels.warn && logger.warn('[%s] graceful-process init already', label);
    return;
  }
  process[init] = true;
 
  const exit = getExitFunction(options.beforeExit, logger, label);
 
  // https://github.com/eggjs/egg-cluster/blob/master/lib/agent_worker.js#L35
  // exit gracefully
  process.once('SIGTERM', () => {
    printLogLevels.info && logger.info('[%s] receive signal SIGTERM, exiting with code:0', label);
    exit(0);
  });
 
  process.once('exit', code => {
    const level = code === 0 ? 'info' : 'error';
    printLogLevels[level] && logger[level]('[%s] exit with code:%s', label, code);
  });
 
  if (cluster.worker) {
    // cluster mode
    // https://github.com/nodejs/node/blob/6caf1b093ab0176b8ded68a53ab1ab72259bb1e0/lib/internal/cluster/child.js#L28
    cluster.worker.once('disconnect', () => {
      // ignore suicide disconnect event
      if (cluster.worker.exitedAfterDisconnect) return;
      logger.error('[%s] receive disconnect event in cluster fork mode, exitedAfterDisconnect:false', label);
    });
  } else {
    // child_process mode
    process.once('disconnect', () => {
      // wait a loop for SIGTERM event happen
      setImmediate(() => {
        // if disconnect event emit, maybe master exit in accident
        logger.error('[%s] receive disconnect event on child_process fork mode, exiting with code:110', label);
        exit(110);
      });
    });
  }
};