這篇文章給大家介紹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 實現一個連接池就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。