af41e4d843
* kill process tree using syscall on windows & cleanup (#80) * kill process tree using syscall on windows & cleanup * use job api * add error check for cmd.Start * modularize monitor, reduce init usage * replace slices with sort * update gopsutil & other dependencies
55 lines
961 B
Go
55 lines
961 B
Go
//go:build !windows
|
|
|
|
package processgroup
|
|
|
|
import (
|
|
"os/exec"
|
|
"sync"
|
|
"syscall"
|
|
)
|
|
|
|
type ProcessExitGroup struct {
|
|
cmds []*exec.Cmd
|
|
}
|
|
|
|
func NewProcessExitGroup() (ProcessExitGroup, error) {
|
|
return ProcessExitGroup{}, nil
|
|
}
|
|
|
|
func NewCommand(arg string) *exec.Cmd {
|
|
cmd := exec.Command("sh", "-c", arg)
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
|
|
return cmd
|
|
}
|
|
|
|
func (g *ProcessExitGroup) Dispose() error {
|
|
var wg sync.WaitGroup
|
|
wg.Add(len(g.cmds))
|
|
|
|
for _, c := range g.cmds {
|
|
go func(c *exec.Cmd) {
|
|
defer wg.Done()
|
|
killChildProcess(c)
|
|
}(c)
|
|
}
|
|
|
|
wg.Wait()
|
|
return nil
|
|
}
|
|
|
|
func (g *ProcessExitGroup) AddProcess(cmd *exec.Cmd) error {
|
|
g.cmds = append(g.cmds, cmd)
|
|
return nil
|
|
}
|
|
|
|
func killChildProcess(c *exec.Cmd) {
|
|
pgid, err := syscall.Getpgid(c.Process.Pid)
|
|
if err != nil {
|
|
// Fall-back on error. Kill the main process only.
|
|
c.Process.Kill()
|
|
}
|
|
// Kill the whole process group.
|
|
syscall.Kill(-pgid, syscall.SIGTERM)
|
|
c.Wait()
|
|
}
|