79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package main
|
||
|
||
import (
|
||
"bufio"
|
||
"context"
|
||
"fmt"
|
||
"io"
|
||
"os/exec"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
func main() {
|
||
//ctx, cancel := context.WithCancel(context.Background())
|
||
//go func(cancelFunc context.CancelFunc) {
|
||
// time.Sleep(10 * time.Second)
|
||
// cancelFunc()
|
||
//}(cancel)
|
||
//err := Command(ctx, "ping www.baidu.com")
|
||
//if err != nil {
|
||
// return
|
||
//}
|
||
//err := RunWithTimeout(10*time.Second, "ping www.baidu.com")
|
||
//if err != nil {
|
||
// return
|
||
//}
|
||
err := RunCmd("ping www.baidu.com")
|
||
if err != nil {
|
||
return
|
||
}
|
||
}
|
||
|
||
func RunWithTimeout(timeout time.Duration, cmd string) error {
|
||
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
||
defer cancel()
|
||
return Command(ctx, cmd)
|
||
}
|
||
|
||
func RunCmd(cmd string) error {
|
||
return Command(context.Background(), cmd)
|
||
}
|
||
|
||
func Command(ctx context.Context, cmd string) error {
|
||
// c := exec.CommandContext(ctx, "cmd", "/C", cmd)
|
||
c := exec.CommandContext(ctx, "bash", "-c", cmd) // mac linux
|
||
stdout, err := c.StdoutPipe()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
var wg sync.WaitGroup
|
||
wg.Add(1)
|
||
go func(wg *sync.WaitGroup) {
|
||
defer wg.Done()
|
||
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 {
|
||
return
|
||
}
|
||
fmt.Print(readString)
|
||
}
|
||
}
|
||
}(&wg)
|
||
err = c.Start()
|
||
wg.Wait()
|
||
return err
|
||
}
|