详解Golang并发控制的三种方案
Channel
Channel是Go在语言层面提供的一种协程间的通信方式，我们可以通过在协程中向管道写入数据和在待等待的协程中读取对应协程的次数来实现并发控制。
func main() {
intChan := make(chan int, 5)
waitCount := 5
for i := 0; i < waitCount; i++ {
go func() {
intChan <- 1
}()
}
for i := 0; i < waitCount; i++ {
<-intChan
}
fmt.Println("主进程结束")
}WaitGroup
waitgroup通常应用于等待一组“工作协程”结束的场景，waitgroup底层是由一个长度为3的数组实现的，其内部有两个计数器，一个是工作协程计数器、一个是坐等协程计数器，还有一个是信号量。工作协程全部运行结束后，工作协程计数器将置为0，会释放对应坐等协程次数的信号量。
两点注意：
Add()方法中的参数大小要于工作协程的数量相等，否则会导致坐等协程一直等待，触发死锁panic
Done()方法执行的次数要与Add()方法中的工作协程计数器的数量一致，否则当工作协程计数器<0时，会触发panic【panic: sync: negative WaitGroup counter】
func main() {
wg := sync.WaitGroup{}
wg.Add(2)
go func() {
time.Sleep(3 * time.Second)
fmt.Println("等待三分钟的协程结束了")
wg.Done()
}()
go func() {
time.Sleep(3 * time.Second)
fmt.Println("等待三分钟的协程结束了")
wg.Done()
}()
wg.Wait()
}Context
适用于一个协程派生出多个协程的情况，可以控制多级的goroutine。我们可以通过一个Context对象，对派生出来的树状goroutine进行统一管理，并且每个goroutine具有相同的上下文。做统一关闭操作、统一定时关闭、统一传值的操作。多个上下文协程之间可以互相嵌套配合。
golang实现了四种原生的上下文对象
emptyCtx: 该上下文对象一般是作为父节点的，如果没有父节点，我们通常使用context.Background()方法来获取emptyCtx对象，并将其作为创建其他节点的父节点。
cancelCtx: 该上下文对象可以关闭所有拥有同一个上下文的goroutine,通过在子协程中监听cancelCtx.Done方法，来结束所有的派生协程。具体代码看下方,我们通过WithCancel（）方法来获取该对象。
timerCtx：该上下文对象是对cancelCtx对象的进一步封装，比cancelCtx主动关闭之外，多了了一个定时关闭功能。我们可以通过WithTimeout()和WithDeadline()这两种方法来获取该对象。其中WithTimeout()和WithDeadline()这两种方法点是WithTimeout()是设置过一段时间关闭上下文，WithDeadline()是设置那一个时间点来关闭这一个上下文。
valueCtx:该上下文对象并不用于进行协程的控制，而是在多级协程之间进行值得传递，方便共享一些相同得上下文内容。
以上除emptyCtx外的上下文对象和获取实例的方法如下图所示:
Context示例代码
cancelCtx
我们在所有的派生协程中传入相同的cancelContext对象，并在每一个子协程中使用switch-case结构监听上下文对象是否关闭，如果上下文对象关闭了，ctx.Done()返回的管道就可以读取到一个元素，使所在的case语句可执行，之后退出switch结构，执行协程中的其他代码。
func main() {
ctx, cancelFunc := context.WithCancel(context.Background())
deadline, ok := ctx.Deadline()
fmt.Println(deadline, ok)
done := ctx.Done()
fmt.Println(reflect.TypeOf(done))
fmt.Println(done)
go HandelRequest(ctx)
//<-done 阻塞当前一层的goroutine
time.Sleep(5 * time.Second)
fmt.Println("all goroutines is stopping!")
cancelFunc()
err := ctx.Err()
fmt.Println(err) //context canceled
time.Sleep(5 * time.Second)
}
func HandelRequest(ctx context.Context) {
go WriteMysql(ctx)
go WriteRedis(ctx)
for {
select {
case <-ctx.Done():
fmt.Println("HandelRequest Done")
return
default:
fmt.Println("等一等,Handler正在执行中")
time.Sleep(2 * time.Second)
}
}
}
func WriteRedis(ctx context.Context) {
for {
select {
case <-ctx.Done():
fmt.Println("WriteRedis Done.")
return
default:
fmt.Println("等一等,Redis正在执行中")
time.Sleep(2 * time.Second)
}
}
}
func WriteMysql(ctx context.Context) {
for {
select {
case <-ctx.Done():
fmt.Println("WriteMysql Done.")
return
default:
fmt.Println("等一等,Mysql正在执行中")
time.Sleep(2 * time.Second)
}
}
}timerCtx
这里代码以WithTimeout举例，相比与我们之前的手动调用关闭，使用timerCtx定时上下文对象后，可以是实现到达指定的时间自动进行关闭的操作。
func main() {
deadline, _ := context.WithTimeout(context.Background(), 5*time.Second)
go HandelRequest(deadline)
time.Sleep(10 * time.Second)
}
func HandelRequest(ctx context.Context) {
go WriteMysql(ctx)
go WriteRedis(ctx)
for {
select {
case <-ctx.Done():
fmt.Println("HandelRequest Done")
return
default:
fmt.Println("等一等,Handler正在执行中")
time.Sleep(2 * time.Second)
}
}
}
func WriteRedis(ctx context.Context) {
for {
select {
case <-ctx.Done():
fmt.Println("WriteRedis Done.")
return
default:
fmt.Println("等一等,Redis正在执行中")
time.Sleep(2 * time.Second)
}
}
}
func WriteMysql(ctx context.Context) {
for {
select {
case <-ctx.Done():
fmt.Println("WriteMysql Done.")
return
default:
fmt.Println("等一等,Mysql正在执行中")
time.Sleep(2 * time.Second)
}
}
}
valueCtx
我们可以通过嵌套WithValue上下文，来进行多个key-value在派生协程中传递，共享常量
func main() {
ctx, cancelFunc := context.WithCancel(context.Background())
value1 := context.WithValue(ctx, "param1", 1)
value2 := context.WithValue(value1, "param2", 2)
go ReadContextValue(value2)
time.Sleep(10 * time.Second)
cancelFunc()
time.Sleep(5 * time.Second)
}
func ReadContextValue(ctx context.Context) {
fmt.Println(ctx.Value("param1"))
fmt.Println(ctx.Value("param2"))
}到此这篇关于详解Golang并发控制的三种方案的文章就介绍到这了,更多相关Golang并发控制内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要给大家介绍了关于golang如何实现mapreduce单进程版本的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。2018-01-01
社区不少人在谈论 golang 为毛不用try/catch模式，而采用苛刻的recovery、panic、defer组合，本文就来详细的介绍一下，感兴趣的可以了解一下2021-07-07
详解golang中的结构体编解码神器Mapstructure库
mapstructure是GO字典(map[string]interface{})和Go结构体之间转换的编解码工具,这篇文章主要为大家介绍一下Mapstructure库的相关使用,希望对大家有所帮助2023-09-09
这篇文章主要介绍了Go语言带缓冲的通道实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-04-04
Channel是go语言内置的一个非常重要的特性,也是go并发编程的两大基石之一,下面这篇文章主要给大家介绍了关于Go中channel的相关资料,需要的朋友可以参考下2023-04-04
Go语言中的init函数为开发者提供了一种在程序正式运行前初始化包级变量的机制,然而,由于init函数的特殊性,不当地使用它可能引起一系列问题,本文将深入探讨如何有效地使用init函数,列举常见误用并提供相应的避免策略2024-01-01
WaitGroup是Golang并发的两种方式之一,一个是Channel,另一个是WaitGroup,下面这篇文章主要给大家介绍了关于golang基础之waitgroup用法以及使用要点的相关资料,需要的朋友可以参考下2023-02-02
在Go语言中,switch语句除了常见的用法外,还有一种不常用但有趣的写法,这种写法中,switch后面不跟任何表达式,而每个case后面跟的是返回bool类型的函数调用表达式,这实际上是一个等价于switch true的用法,通过从上到下逐一比较case后的表达式是否为true来决定执行哪个分支2024-10-10
Go语言支持单常量和多常量的定义方法,通过const关键字实现,常量用于存储不变的值,如圆周率或固定的错误信息,旨在提高代码的维护性和可读性,感兴趣的可以了解一下2024-10-10
Golang 中的 archive/zip 包用于处理 ZIP 格式的压缩文件,提供了一系列用于创建、读取和解压缩 ZIP 格式文件的函数和类型,下面小编就来和大家讲解下常用函数吧2023-08-08
最新评论