可靠终止Go语言子进程：Unix系统中的进程组管理

package main

import (
    "bytes"
    "fmt"
    "os/exec"
    "path/filepath"
    "syscall"
    "time"
)

func main() {
    // 模拟一个会创建子进程且可能长时间运行的命令
    // 例如，一个长时间运行的脚本，或一个内部会启动其他进程的程序
    // 这里使用一个简单的sleep命令来模拟，实际场景可能是go test等
    cmd := exec.Command("sleep", "10") // 模拟一个会运行10秒的命令
    // 如果是go test html，它会启动go进程，go进程再启动测试进程
    // cmd := exec.Command("go", "test", "html") 

    var output bytes.Buffer
    cmd.Stdout = &output
    cmd.Stderr = &output

    fmt.Println("Starting command...")
    if err := cmd.Start(); err != nil {
        fmt.Printf("Failed to start command: %s\n", err)
        return
    }

    // 设置一个2秒的超时
    timer := time.AfterFunc(time.Second*2, func() {
        fmt.Printf("Timeout occurred: Nobody got time fo that\n")
        // 尝试杀死进程，但可能只杀死父进程
        if cmd.Process != nil {
            if err := cmd.Process.Signal(syscall.SIGKILL); err != nil {
                fmt.Printf("Error sending SIGKILL to main process: %s\n", err)
            }
            fmt.Printf("Sent SIGKILL to main process (PID: %d)\n", cmd.Process.Pid)
        }
    })
    defer timer.Stop() // 确保在cmd.Wait()完成时停止计时器

    err := cmd.Wait()
    if err != nil {
        fmt.Printf("Command finished with error: %s\n", err)
    } else {
        fmt.Printf("Command finished successfully.\n")
    }
    fmt.Printf("Output:\n%s\n", output.String())
    fmt.Printf("Done waiting.\n")
}

package main

import (
    "bytes"
    "fmt"
    "os/exec"
    "syscall" // 导入syscall包
    "time"
)

func main() {
    // 模拟一个会创建子进程且可能长时间运行的命令
    // 例如，一个shell脚本，它会启动后台进程
    // 或者像go test这样的命令
    // 这里使用一个简单的bash脚本，它会启动一个sleep子进程
    cmd := exec.Command("bash", "-c", "sleep 10 & echo 'Child process started'; wait") 
    // 关键设置：将命令放入一个新的进程组
    cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}

    var output bytes.Buffer
    cmd.Stdout = &output
    cmd.Stderr = &output

    fmt.Println("Starting command with process group management...")
    if err := cmd.Start(); err != nil {
        fmt.Printf("Failed to start command: %s\n", err)
        return
    }

    // 获取进程组ID (PGID)
    pgid, err := syscall.Getpgid(cmd.Process.Pid)
    if err != nil {
        fmt.Printf("Failed to get process group ID: %s\n", err)
        // 如果无法获取PGID，可以尝试杀死单个进程作为回退
        if cmd.Process != nil {
            _ = cmd.Process.Signal(syscall.SIGKILL)
        }
        return
    }

    // 设置一个2秒的超时
    timer := time.AfterFunc(time.Second*2, func() {
        fmt.Printf("Timeout occurred: Nobody got time fo that\n")
        // 向整个进程组发送SIGTERM信号，尝试优雅关闭
        // 负值的PGID表示向整个进程组发送信号
        if err := syscall.Kill(-pgid, syscall.SIGTERM); err != nil {
            fmt.Printf("Error sending SIGTERM to process group %d: %s\n", pgid, err)
            // 如果SIGTERM失败，可以尝试发送SIGKILL强制终止
            if err := syscall.Kill(-pgid, syscall.SIGKILL); err != nil {
                fmt.Printf("Error sending SIGKILL to process group %d: %s\n", pgid, err)
            }
        }
        fmt.Printf("Sent signal to process group (PGID: %d)\n", pgid)
    })
    defer timer.Stop() // 确保在cmd.Wait()完成时停止计时器

    err = cmd.Wait()
    if err != nil {
        fmt.Printf("Command finished with error: %s\n", err)
    } else {
        fmt.Printf("Command finished successfully.\n")
    }
    fmt.Printf("Output:\n%s\n", output.String())
    fmt.Printf("Done waiting.\n")
}