go channel应用系列二:协程间信息同步

2023-08-29 16:49:19 浏览数 (1)

大家好,我是渔夫子。

今天给大家介绍的go channel的第二种应用:协程间同步信息。

通过channel,能够确保一个协程在另一个协程完成工作之后才能继续。如果需要在两个或多个协程之间共享数据的场景中,这种用法就特别有用,并且能够确保数据不会同时被多个协程修改非常重要。

我们先看一个简单的示例:

代码语言:javascript复制
package main

import (
    "fmt"
    "time"
)

func worker(done chan bool) {
    fmt.Print("working...")
    time.Sleep(time.Second)
    fmt.Println("done")

    done <- true
}

func main() {
    done := make(chan bool, 1)
    go worker(done)

    <-done
}

在这个示例中,我们创建了一个worker协程,同时在main协程中创建了一个done通道。当worker协程完成工作后,往done通道中发送了一个true,代表通知main协程worker执行完毕了。

开源项目中的应用

接下来我们看几个开源项目中的示例。

应用一:利用通道进行平滑关闭

在gin框架的example中,有一个关闭服务的示例,就是利用了通道来在两个协程间进行通讯的特性。如下:

这里就是在main协程中创建了一个quit通道,然后并将该quit通道传递给signal.Notify函数,然后通过<-quit阻塞等待signal.Notify函数执行完毕。在signal.Notify中其实是注册并监听syscall.SIGTERM信号,通过启动了一个新的协程来监听该信号。当该信号发生时,就往quit通道中写入一个os.Signal的数据。

应用二:fastcache中利用通道输出结果

在fastcache开源项目中,有个功能是将数据保存到文件中。在保存函数中用到了并发保存,同时需要将每个保存的结果输出。下面就是通过通道来接收每个协程的保存结果的功能。如下:

在上图中,首先在save函数中初始化了一个results通道,然后将saveBuckets的结果输出到results。在save函数的最下面,通过从results等待输出每次saveBuckets的结果。你看,这里就是通过results通道将子协程中的结果输出给save函数(父协程)了。

好了,今天通道的应用案例就分享到这里了。

0 人点赞