登录
首页 >  Golang >  Go问答

获取exec.StdoutPipe()输出流的颜色方案

来源:stackoverflow

时间:2024-03-22 15:19:21 188浏览 收藏

在 Go 中获取通过 `exec.StdoutPipe()` 传递的命令输出的颜色方案时,使用 `bufio.NewScanner` 读取标准输出时,颜色信息会丢失。本文介绍了如何在 Go 中解决此问题,方法是将 `os.Stdout` 分配给 `cmd.Stdout`,从而允许输出使用正确的颜色进行打印。通过这种修改,程序可以无缝打印未知命令的输出,包括文本和颜色信息。

问题内容

我希望能够无缝打印 go 中通过 io.readcloser 传递的未知(用户定义)命令的输出。 bufio.newscanner 读取标准并正确打印文本,但是子进程打印的颜色没有记录并通过管道传递(或者我不知道如何访问它)。

我尝试使用 execerr := syscall.exec(binary, cmd.args, os.environ()) 但是由于它接管了 go 进程,所以我无法运行一组进程。

// spawngroup spawns a group of processes
func spawngroup(cmds []*exec.cmd) {
    spawnwg := &sync.waitgroup{}
    spawnwg.add(len(cmds))
    defer spawnwg.wait()
    for _, cmd := range cmds {
        go spawn(cmd, spawnwg)
    }
}

// spawn spawn a child process
func spawn(cmd *exec.cmd, wg *sync.waitgroup) {
    defer wg.done()
    stdout, _ := cmd.stdoutpipe()
    stderr, _ := cmd.stderrpipe()
    err := cmd.start()
    if err != nil {
        color.red(err.error())
    }
    scanner := bufio.newscanner(stdout)
    for scanner.scan() {
        fmt.println(scanner.text())
    }
    errscanner := bufio.newscanner(stderr)
    for errscanner.scan() {
        color.red(scanner.text())
    }
    cmd.wait()
}
例如,如果我尝试运行在数组中传递的两个命令,例如 sls offlinevue-cli-serviceserve 一切都会按预期工作,但记录的输出没有颜色。这两个进程都用颜色将一些输出打印到命令行。确切的命令数组是未知的,所以我需要一种方法来打印它们的确切输出。

我可以通过使用以下方法使其在节点中工作:

let cmd = await spawn(command, args, {
    stdio: 'inherit',
    shell: true,
    ...options
  });

我一直无法找到如何在 go 中做到这一点。

感谢您的任何建议或帮助,这是我第一次真正深入了解 go,但它似乎是一门出色的语言!


解决方案


更改生成代码以将 os.stdout 分配给 cmd.stdout 使输出能够使用正确的颜色进行打印。

func Spawn(cmd *exec.Cmd, wg *sync.WaitGroup) {
    defer wg.Done()
    cmd.Stdout = os.Stdout
    cmd.Stderr = os.Stderr
    err := cmd.Start()
    if err != nil {
        color.Red(err.Error())
    }
    defer cmd.Wait()
}```

到这里,我们也就讲完了《获取exec.StdoutPipe()输出流的颜色方案》的内容了。个人认为,基础知识的学习和巩固,是为了更好的将其运用到项目中,欢迎关注golang学习网公众号,带你了解更多关于的知识点!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>