加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 大数据 > 正文

将cmd stdout和stderr作为字符串返回,而不是在golang中打印到控

发布时间:2020-12-16 19:04:01 所属栏目:大数据 来源:网络整理
导读:我正在从golang应用程序执行bash命令.现在stdout和stderr直接进入控制台: cmd.Stdout = os.Stdout cmd.Stderr = os.Stderr 但我希望stdout和stderr作为字符串变量从runBashCommandAndKillIfTooSlow函数返回,而不立即打印到控制台. 怎么实现这个? 代码: pa
我正在从golang应用程序执行bash命令.现在stdout和stderr直接进入控制台:
cmd.Stdout = os.Stdout 
cmd.Stderr = os.Stderr

但我希望stdout和stderr作为字符串变量从runBashCommandAndKillIfTooSlow函数返回,而不立即打印到控制台.
怎么实现这个?

代码:

package main

import (
    "fmt"
    "log"
    "os"
    "os/exec"
    "time"
)

func main() {
    ok,outString,errString := runBashCommandAndKillIfTooSlow("ls -la",2000)
    fmt.Println("ok")
    fmt.Println(ok)
    fmt.Println("outString")
    fmt.Println(outString)
    fmt.Println("errString")
    fmt.Println(errString)
}

/*
run bash command and kill it if it works longer than "killInMilliSeconds" milliseconds
*/
func runBashCommandAndKillIfTooSlow(command string,killInMilliSeconds time.Duration) (okResult bool,stdout,stderr string) {
    fmt.Println("running bash command...")
    fmt.Println(command)
    cmd := exec.Command("sh","-c",command)

    cmd.Stdout = os.Stdout // cmd.Stdout -> stdout
    cmd.Stderr = os.Stderr // cmd.Stderr -> stderr

    okResult = true

    err := cmd.Start()
    log.Printf("Waiting for command to finish...")
    done := make(chan error,1)
    go func() {
        done <- cmd.Wait()
    }()
    select {
    case <-time.After(killInMilliSeconds * time.Millisecond):
        if err := cmd.Process.Kill(); err != nil {
            log.Fatal("failed to kill: ",err)
            okResult = false
        }
        <-done // allow goroutine to exit
        // log.Println("process killed")
    case err := <-done:

        if err != nil {
            log.Printf("process done with error = %v",err)
            okResult = false
        }
    }
    if err != nil {
        log.Fatal(err)
        okResult = false
    }
    return
}

顺便说一下,如果程序太慢(killInMilliSeconds参数),程序应该保留杀死bash命令的能力.

将输出设置为 bytes.Buffer:
var outbuf,errbuf bytes.Buffer
cmd.Stdout = &outbuf
cmd.Stderr = &errbuf

运行该命令后,可以通过调用Buffer.String()方法将stdout和stderr作为字符串获取:

stdout := outbuf.String()
stderr := errbuf.String()

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读