-
Notifications
You must be signed in to change notification settings - Fork 36
/
exec_process.go
91 lines (74 loc) · 1.92 KB
/
exec_process.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package system
import (
"bytes"
"os/exec"
"strings"
"syscall"
bosherr "github.com/cloudfoundry/bosh-utils/errors"
boshlog "github.com/cloudfoundry/bosh-utils/logger"
)
const (
execProcessLogTag = "Cmd Runner"
)
type execProcess struct {
cmd *exec.Cmd
stdoutWriter *bytes.Buffer
stderrWriter *bytes.Buffer
keepAttached bool
quiet bool
pid int
pgid int //nolint:unused
logger boshlog.Logger
waitCh chan Result
}
func NewExecProcess(cmd *exec.Cmd, keepAttached bool, quiet bool, logger boshlog.Logger) *execProcess {
return &execProcess{
cmd: cmd,
stdoutWriter: bytes.NewBufferString(""),
stderrWriter: bytes.NewBufferString(""),
keepAttached: keepAttached,
quiet: quiet,
logger: logger,
}
}
func (p *execProcess) Wait() <-chan Result {
if p.waitCh != nil {
panic("Wait() must be called only once")
}
// Use buffer=1 to allow goroutine below to finish
p.waitCh = make(chan Result, 1)
go func() {
p.waitCh <- p.wait()
}()
return p.waitCh
}
func (p *execProcess) wait() Result {
// err will be non-nil if command exits with non-0 status
err := p.cmd.Wait()
stdout := p.stdoutWriter.String()
if !p.quiet {
p.logger.Debug(execProcessLogTag, "Stdout: %s", stdout)
}
stderr := p.stderrWriter.String()
if !p.quiet {
p.logger.Debug(execProcessLogTag, "Stderr: %s", stderr)
}
exitStatus := -1
waitStatus := p.cmd.ProcessState.Sys().(syscall.WaitStatus)
if waitStatus.Exited() {
exitStatus = waitStatus.ExitStatus()
} else if waitStatus.Signaled() {
exitStatus = 128 + int(waitStatus.Signal())
}
p.logger.Debug(execProcessLogTag, "Successful: %t (%d)", err == nil, exitStatus)
if err != nil {
cmdString := strings.Join(p.cmd.Args, " ")
err = bosherr.WrapComplexError(err, NewExecError(cmdString, stdout, stderr))
}
return Result{
Stdout: stdout,
Stderr: stderr,
ExitStatus: exitStatus,
Error: err,
}
}