56 lines
857 B
Go
56 lines
857 B
Go
package utils
|
|
|
|
import (
|
|
"bufio"
|
|
"os/exec"
|
|
)
|
|
|
|
type Stdout string
|
|
type Stderr string
|
|
type ExitCode int
|
|
|
|
func Run(cmd *exec.Cmd) (chan interface{}, error) {
|
|
stdout, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
stderr, err := cmd.StderrPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := cmd.Start(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
c := make(chan interface{}, 10)
|
|
|
|
go func() {
|
|
scan := bufio.NewScanner(stdout)
|
|
for scan.Scan() {
|
|
c <- Stdout(scan.Text())
|
|
}
|
|
}()
|
|
go func() {
|
|
scan := bufio.NewScanner(stderr)
|
|
for scan.Scan() {
|
|
c <- Stderr(scan.Text())
|
|
}
|
|
}()
|
|
|
|
go func() {
|
|
defer close(c)
|
|
|
|
if err := cmd.Wait(); err != nil {
|
|
if exitError, ok := err.(*exec.ExitError); ok {
|
|
c <- ExitCode(exitError.ExitCode())
|
|
} else {
|
|
c <- ExitCode(1)
|
|
}
|
|
} else {
|
|
c <- ExitCode(0)
|
|
}
|
|
}()
|
|
|
|
return c, nil
|
|
}
|