nezha/cmd/agent/processgroup/process_group.go
2021-08-18 17:42:26 +08:00

32 lines
559 B
Go

// +build !windows
package processgroup
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
}