golang執行命令行(一)

golang中會經常遇到要 fork 子進程的需求。go 標准庫為我們封裝了 os/exec標准包,當我們要運行外部命令時應該優先使用這個庫。

執行 command

這里我簡單結合context 和 Cmd 模塊寫一個通用的執行 command 方法。代碼如下:

// RunCmd ...
func RunCmd(ctx context.Context, cmd *exec.Cmd) error {
	if err := cmd.Start(); err != nil {
		return err
	}

	errCh := make(chan error, 1)
	go func() {
		errCh <- cmd.Wait()
	}()

	done := ctx.Done()
	for {
		select {
		case <-done:
			done = nil
			pid := cmd.Process.Pid
			if err := syscall.Kill(-1*pid, syscall.SIGKILL); err != nil {
				return err
			}
		case err := <-errCh:
			if done == nil {
				return ctx.Err()
			}
			return err
		}
	}
}