golang 连接池 php,golang通用连接池的实现

golang的channel除了goroutine通信之外还有很多其他的功能，本文将实现一种基于channel的通用连接池。

更新

添加超时处理，需要实现GetActiveTime方法获取最后活跃时间。

何为通用？

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

io.Closer

GetActiveTime() time.Time

}

type factory func() (Poolable, error)

type Pool interface {

Acquire() (Poolable, error) // 获取资源

Release(Poolable) error // 释放资源

Close(Poolable) error // 关闭资源

Shutdown() error // 关闭池

}

type GenericPool struct {

sync.Mutex

pool chan Poolable

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 Poolable, maxOpen),

}

for i := 0; i < minOpen; i++ {

closer, err := factory()

if err != nil {

continue

}

p.numOpen++

p.pool

}

return p, nil

}

func (p *GenericPool) Acquire() (Poolable, error) {

if p.closed {

return nil, ErrPoolClosed

}

for {

closer, err := p.getOrCreate()

if err != nil {

return nil, err

}

// 如果设置了超时且当前连接的活跃时间+超时时间早于现在，则当前连接已过期

if p.maxLifetime > 0 && closer.GetActiveTime().Add(p.maxLifetime).Before(time.Now()) {

p.Close(closer)

continue

}

return closer, nil

}

}

func (p *GenericPool) getOrCreate() (Poolable, error) {

select {

case closer :=

return closer, nil

default:

}

p.Lock()

if p.numOpen >= p.maxOpen {

closer :=

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 Poolable) error {

if p.closed {

return ErrPoolClosed

}

p.pool

return nil

}

// 关闭单个资源

func (p *GenericPool) Close(closer Poolable) 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等等，非常方便！

项目地址