背景
是这样的，最近在研究一个定时任务系统的改造，可能有点像jenkins做到的那种吧。
可以输入shell命令，也可以执行py脚本等等，相比之前来说，也要能够及时停止！
但是遇到了这么个问题，golang执行py脚本的时候获取不到脚本的输出。
首先来看看go里面怎么运行shell脚本吧，我比较喜欢执行全部命令。
普通用法（一次性获取所有输出）
package main
import (
"fmt"
"os/exec"
)
func main() {
Command("ls")
}
// 这里为了简化，我省去了stderr和其他信息
func Command(cmd string) error {
c := exec.Command("bash", "-c", cmd)
// 此处是windows版本
// c := exec.Command("cmd", "/C", cmd)
output, err := c.CombinedOutput()
fmt.Println(string(output))
return err
}
可以看到，当前命令执行的是输出当前目录下的文件/文件夹
image.png
实时显示
效果图:
image.png
package main
import (
"bufio"
"fmt"
"io"
"os/exec"
)
func main() {
// 执行ping baidu的命令, 命令不会结束
Command("ping www.baidu.com")
}
func Command(cmd string) error {
c := exec.Command("bash", "-c", cmd)
stdout, err := c.StdoutPipe()
if err != nil {
return err
}
go func() {
reader := bufio.NewReader(stdout)
for {
readString, err := reader.ReadString('\n')
if err != nil || err == io.EOF {
break
}
fmt.Print(readString)
}
}()
return c.Run()
}
可关闭+实时输出
package main
import (
"bufio"
"context"
"fmt"
"io"
"os/exec"
"time"
)
func main() {
ctx, cancel := context.WithCancel(context.Background())
go func(cancelFunc context.CancelFunc) {
time.Sleep(3 * time.Second)
cancelFunc()
}(cancel)
Command(ctx, "ping www.baidu.com")
}
func Command(ctx context.Context, cmd string) error {
c := exec.CommandContext(ctx, "bash", "-c", cmd)
stdout, err := c.StdoutPipe()
if err != nil {
return err
}
go func() {
reader := bufio.NewReader(stdout)
for {
// 其实这段去掉程序也会正常运行，只是我们就不知道到底什么时候Command被停止了，而且如果我们需要实时给web端展示输出的话，这里可以作为依据 取消展示
select {
// 检测到ctx.Done()之后停止读取
case <-ctx.Done():
if ctx.Err() != nil {
fmt.Printf("程序出现错误: %q", ctx.Err())
} else {
fmt.Println("程序被终止")
}
return
default:
readString, err := reader.ReadString('\n')
if err != nil || err == io.EOF {
break
}
fmt.Print(readString)
}
}
}()
return c.Run()
}
效果图:
image.png
可以看到输出了3次（1秒1次）之后程序就被终止了，确切的说是读取输出流的循环结束了。
执行Python脚本(阻塞)
其实很简单，只要python -u xxx.py这样执行就可以了, -u参数
简单的说就是python的输出是有缓存的，-u会强制往标准流输出，当Python脚本阻塞的时候
也不会拿不到输出！
其他
"bash" 和"-c"，据我的观察，这2个参数代表在当前cmd窗口执行，而不加这2个参数，直接上shell的话，会启动一个新窗口，目前观察是stdout拿不到数据。
有疑问加站长微信联系（非本文作者）
