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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
/**
 * Copyright(c) node-modules and other contributors.
 * MIT Licensed
 *
 * Authors:
 *   fengmk2 <m@fengmk2.com> (http://fengmk2.com)
 */
 
"use strict";
 
/**
 * Module dependencies.
 */
 
var cluster = require('cluster');
 
module.exports = reload;
 
// Windows not support SIGQUIT https://nodejs.org/api/process.html#process_signal_events
var KILL_SIGNAL = 'SIGTERM';
var reloading = false;
var reloadPedding = false;
function reload(count) {
  if (reloading) {
    reloadPedding = true;
    return;
  }
  if (!count) {
    count = require('os').cpus().length;
  }
  reloading = true;
  // find out all alive workers
  var aliveWorkers = [];
  var worker;
  for (var id in cluster.workers) {
    worker = cluster.workers[id];
    if (worker.state === 'disconnected') {
      continue;
    }
    aliveWorkers.push(worker);
  }
 
  var firstWorker;
  var newWorker;
 
  function reset() {
    // don't leak
    newWorker.removeListener('listening', reset);
    newWorker.removeListener('error', reset);
 
    if (firstWorker) {
      // console.log('firstWorker %s %s', firstWorker.id, firstWorker.state);
      firstWorker.kill(KILL_SIGNAL);
      setTimeout(function () {
        firstWorker.process.kill(KILL_SIGNAL);
      }, 100);
    }
    reloading = false;
    if (reloadPedding) {
      // has reload jobs, reload again
      reloadPedding = false;
      reload(count);
    }
  }
 
  firstWorker = aliveWorkers[0];
  newWorker = cluster.fork();
  newWorker.on('listening', reset).on('exit', reset);
 
  // kill other workers
  for (var i = 1; i < aliveWorkers.length; i++) {
    worker = aliveWorkers[i];
    // console.log('worker %s %s', worker.id, worker.state);
    worker.kill(KILL_SIGNAL);
  }
 
  // keep workers number as before
  var left = count - 1;
  for (var j = 0; j < left; j++) {
    cluster.fork();
  }
}