mirror of https://github.com/siddontang/go.git
37 lines
542 B
Go
37 lines
542 B
Go
|
package exec
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os/exec"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
//exec another process
|
||
|
|
//if wait d Duration, it will kill the process
|
||
|
|
//d is <= 0, wait forever
|
||
|
|
func ExecTimeout(d time.Duration, name string, args ...string) error {
|
||
|
|
cmd := exec.Command(name, args...)
|
||
|
|
|
||
|
|
if err := cmd.Start(); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if d <= 0 {
|
||
|
|
return cmd.Wait()
|
||
|
|
}
|
||
|
|
|
||
|
|
done := make(chan error)
|
||
|
|
go func() {
|
||
|
|
done <- cmd.Wait()
|
||
|
|
}()
|
||
|
|
|
||
|
|
select {
|
||
|
|
case <-time.After(d):
|
||
|
|
cmd.Process.Kill()
|
||
|
|
|
||
|
|
//wait goroutine return
|
||
|
|
return <-done
|
||
|
|
case err := <-done:
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|