在 Go 语言中，通过线程并行执行任务时，输出可能会混杂在一起。为了按顺序显示输出，可以使用并发模式或缓冲输出。并发模式允许多个任务同时执行，而缓冲输出则将任务的输出暂存在缓冲区中，以便按顺序打印。

我正在尝试用 go 实现流程自动化。我已经能够实现线程并相应地执行该过程，但是输出是混合和匹配的。

我想知道是否有一种方法可以显示程序根据程序进程生成的输出。因此，如果任务 a 在任务 b 之前完成，我们会在 b 之前显示 a 的输出，反之亦然。

package main

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

var url string
var wg sync.WaitGroup

func nikto() {
    cmd := exec.Command("nikto", "-h", url)
    cmd.Stdout = os.Stdout
    err := cmd.Run()
    if err != nil {
        log.Fatal(err)
    }
    wg.Done()
}

func whois() {

    cmd := exec.Command("whois", "google.co")
    cmd.Stdout = os.Stdout
    err := cmd.Run()
    if err != nil {
        log.Fatal(err)
    }
    wg.Done()
}
func main() {
    fmt.Printf("Please input URL")
    fmt.Scanln(&url)
    wg.Add(1)
    go nikto()
    wg.Add(1)
    go whois()
    wg.Wait()
}

解决方案

在您的进程中，您将 os.stdout 文件描述符直接传递给您调用以运行子进程的命令。这意味着子进程的 stdout 管道将直接连接到 go 程序的标准输出，并且如果两个子进程同时写入，则可能会交错。

解决这个问题的最简单方法需要你在 go 程序中缓冲子进程的 stdout 管道的输出，这样你就可以拦截输出并在打印时进行控制。

os/exec 包中的 cmd 类型提供了一个函数调用 Output()，它将调用子进程并以字节切片形式返回 stdout 的内容。您可以轻松调整代码来实现此模式并处理结果，例如：

func whois() {
    cmd := exec.command("whois", "google.co")
    out, err := cmd.output()
    if err != nil {
        log.fatal(err)
    }
    fmt.println(out)
    wg.done()
}

输出交错

如果使用 fmt 包中的函数来打印输出，则有 no guarantee 个对 fmt.println 的并发调用不会交错。

为了防止交错，您可以选择串行化对 stdout 的访问，或使用可安全并发使用的记录器（例如 log 包）。下面是 go 进程中序列化访问 stdout 的示例：

package main

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

var url string

func nikto(outChan chan<- []byte) {
    cmd := exec.Command("nikto", "-h", url)
    bs, err := cmd.Output()
    if err != nil {
        log.Fatal(err)
    }
    outChan <- bs
}

func whois(outChan chan<- []byte) {
    cmd := exec.Command("whois", "google.com")
    bs, err := cmd.Output()
    if err != nil {
        log.Fatal(err)
    }
    outChan <- bs
}

func main() {
    outChan := make(chan []byte)

    fmt.Printf("Please input URL")
    fmt.Scanln(&url)
    go nikto(outChan)
    go whois(outChan)

    for i := 0; i < 2; i++ {
        bs := <-outChan
        fmt.Println(string(bs))
    }
}

以上就是《Go中如何完成睡眠操作》的详细内容，更多关于的资料请关注golang学习网公众号！