qwen-free-api/src/daemon.ts
2024-03-24 01:48:06 +08:00

83 lines
3.1 KiB
TypeScript

/**
* 守护进程
*/
import process from 'process';
import path from 'path';
import { spawn } from 'child_process';
import fs from 'fs-extra';
import { format as dateFormat } from 'date-fns';
import 'colors';
const CRASH_RESTART_LIMIT = 600; //进程崩溃重启次数限制
const CRASH_RESTART_DELAY = 5000; //进程崩溃重启延迟
const LOG_PATH = path.resolve("./logs/daemon.log"); //守护进程日志路径
let crashCount = 0; //进程崩溃次数
let currentProcess; //当前运行进程
/**
* 写入守护进程日志
*/
function daemonLog(value, color?: string) {
try {
const head = `[daemon][${dateFormat(new Date(), "yyyy-MM-dd HH:mm:ss.SSS")}] `;
value = head + value;
console.log(color ? value[color] : value);
fs.ensureDirSync(path.dirname(LOG_PATH));
fs.appendFileSync(LOG_PATH, value + "\n");
}
catch(err) {
console.error("daemon log write error:", err);
}
}
daemonLog(`daemon pid: ${process.pid}`);
function createProcess() {
const childProcess = spawn("node", ["index.js", ...process.argv.slice(2)]); //启动子进程
childProcess.stdout.pipe(process.stdout, { end: false }); //将子进程输出管道到当前进程输出
childProcess.stderr.pipe(process.stderr, { end: false }); //将子进程错误输出管道到当前进程输出
currentProcess = childProcess; //更新当前进程
daemonLog(`process(${childProcess.pid}) has started`);
childProcess.on("error", err => daemonLog(`process(${childProcess.pid}) error: ${err.stack}`, "red"));
childProcess.on("close", code => {
if(code === 0) //进程正常退出
daemonLog(`process(${childProcess.pid}) has exited`);
else if(code === 2) //进程已被杀死
daemonLog(`process(${childProcess.pid}) has been killed!`, "bgYellow");
else if(code === 3) { //进程主动重启
daemonLog(`process(${childProcess.pid}) has restart`, "yellow");
createProcess(); //重新创建进程
}
else { //进程发生崩溃
if(crashCount++ < CRASH_RESTART_LIMIT) { //进程崩溃次数未达重启次数上限前尝试重启
daemonLog(`process(${childProcess.pid}) has crashed! delay ${CRASH_RESTART_DELAY}ms try restarting...(${crashCount})`, "bgRed");
setTimeout(() => createProcess(), CRASH_RESTART_DELAY); //延迟指定时长后再重启
}
else //进程已崩溃,且无法重启
daemonLog(`process(${childProcess.pid}) has crashed! unable to restart`, "bgRed");
}
}); //子进程关闭监听
}
process.on("exit", code => {
if(code === 0)
daemonLog("daemon process exited");
else if(code === 2)
daemonLog("daemon process has been killed!");
}); //守护进程退出事件
process.on("SIGTERM", () => {
daemonLog("received kill signal", "yellow");
currentProcess && currentProcess.kill("SIGINT");
process.exit(2);
}); //kill退出守护进程
process.on("SIGINT", () => {
currentProcess && currentProcess.kill("SIGINT");
process.exit(0);
}); //主动退出守护进程
createProcess(); //创建进程