前言
golang的channel除了goroutine通信之外还有很多其他的功能,本文将实现一种基于channel的通用连接池。下面话不多说了,来一起看看详细的介绍吧。
功能
* 连接池中连接类型为interface{},使得更加通用
* 链接的最大空闲时间,超时的链接将关闭丢弃,可避免空闲时链接自动失效问题
* 使用channel处理池中的链接,高效
何为通用?
连接池的实现不依赖具体的实例,而依赖某个接口,本文的连接池选用的是io.Closer接口,只要是实现了该接口的对象都可以被池管理。
当然,你可以实现基于interface{}的连接池,这样任何对象都可以被管理。
实现原理
将连接句柄存入channel中,由于缓存channel的特性,获取连接时如果池中有连接,将直接返回,如果池中没有连接,将阻塞或者新建连接(没超过最大限制的情况下)。
由于面向接口编程,所有创建连接的逻辑是不清楚的,这里需要传入一个函数,该函数返回一个io.Closer对象。
实现
由于并发问题,在需要操作池中互斥数据的时候需要加锁。
package pool import ( "errors" "io" "sync" "time" ) var ( ErrInvalidConfig = errors.New("invalid pool config") ErrPoolClosed = errors.New("pool closed") ) type factory func() (io.Closer, error) type Pool interface { Acquire() (io.Closer, error) // 获取资源 Release(io.Closer) error // 释放资源 Close(io.Closer) error // 关闭资源 Shutdown() error // 关闭池 } type GenericPool struct { sync.Mutex pool chan io.Closer maxOpen int // 池中最大资源数 numOpen int // 当前池中资源数 minOpen int // 池中最少资源数 closed bool // 池是否已关闭 maxLifetime time.Duration factory factory // 创建连接的方法 } func NewGenericPool(minOpen, maxOpen int, maxLifetime time.Duration, factory factory) (*GenericPool, error) { if maxOpen <= 0 || minOpen > maxOpen { return nil, ErrInvalidConfig } p := &GenericPool{ maxOpen: maxOpen, minOpen: minOpen, maxLifetime: maxLifetime, factory: factory, pool: make(chan io.Closer, maxOpen), } for i := 0; i < minOpen; i++ { closer, err := factory() if err != nil { continue } p.numOpen++ p.pool <- closer } return p, nil } func (p *GenericPool) Acquire() (io.Closer, error) { if p.closed { return nil, ErrPoolClosed } for { closer, err := p.getOrCreate() if err != nil { return nil, err } // todo maxLifttime处理 return closer, nil } } func (p *GenericPool) getOrCreate() (io.Closer, error) { select { case closer := <-p.pool: return closer, nil default: } p.Lock() if p.numOpen >= p.maxOpen { closer := <-p.pool p.Unlock() return closer, nil } // 新建连接 closer, err := p.factory() if err != nil { p.Unlock() return nil, err } p.numOpen++ p.Unlock() return closer, nil } // 释放单个资源到连接池 func (p *GenericPool) Release(closer io.Closer) error { if p.closed { return ErrPoolClosed } p.Lock() p.pool <- closer p.Unlock() return nil } // 关闭单个资源 func (p *GenericPool) Close(closer io.Closer) error { p.Lock() closer.Close() p.numOpen-- p.Unlock() return nil } // 关闭连接池,释放所有资源 func (p *GenericPool) Shutdown() error { if p.closed { return ErrPoolClosed } p.Lock() close(p.pool) for closer := range p.pool { closer.Close() p.numOpen-- } p.closed = true p.Unlock() return nil }
结论
基于该连接池,可以管理所有io.Closer对象。比如memcached,redis等等,非常方便!
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对的支持。
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
《魔兽世界》大逃杀!60人新游玩模式《强袭风暴》3月21日上线
暴雪近日发布了《魔兽世界》10.2.6 更新内容,新游玩模式《强袭风暴》即将于3月21 日在亚服上线,届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。
艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕,并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时,他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中,玩家将会进入一个全新的海盗主题大逃杀式限时活动,其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场,作为一个独立于主游戏之外的活动,玩家可以用大逃杀的风格来体验《魔兽世界》,不分职业、不分装备(除了你在赛局中捡到的),光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式,玩家在加入海盗主题的预赛大厅区域前,可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹,《巨龙崛起》和《魔兽世界:巫妖王之怒 经典版》的玩家都可以获得奖励。