go-shell

easy execute shell, better os/exec

Feature

• simple api
• add timeout
• add stop()
• add yum api
• use channel to send stdout and stderr
• merge stdout and stderr to new output
• use sync.pool to reduce alloc buffer

Usage

😁 Look at the code for yourself

func TestCheckOutput(t *testing.T) {
	cmd := NewCommand("echo -n 123123")
	cmd.Run() // start and wait
	status := cmd.Status
assert.Equal(t, status.Output, "123123") 	assert.Equal(t, status.Stdout, "123123") 	assert.Equal(t, status.Stderr, "") }
func TestCheckStderr(t *testing.T) { 	cmd := NewCommand("echo -n \"123123\" >&2") 	cmd.Run() // start and wait 	status := cmd.Status
assert.Equal(t, status.Stdout, "") 	assert.Equal(t, status.Output, "123123") 	assert.Equal(t, status.Stderr, "123123") }
func TestDelayStop(t *testing.T) { 	start := time.Now() 	cmd := NewCommand("sleep 5;echo 123;sleep 123") 	cmd.Start() // async start
go func() { 		time.Sleep(1 * time.Second) 		cmd.Stop() 	}()
cmd.Wait() // wait 	end := time.Since(start) 	assert.Less(t, end.Seconds(), float64(2)) }
func TestRunTimeout(t *testing.T) { 	cmd := NewCommand("echo 123; sleep 5", WithTimeout(2)) 	cmd.Start() 	cmd.Wait() 	status := cmd.Status
assert.Equal(t, status.Error, ErrProcessTimeout) 	assert.Greater(t, status.CostTime.Seconds(), float64(2)) 	assert.Less(t, status.CostTime.Seconds(), float64(3)) }
func TestCheckStderr(t *testing.T) { 	cmd := NewCommand("echo -n \"123123\" >&2") // echo stderr 	cmd.Run() // start and wait 	status := cmd.Status
assert.Equal(t, status.Stdout, "") 	assert.Equal(t, status.Output, "123123") 	assert.Equal(t, status.Stderr, "123123") }
func TestCheckStream(t *testing.T) { 	stdoutChan := make(chan string, 100) 	incr := 0 	go func() { 		for line := range stdoutChan { 			incr++ 			fmt.Println(incr, line) 		} 	}()
cmd := exec.Command("bash", "-c", "echo 123;sleep 1;echo 456; echo 789") 	stdout := NewOutputStream(stdoutChan) 	cmd.Stdout = stdout 	cmd.Run()
assert.Equal(t, incr, 3) }
func TestCheckBuffer(t *testing.T) { 	cmd := exec.Command("bash", "-c", "echo 123") 	stdout := NewOutputBuffer() 	cmd.Stdout = stdout 	cmd.Run()
assert.Equal(t, stdout.buf.String(), "123\n") 	assert.Equal(t, stdout.Lines()[0], "123") }