流式操作系统/exec.Command

Question

我想构建一个类似于 unix 工具time 的基准测试工具。我目前拥有的是这样的：

package main

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

func main() {
    command := os.Args[1]
    args := os.Args[2:]
    cmd := exec.Command(command, args...)
    start_time := time.Now().UnixNano()
    stdout, err := cmd.Output()

    if err != nil {
        println(err.Error())
        return
    }

    print(string(stdout))
    total_time := int64(time.Nanosecond) * (time.Now().UnixNano() - start_time) / int64(time.Millisecond)
    fmt.Println(total_time)
}

我的问题是，输出不是流式传输，而是一次全部打印出来，而且对于某些程序来说，这很奇怪。

Answer 1

最好的方法是完全避免接触流。您可以直接为 stdout 和 stderr 传递您自己的文件描述符。

package main

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

func main() {
    command := os.Args[1]
    args := os.Args[2:]
    cmd := exec.Command(command, args...)
    cmd.Stdout = os.Stdout
    cmd.Stderr = os.Stderr
    start_time := time.Now()

    err := cmd.Run()
    if err != nil {
        println(err.Error())
        return
    }

    total_time := int64(time.Since(start_time) / time.Millisecond)
    fmt.Println(total_time)
}

---

此外，永远不应该使用 print()。相反，请使用 fmt.Print* 函数之一。

---

total_time := int64(time.Nanosecond) * (time.Now().UnixNano() - start_time) / int64(time.Millisecond)

这条线非常混乱。 int64(time.Nanosecond) * x 解析为 1 * x 或只是 x。

Answer 2

我不确定您在此处尝试使用 print(string(stdout)) 做什么，但这是不必要的，您面临的问题可能是那里的一些误用的副作用。

func main() {
    command := os.Args[1]
    args := os.Args[2:]
    cmd := exec.Command(command, args...)
    cmd.Stdout = os.Stdout
    cmd.Stderr = os.Stderr
    start_time := time.Now().UnixNano()

    if err != nil {
        fmt.Println(err.Error())
        return
    }

    total_time := int64(time.Nanosecond) * (time.Now().UnixNano() - start_time) / int64(time.Millisecond)
    fmt.Println(total_time)
}