Golang语言社区--golang通用连接池的实现

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

何为通用？

连接池的实现不依赖具体的实例，而依赖某个接口，本文的连接池选用的是io.Closer接口，只要是实现了该接口的对象都可以被池管理。
当然，你可以实现基于interface{}的连接池，这样任何对象都可以被管理。

实现原理

将连接句柄存入channel中，由于缓存channel的特性，获取连接时如果池中有连接，将直接返回，如果池中没有连接，将阻塞或者新建连接（没超过最大限制的情况下）。
由于面向接口编程，所有创建连接的逻辑是不清楚的，这里需要传入一个函数，该函数返回一个io.Closer对象。

实现

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

1. package pool
   
2. 
   
3. import (
   
4. "errors"
   
5. "io"
   
6. "sync"
   
7. "time"
   
8. )
   
9. var (
   
10. ErrInvalidConfig = errors.New("invalid pool config")
    
11. ErrPoolClosed = errors.New("pool closed")
    
12. type factory func() (io.Closer,error)
    
13. type Pool interface {
    
14. Acquire() (io.Closer,error) // 获取资源
    
15. Release(io.Closer) error // 释放资源
    
16. Close(io.Closer) error // 关闭资源
    
17. Shutdown() error // 关闭池
    
18. }
    
19. type GenericPool struct {
    
20. sync.Mutex
    
21. pool chan io.Closer
    
22. maxOpen int// 池中最大资源数
    
23. numOpen int// 当前池中资源数
    
24. minopen int// 池中最少资源数
    
25. closed bool // 池是否已关闭
    
26. maxLifetime time.Duration
    
27. factory factory // 创建连接的方法
    
28. func NewGenericPool(minopen,maxOpen int,maxLifetime time.Duration,factory factory) (*GenericPool,error) {
    
29. if maxOpen <= 0 || minopen > maxOpen {
    
30. return nil,ErrInvalidConfig
    
31. }
    
32. p := &GenericPool{
    
33. maxOpen: maxOpen,
    
34. minopen: minopen,serif;font-size:12px;line-height:1.8em;"> maxLifetime: maxLifetime,serif;font-size:12px;line-height:1.8em;"> factory: factory,serif;font-size:12px;line-height:1.8em;"> pool: make(chan io.Closer,maxOpen),serif;font-size:12px;line-height:1.8em;"> for i := 0; i < minopen; i++ {
    
35. closer,err := factory()
    
36. if err != nil {
    
37. continue
    
38. p.numOpen++
    
39. p.pool <- closer
    
40. return p,nil
    
41. func (p *GenericPool) Acquire() (io.Closer,serif;font-size:12px;line-height:1.8em;"> if p.closed {
    
42. for {
    
43. etorCreate()
    
44. return nil,err
    
45. // todo maxLifttime处理
    
46. return closer,serif;font-size:12px;line-height:1.8em;"> func (p *GenericPool) getorCreate() (io.Closer,serif;font-size:12px;line-height:1.8em;"> select {
    
47. case closer := <-p.pool:
    
48. default:
    
49. p.Lock()
    
50. if p.numOpen >= p.maxOpen {
    
51. closer := <-p.pool
    
52. p.Unlock()
    
53. // 新建连接
    
54. // 释放单个资源到连接池
    
55. func (p *GenericPool) Release(closer io.Closer) error {
    
56. return ErrPoolClosed
    
57. return nil
    
58. // 关闭单个资源
    
59. func (p *GenericPool) Close(closer io.Closer) error {
    
60. closer.Close()
    
61. p.numOpen--
    
62. // 关闭连接池，释放所有资源
    
63. func (p *GenericPool) Shutdown() error {
    
64. close(p.pool)
    
65. for closer := range p.pool {
    
66. p.closed = true
    
67. }

复制代码

结论

基于该连接池，可以管理所有io.Closer对象。比如memcached,redis等等，非常方便！

Golang语言社区论坛网址：www.golang.ltd