golang实现基于channel的通用连接池详解

Go  /  管理员 发布于 7年前   153

前言

golang的channel除了goroutine通信之外还有很多其他的功能，本文将实现一种基于channel的通用连接池。下面话不多说了，来一起看看详细的介绍吧。

功能

* 连接池中连接类型为interface{}，使得更加通用

* 链接的最大空闲时间，超时的链接将关闭丢弃，可避免空闲时链接自动失效问题

* 使用channel处理池中的链接，高效

何为通用？

连接池的实现不依赖具体的实例，而依赖某个接口，本文的连接池选用的是io.Closer接口，只要是实现了该接口的对象都可以被池管理。

当然，你可以实现基于interface{}的连接池，这样任何对象都可以被管理。

实现原理

将连接句柄存入channel中，由于缓存channel的特性，获取连接时如果池中有连接，将直接返回，如果池中没有连接，将阻塞或者新建连接（没超过最大限制的情况下）。

由于面向接口编程，所有创建连接的逻辑是不清楚的，这里需要传入一个函数，该函数返回一个io.Closer对象。

实现

由于并发问题，在需要操作池中互斥数据的时候需要加锁。

package poolimport (  "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等等，非常方便！

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对AIDI的支持。