使用 goroutine 进行并发

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《使用 goroutine 进行并发》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

我正在尝试使用 go 自动化我的侦察工具。到目前为止，我可以在 kali 中运行两个基本工具（nikto/whois）。现在我希望它们并行执行，而不是等待一个函数完成。读了一下之后，我发现这可以通过使用 goroutine 来实现。但我的代码似乎不起作用：

package main

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

var url string

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

func whois() {
    cmd := exec.Command("whois","google.co")
    cmd.Stdout = os.Stdout
    err := cmd.Run()
    if err !=nil {
        log.Fatal(err)
    }
}

func main(){
    fmt.Printf("Please input URL")
    fmt.Scanln(&url)
    nikto()
    go whois()
}

我确实明白，这里 go whois() 会执行到 main() 为止，但我仍然看不到它们都并行执行。

解决方案

如果我正确理解你的问题，你想并行执行 nikto() 和 whois() 并等待两者返回。要等到一组 goroutine 完成，sync.WaitGroup 是归档的好方法。对于您来说，它可能看起来像这样：

func main(){
    fmt.Printf("Please input URL")
    fmt.Scanln(&url)

    var wg sync.WaitGroup
    wg.Add(2)
    go func() {
        defer wg.Done()
        nikto()
    }()
    go func() {
        defer wg.Done()
        whois()
    }()
    wg.Wait()
}

这里 wg.add(2) 告诉 waitgroup，我们将等待 2 个 goroutine。

然后你的两个函数在小包装函数中被调用，一旦每个函数完成，它也会调用 wg.done() 。这告诉 waitgroup 该函数已完成。 defer 关键字只是告诉 go，一旦周围的函数返回，就执行函数调用。另请注意两次调用包装器函数之前的 go 关键字，这会导致执行发生在 2 个单独的 goroutine 中。

最后，一旦两个 goroutine 都启动了，主函数中对 wg.wait() 的调用就会阻塞，直到 wg.done() 被调用两次，这将在 nikto() 和 whois() 都调用一次时发生。 已经完成了。

理论要掌握，实操不能落！以上关于《使用 goroutine 进行并发》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！