[Go] golang缓冲通道实现管理一组goroutine工作
程序员文章站
2023-11-06 08:15:28
通道1.当一个资源需要在goroutine之间共享时,通道在goroutine之间架起了一个管道2.无缓冲通道和有缓冲通道,make的第二个参数就是缓冲区大小3.无缓冲通道需要发送和接收都准备好,否则先执行的goroutine会阻塞等待4.有缓冲的通道,在缓冲区没满之前,发送和接收动作都不会阻塞,空 ......
通道
1.当一个资源需要在goroutine之间共享时,通道在goroutine之间架起了一个管道
2.无缓冲通道和有缓冲通道,make的第二个参数就是缓冲区大小
3.无缓冲通道需要发送和接收都准备好,否则先执行的goroutine会阻塞等待
4.有缓冲的通道,在缓冲区没满之前,发送和接收动作都不会阻塞,空的时候接收才会阻塞
time.now().unix() 当前时间戳 time.millisecond 毫秒
time.sleep(1 * time.second) 睡眠一秒
package main
import (
"fmt"
"math/rand"
"sync"
"time"
)
const (
//全局常量
numbergoroutines = 4 //使用goroutine的数量
taskload = 10 //要处理的工作量
)
var wg sync.waitgroup
//init函数会在main之前执行
func init() {
//初始化随机数
rand.seed(time.now().unix())
}
func main() {
//创建有缓冲的通道管理,缓冲区是10
tasks := make(chan string, taskload)
//启动4个goroutines来处理工作
wg.add(numbergoroutines) //加入计数信号量
for i := 1; i <= numbergoroutines; i++ {
go worker(tasks, i)
}
//主goroutine将10个字符串发送到通道,模拟分发给子goroutine的工作
for j := 1; j <= taskload; j++ {
tasks <- fmt.sprintf("task: %d", j)
}
//发送完成后,把通道关闭
close(tasks)
//如果一直往通道塞,子goroutine就可以一直工作,可以当队列用
// for {
// tasks <- fmt.sprintf("task: %d", rand.int63n(10000000000))
// }
//等待所有goroutines完成
wg.wait()
}
//处理工作
func worker(tasks chan string, worker int) {
defer wg.done()
//无限循环处理接收到的工作,可以处理完一个以后继续处理下一个
for {
//从已经关闭的通道中,依然可以接收数据,并且返回一个通道类型的零值,如果一个都没接收到的时候会阻塞
//接收到一个以后,会继续往下执行
task, ok := <-tasks
//判断通道是否清空并关闭
if !ok {
fmt.printf("worker: %d:关闭\n", worker)
//直接退出这个goroutine
return
}
//正式开始工作
fmt.printf("worker: %d:开始工作\n", worker)
//用随机睡眠来模拟执行中
sleep := rand.int63n(100)
time.sleep(time.duration(sleep) * time.millisecond)
//显示完成了
fmt.printf("worker: %d :完成 %s \n", worker, task)
}
}
上一篇: flash怎么设置动画部不循环播放?
下一篇: Docker