Node.js中正确终止Go进程的实践指南

go build -o main main.go

const { spawn } = require('child_process');
const path = require('path');

let goProcess;

// 假设Go二进制文件在当前目录或指定路径
const goBinaryPath = path.join(__dirname, 'main'); // 或 './main.exe' for Windows

function startGoProcess() {
    // 确保二进制文件存在
    // 实际应用中可能需要先执行 go build
    // const buildProcess = spawn('go', ['build', '-o', 'main', 'main.go']);
    // buildProcess.on('close', (code) => {
    //     if (code === 0) {
    //         console.log('Go binary built successfully.');
    //         goProcess = spawn(goBinaryPath);
    //         console.log(`Go process started with PID: ${goProcess.pid}`);
    //         setupProcessListeners();
    //     } else {
    //         console.error(`Go build failed with code: ${code}`);
    //     }
    // });

    // 简化示例，假设二进制已存在
    goProcess = spawn(goBinaryPath);

    console.log(`Go process started with PID: ${goProcess.pid}`);
    setupProcessListeners();
}

function setupProcessListeners() {
    goProcess.stdout.on('data', (data) => {
        console.log(`Go stdout: ${data}`);
    });

    goProcess.stderr.on('data', (data) => {
        console.error(`Go stderr: ${data}`);
    });

    goProcess.on('close', (code) => {
        console.log(`Go process exited with code: ${code}`);
        goProcess = null;
    });

    goProcess.on('error', (err) => {
        console.error(`Failed to start Go process: ${err}`);
    });
}

function killGoProcess() {
    if (goProcess) {
        console.log(`Attempting to kill Go process with PID: ${goProcess.pid}`);
        // 发送 SIGINT 信号，Go应用可以捕获此信号进行优雅关闭
        goProcess.kill('SIGINT');
        // 如果 SIGINT 无效，可以尝试 SIGTERM 或 SIGKILL
        // setTimeout(() => {
        //     if (goProcess) {
        //         console.log('Go process did not respond to SIGINT, sending SIGTERM...');
        //         goProcess.kill('SIGTERM');
        //     }
        // }, 5000); // 5秒后如果未关闭则发送SIGTERM
    } else {
        console.log('No Go process is running.');
    }
}

// 示例用法
startGoProcess();

// 模拟一段时间后终止进程
setTimeout(() => {
    killGoProcess();
}, 10000); // 10秒后尝试终止Go进程

// main.go (Go应用程序示例)
package main

import (
    "fmt"
    "os"
    "os/signal"
    "syscall"
    "time"
)

func main() {
    fmt.Println("Go application started...")

    // 创建一个通道用于接收操作系统信号
    sigChan := make(chan os.Signal, 1)
    // 注册要监听的信号，例如 SIGINT (Ctrl+C) 和 SIGTERM
    signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

    // 启动一个goroutine模拟工作
    go func() {
        for {
            fmt.Println("Go application is working...")
            time.Sleep(2 * time.Second)
        }
    }()

    // 阻塞等待信号
    sig := <-sigChan
    fmt.Printf("Received signal: %v. Shutting down...\n", sig)

    // 在这里执行清理工作，例如关闭数据库连接、保存数据等
    time.Sleep(1 * time.Second) // 模拟清理工作

    fmt.Println("Go application gracefully shut down.")
    os.Exit(0)
}