kimi-free-api/src/daemon.ts

/**
 * 守护进程
 */

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();  //创建进程