forked from rook/rook
Travis Nielsen
f2ecaa2bda
The helpers for executing a process have long required an actionName param which is not being used. Now we remove the old param while also cleaning up various other usages of the exec package. Signed-off-by: Travis Nielsen <tnielsen@redhat.com>
124 lines
2.8 KiB
Go
124 lines
2.8 KiB
Go
/*
|
|
Copyright 2016 The Rook Authors. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package utils
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/coreos/pkg/capnslog"
|
|
utilexec "github.com/rook/rook/pkg/util/exec"
|
|
)
|
|
|
|
var logger = capnslog.NewPackageLogger("github.com/rook/rook", "testutil")
|
|
|
|
// CommandArgs is a warpper for cmd args
|
|
type CommandArgs struct {
|
|
Command string
|
|
SubCommand string
|
|
CmdArgs []string
|
|
OptionalArgs []string
|
|
PipeToStdIn string
|
|
EnvironmentVariable []string
|
|
}
|
|
|
|
// CommandOut is a wrapper for cmd out returned after executing command args
|
|
type CommandOut struct {
|
|
StdOut string
|
|
StdErr string
|
|
ExitCode int
|
|
Err error
|
|
}
|
|
|
|
// ExecuteCommand executes a os command with stdin and returns output
|
|
func ExecuteCommand(cmdStruct CommandArgs) CommandOut {
|
|
logger.Infof("Running %s %v", cmdStruct.Command, cmdStruct.CmdArgs)
|
|
|
|
var outBuffer, errBuffer bytes.Buffer
|
|
|
|
cmd := exec.Command(cmdStruct.Command, cmdStruct.CmdArgs...)
|
|
|
|
cmd.Env = append(cmd.Env, cmdStruct.EnvironmentVariable...)
|
|
|
|
stdOut, err := cmd.StdoutPipe()
|
|
|
|
if err != nil {
|
|
return CommandOut{StdErr: errBuffer.String(), StdOut: outBuffer.String(), Err: err}
|
|
}
|
|
|
|
stdin, err := cmd.StdinPipe()
|
|
|
|
if err != nil {
|
|
return CommandOut{Err: err}
|
|
}
|
|
|
|
defer stdOut.Close()
|
|
|
|
scanner := bufio.NewScanner(stdOut)
|
|
go func() {
|
|
for scanner.Scan() {
|
|
outBuffer.WriteString(scanner.Text())
|
|
fmt.Printf("%s\n", scanner.Text())
|
|
}
|
|
}()
|
|
|
|
stdErr, err := cmd.StderrPipe()
|
|
|
|
if err != nil {
|
|
return CommandOut{StdErr: errBuffer.String(), StdOut: outBuffer.String(), Err: err}
|
|
}
|
|
|
|
defer stdErr.Close()
|
|
|
|
stdErrScanner := bufio.NewScanner(stdErr)
|
|
go func() {
|
|
for stdErrScanner.Scan() {
|
|
|
|
txt := stdErrScanner.Text()
|
|
|
|
if !strings.Contains(txt, "no buildable Go source files in") {
|
|
errBuffer.WriteString(txt)
|
|
fmt.Printf("%s\n", txt)
|
|
}
|
|
}
|
|
}()
|
|
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
return CommandOut{StdErr: errBuffer.String(), StdOut: outBuffer.String(), Err: err}
|
|
}
|
|
|
|
if cmdStruct.PipeToStdIn != "" {
|
|
stdin.Write([]byte(cmdStruct.PipeToStdIn))
|
|
stdin.Close()
|
|
}
|
|
|
|
err = cmd.Wait()
|
|
out := CommandOut{StdErr: errBuffer.String(), StdOut: outBuffer.String()}
|
|
if err != nil {
|
|
out.Err = err
|
|
if code, ok := utilexec.ExitStatus(err); ok {
|
|
out.ExitCode = code
|
|
}
|
|
}
|
|
|
|
return out
|
|
}
|