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

前言

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等等，非常方便！

总结

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