nezha/pkg/utils/proccess_group.go
2021-01-29 11:59:35 +08:00

32 lines
551 B
Go

// +build !windows
package utils
import (
"os/exec"
"syscall"
)
type ProcessExitGroup struct {
cmds []*exec.Cmd
}
func NewProcessExitGroup() (ProcessExitGroup, error) {
return ProcessExitGroup{}, nil
}
func (g ProcessExitGroup) Dispose() error {
for _, c := range g.cmds {
if err := syscall.Kill(-c.Process.Pid, syscall.SIGKILL); err != nil {
return err
}
}
return nil
}
func (g *ProcessExitGroup) AddProcess(cmd *exec.Cmd) error {
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
g.cmds = append(g.cmds, cmd)
return nil
}