golang中怎么利用channel实现一个连接池

这篇文章给大家介绍golang中怎么利用channel 实现一个连接池,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。

成都服务器托管,创新互联建站提供包括服务器租用、资阳移动机房、带宽租用、云主机、机柜租用、主机租用托管、CDN网站加速、域名注册等业务的一体化完整服务。电话咨询:18982081108

何为通用?

连接池的实现不依赖具体的实例,而依赖某个接口,本文的连接池选用的是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
}

关于golang中怎么利用channel 实现一个连接池就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。


分享文章:golang中怎么利用channel实现一个连接池
文章网址:http://pwwzsj.com/article/jscies.html