const readline = require('readline');

process.on('SIGINT', () => {
  // 我們通過 readline 來簡單地實現(xiàn)命令行里面的交互
  const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
  });
  rl.question('任務還沒執(zhí)行完，確定要退出嗎？', answer => {
    if (answer === 'yes') {
      console.log('任務執(zhí)行中斷，退出進程');
      process.exit(0);
    } else {
      console.log('任務繼續(xù)執(zhí)行...');
    }
    rl.close();
  });
});

// 模擬一個需要執(zhí)行 1 分鐘的任務
const longTimeTask = () => {
  console.log('task start...');
  setTimeout(() => {
    console.log('task end');
  }, 1000 * 60);
};

longTimeTask();

const cluster = require('cluster');
const http = require('http');
const numCPUs = require('os').cpus().length;
const process = require('process');

// 主進程代碼
if (cluster.isMaster) {
  console.log(`啟動主進程: ${process.pid}`);
  // 根據(jù) cpu 核數(shù)，創(chuàng)建工作進程
  for (let i = 0; i < numCPUs; i++) {
    cluster.fork();
  }
  // 監(jiān)聽工作進程退出事件
  cluster.on('exit', (worker, code, signal) => {
    console.log(`工作進程 ${worker.process.pid} 退出，錯誤碼: ${code || signal}, 重啟中...`);
    // 重啟子進程
    cluster.fork();
  });
}

// 工作進程代碼
if (cluster.isWorker) {
  // 監(jiān)聽未捕獲錯誤事件
  process.on('uncaughtException', error => {
    console.log(`工作進程 ${process.pid} 發(fā)生錯誤`, error);
    process.emit('disconnect');
    process.exit(1);
  });
  // 創(chuàng)建 web server
  // 各個工作進程都會監(jiān)聽端口 8000（Node.js 內(nèi)部會做處理，不會導致端口沖突）
  http.createServer((req, res) => {
    res.writeHead(200);
    res.end('hello world\n');
  }).listen(8000);
  console.log(`啟動工作進程: ${process.pid}`);
}

// exit-hook.js
// 保存需要執(zhí)行的退出任務
const tasks = [];
// 添加退出任務
const addExitTask = fn => tasks.push(fn);
const handleExit = (code, error) => {  
  // ...handleExit 的實現(xiàn)見下面
};
// 監(jiān)聽各種退出事件
process.on('exit', code => handleExit(code));
// 按照 POSIX 的規(guī)范，我們用 128 + 信號編號 得到最終的退出碼
// 信號編號參考下面的圖片，大家可以在 linux 系統(tǒng)下執(zhí)行 kill -l 查看所有的信號編號
process.on('SIGHUP', () => handleExit(128 + 1));
process.on('SIGINT', () => handleExit(128 + 2));
process.on('SIGTERM', () => handleExit(128 + 15));
// windows 下按下 ctrl+break 的退出信號
process.on('SIGBREAK', () => handleExit(128 + 21));
// 退出碼 1 代表未捕獲的錯誤導致進程退出
process.on('uncaughtException', error => handleExit(1, error));
process.on('unhandledRejection', error => handleExit(1, error));

// 標記是否正在退出，避免多次執(zhí)行
let isExiting = false;
const handleExit = (code, error) => {
  if (isExiting) return;
  isExiting = true;

  // 標記已經(jīng)執(zhí)行了退出動作，避免多次調(diào)用
  let hasDoExit = fasle;
  const doExit = () => {
      if (hasDoExit) return;
      hasDoExit = true
      process.nextTick(() => process.exit(code))
  }

  // 記錄有多少個異步任務
  let asyncTaskCount = 0;
  // 異步任務結(jié)束后，用戶需要調(diào)用的回調(diào)
  let ayncTaskCallback = () => {
      process.nextTick(() => {
        asyncTaskCount--
        if (asyncTaskCount === 0) doExit() 
      })
  }
  // 執(zhí)行所有的退出任務

  tasks.forEach(taskFn => {
      // 如果 taskFn 函數(shù)的參數(shù)個數(shù)大于 1，認為傳遞了 callback 參數(shù)，是一個異步任務
      if (taskFn.length > 1) {
         asyncTaskCount++
         taskFn(error, ayncTaskCallback)
      } else {
          taskFn(error)
      }
  });

  // 如果存在異步任務
  if (asyncTaskCount > 0) {
      // 超過 10s 后，強制退出
      setTimeout(() => {
          doExit();
      }, 10 * 1000)
  } else {
      doExit()
  }
};

const http = require('http');

// 創(chuàng)建 web server
const server = http.createServer((req, res) => {
  res.writeHead(200);
  res.end('hello world\n');
}).listen(8000);

// 使用我們在上面開發(fā)的工具添加進程退出任務
addExitTask((error, callback) => {
   // 打印錯誤日志、觸發(fā)告警、釋放數(shù)據(jù)庫連接等
   console.log('進程異常退出'， error)
   // 停止接受新的請求
   server.close((error) => {
       if (error) {
         console.log('停止接受新請求錯誤', error)
       } else {
         console.log('已停止接受新的請求')
       }
   })
   // 比較簡單的做法是，等待一定的時間（這里我們等待 5s），讓存量請求執(zhí)行完畢
   // 如果要完全保證所有請求都處理完畢，需要記錄每一個連接，在所有連接都釋放后，才執(zhí)行退出動作
   // 可以參考開源庫 https://github.com/sebhildebrandt/http-graceful-shutdown
   setTimout(callback, 5 * 1000)
})