Kill child processes in Go

go.mod

module sample

go 1.22.1

main.go

package main

import (
	"context"
	"fmt"
	"os"
	"os/exec"
	"os/signal"
	"sample/task"
	"sync"
	"time"
)

func main() {
	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
	defer stop()
	cmd := exec.CommandContext(ctx, "bash", "-c", "yes &")
	cmd.SysProcAttr = task.NewSysProcAttr()
	cmd.Cancel = sync.OnceValue(func() error {
		return task.Kill(cmd.Process.Pid)
	})
	cmd.WaitDelay = 3 * time.Second

	err := cmd.Start()
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		return
	}

	time.Sleep(5 * time.Second)

	err = cmd.Cancel()
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
	}

	cmd.Wait()
}

task.go

//go:build !windows && (linux || darwin)

package task

import "syscall"

func NewSysProcAttr() *syscall.SysProcAttr {
	return &syscall.SysProcAttr{Setpgid: true}
}

func Kill(pid int) error {
	return syscall.Kill(-pid, syscall.SIGKILL)
}

task_windows.go

package task

import (
	"os/exec"
	"strconv"
	"syscall"
)

func NewSysProcAttr() *syscall.SysProcAttr {
	return nil
}

func Kill(pid int) error {
	return exec.Command("taskkill", "/T", "/F", "/PID", strconv.Itoa(pid)).Run()
}