增加 带 Cancel 的外部命令调用.

Signed-off-by: chen.yang <chen.yang@yuzhen-iot.com>
This commit is contained in:
chen.yang 2021-08-11 14:31:39 +08:00
parent a4217a7ded
commit d07f5e449a
1 changed files with 30 additions and 0 deletions

View File

@ -2,6 +2,8 @@
需要使用 os/exec 包。 需要使用 os/exec 包。
## 1.基本用法
示例如下: 示例如下:
```go ```go
@ -28,3 +30,31 @@ exec.Cmd 有 Start、Wait、Run、Output 等方法,其中
* Wait阻塞进程等待命令执行结束, 经常与 Star 配合使用; * Wait阻塞进程等待命令执行结束, 经常与 Star 配合使用;
* Run阻塞进程直到命令执行结束相当于 Start+Wait 的组合; * Run阻塞进程直到命令执行结束相当于 Start+Wait 的组合;
* Output阻塞进程直到命令执行结束比 Run 方法多返回程序的 stdout。 * Output阻塞进程直到命令执行结束比 Run 方法多返回程序的 stdout。
## 2.带 Cancel 的外部命令调用
```go
func MakeCmdWithCancel(name string, arg ...string) (*exec.Cmd, context.CancelFunc) {
ctx, cancel := context.WithCancel(context.Background())
cmd := exec.CommandContext(ctx, name, arg...)
cmd.Env = append(cmd.Env, "USER=root")
cmd.Env = append(cmd.Env, "HOME=/home/root")
cmd.Env = append(cmd.Env, "SHELL=/bin/sh")
cmd.Env = append(cmd.Env, "SHLVL=1")
cmd.Env = append(cmd.Env, "LOGNAME=root")
cmd.Env = append(cmd.Env, "XDG_RUNTIME_DIR=/run/user/0")
cmd.Env = append(cmd.Env, "PATH=/usr/local/bin:/usr/bin:/bin:/xt/scripts:/usr/local/sbin:/usr/sbin:/sbin:/xt/scripts")
return cmd, cancel
}
func main() {
// Sleep 20s
sleep, cancel := MakeCmdWithCancel("sleep", "20")
sleep.Start()
// 3s 后 Cancel 掉
time.Sleep(3 * time.Second)
cancel()
// 会立刻退出
sleep.Wait()
}
```