golang实现基于channel的通用连接池详解
前言
golang的channel除了goroutine通信之外还有很多其他的功能,本文将实现一种基于channel的通用连接池。下面话不多说了,来一起看看详细的介绍吧。
功能
*连接池中连接类型为interface{},使得更加通用
*链接的最大空闲时间,超时的链接将关闭丢弃,可避免空闲时链接自动失效问题
*使用channel处理池中的链接,高效
何为通用?
连接池的实现不依赖具体的实例,而依赖某个接口,本文的连接池选用的是io.Closer接口,只要是实现了该接口的对象都可以被池管理。
当然,你可以实现基于interface{}的连接池,这样任何对象都可以被管理。
实现原理
将连接句柄存入channel中,由于缓存channel的特性,获取连接时如果池中有连接,将直接返回,如果池中没有连接,将阻塞或者新建连接(没超过最大限制的情况下)。
由于面向接口编程,所有创建连接的逻辑是不清楚的,这里需要传入一个函数,该函数返回一个io.Closer对象。
实现
由于并发问题,在需要操作池中互斥数据的时候需要加锁。
packagepool import( "errors" "io" "sync" "time" ) var( ErrInvalidConfig=errors.New("invalidpoolconfig") ErrPoolClosed=errors.New("poolclosed") ) typefactoryfunc()(io.Closer,error) typePoolinterface{ Acquire()(io.Closer,error)//获取资源 Release(io.Closer)error//释放资源 Close(io.Closer)error//关闭资源 Shutdown()error//关闭池 } typeGenericPoolstruct{ sync.Mutex poolchanio.Closer maxOpenint//池中最大资源数 numOpenint//当前池中资源数 minOpenint//池中最少资源数 closedbool//池是否已关闭 maxLifetimetime.Duration factoryfactory//创建连接的方法 } funcNewGenericPool(minOpen,maxOpenint,maxLifetimetime.Duration,factoryfactory)(*GenericPool,error){ ifmaxOpen<=0||minOpen>maxOpen{ returnnil,ErrInvalidConfig } p:=&GenericPool{ maxOpen:maxOpen, minOpen:minOpen, maxLifetime:maxLifetime, factory:factory, pool:make(chanio.Closer,maxOpen), } fori:=0;i=p.maxOpen{ closer:=<-p.pool p.Unlock() returncloser,nil } //新建连接 closer,err:=p.factory() iferr!=nil{ p.Unlock() returnnil,err } p.numOpen++ p.Unlock() returncloser,nil } //释放单个资源到连接池 func(p*GenericPool)Release(closerio.Closer)error{ ifp.closed{ returnErrPoolClosed } p.Lock() p.pool<-closer p.Unlock() returnnil } //关闭单个资源 func(p*GenericPool)Close(closerio.Closer)error{ p.Lock() closer.Close() p.numOpen-- p.Unlock() returnnil } //关闭连接池,释放所有资源 func(p*GenericPool)Shutdown()error{ ifp.closed{ returnErrPoolClosed } p.Lock() close(p.pool) forcloser:=rangep.pool{ closer.Close() p.numOpen-- } p.closed=true p.Unlock() returnnil }
结论
基于该连接池,可以管理所有io.Closer对象。比如memcached,redis等等,非常方便!
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对毛票票的支持。