Golang 中 Goroutines 和 Channels 的执行顺序控制方法-Golang-PHP中文网

Golang 中 Goroutines 和 Channels 的执行顺序控制方法

WBOY

发布： 2023-08-09 09:06:24

原创

1443 人浏览过

Golang 中 Goroutines 和 Channels 的执行顺序控制方法

在 Golang 编程中，Goroutine 和 Channel 是两个非常重要的概念。Goroutine 是一种轻量级的线程，可以在程序的执行过程中同时运行多个函数。而 Channel 则是用于 Goroutine 之间进行通信的机制。

在某些情况下，我们需要控制 Goroutine 和 Channel 的执行顺序，确保程序按照我们预期的方式运行。本文将介绍一些常见的方法，来实现 Goroutine 和 Channel 的顺序控制。

方法一：使用 WaitGroup

WaitGroup 是一个计数器，用于等待一组 Goroutine 的结束。它的工作原理是每启动一个 Goroutine，就将计数器加一；每个 Goroutine 执行完毕之后，将计数器减一。当计数器归零时，主线程会继续执行。

下面是一个使用 WaitGroup 进行顺序控制的示例代码：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup

    wg.Add(2)

    go func() {
        defer wg.Done()
        fmt.Println("Goroutine 1")
    }()

    go func() {
        defer wg.Done()
        fmt.Println("Goroutine 2")
    }()

    wg.Wait()

    fmt.Println("Main Goroutine")
}

登录后复制

在这个例子中，我们创建了两个 Goroutine，每个 Goroutine 打印一条消息。在每个 Goroutine 中，我们通过 defer wg.Done() 语句将计数器减一。最后，wg.Wait() 会阻塞主线程，直到所有 Goroutine 执行完毕。

运行以上代码，输出结果为：

Goroutine 1
Goroutine 2
Main Goroutine

登录后复制

可以看到，Goroutine 1 和 Goroutine 2 的执行顺序是不确定的，但是它们一定在主线程打印 "Main Goroutine" 之前完成。

方法二：使用无缓冲的 Channel

无缓冲的 Channel 可以用来阻塞 Goroutine 的执行，直到另一个 Goroutine 从 Channel 中接收到数据。通过构建一个 Goroutine 的依赖关系链，我们可以实现顺序执行多个 Goroutine。

下面是一个使用无缓冲 Channel 进行顺序控制的示例代码：

package main

import (
    "fmt"
)

func main() {
    ch1 := make(chan struct{})
    ch2 := make(chan struct{})

    go func() {
        fmt.Println("Goroutine 1")
        ch1 <- struct{}{}
    }()

    go func() {
        <-ch1
        fmt.Println("Goroutine 2")
        ch2 <- struct{}{}
    }()

    <-ch2

    fmt.Println("Main Goroutine")
}

登录后复制

在这个例子中，Goroutine 1 打印一条消息后，向 ch1 Channel 中发送一个空结构体。接着，Goroutine 2 从 ch1 Channel 中接收到数据后打印另一条消息，并向 ch2 Channel 中发送一个空结构体。最后，主线程从 ch2 Channel 中接收到数据后打印 "Main Goroutine"。

运行以上代码，输出结果为：

Goroutine 1
Goroutine 2
Main Goroutine

登录后复制

可以看到，Goroutine 1 和 Goroutine 2 的执行顺序是确定的，它们一定在主线程打印 "Main Goroutine" 之前完成。

方法三：使用有缓冲的 Channel

有缓冲的 Channel 允许我们在创建时指定一个容量，可以存储一定数量的数据。通过合理设置缓冲区大小，我们可以控制 Goroutine 的并发执行数量，实现顺序控制。

下面是一个使用有缓冲的 Channel 进行顺序控制的示例代码：

package main

import (
    "fmt"
)

func main() {
    ch := make(chan struct{}, 2)

    go func() {
        fmt.Println("Goroutine 1")
        ch <- struct{}{}
    }()

    go func() {
        fmt.Println("Goroutine 2")
        ch <- struct{}{}
    }()

    <-ch
    <-ch

    fmt.Println("Main Goroutine")
}

登录后复制

在这个例子中，我们创建了一个有容量为2的缓冲区的 ch Channel。在每个 Goroutine 中，我们发送了一个空结构体到 ch Channel 中。最后，主线程从 ch Channel 中接收两次数据后打印 "Main Goroutine"。

运行以上代码，输出结果为：