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等等，非常方便！
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对毛票票的支持。