go语言消息机制利用--实现连接池与goroutine同步

在go语言中,我们使用基于消息传递的通信方式(而不是大多数语言所使用的基于共享内存的通信方式)进行协程间通信。

消息管道(channel)属于go语言基本的数据类型,使用类型关键字(chan)进行定义,并发操作时线程安全。

我们可以利用消费者生产者模型,结合go语言的消息管道,实现同步。

1、连接池(以Redis为例)

package main

const poolSize = 10

func main() {
	// 初始化创建poolSize连接池的channel
	pool := make(chan redis.Conn, poolSize)

	// 通过goroutine并发执行1000个协程
	for i:=0; i<1000; i++ {
		go func() {
			// 先从连接池中取出一个连接,如果取不到则挂起等待
			rcon := <- pool
			// 处理逻辑
			rcon.dosomething
			// 完成后将链接放回连接池
			pool <- rcon
		}()
	}
}

2、goroutine并发操作后的同步

package main

const chanSize = 10

func main() {
	syncChan := make(chan int, chanSize)

	// 生产者
	// 异步做完任务后,释放资源
	for i := 0; i < chanSize; i++ {
		go func() {
			//dosomething
			syncChan <- 1
		}()
	}

	// 消费者,实现同步等待
	// 一开始取不到则挂起
	// 直到chanSize个资源全部获取完毕
	for i := 0; i < chanSize; i++ {
		<-syncChan
	}

	// after同步后的操作
}
例子来源于 大道至简—GO语言最佳实践

猜你喜欢

转载自blog.csdn.net/yuanlaidewo000/article/details/80746293