// Pool 定義一個結構體 包含重量級鎖 有緩沖區(qū)Chanel 工廠方法 連接池關閉狀態(tài)
type Pool struct {
 m        sync.Mutex
 resource chan io.Closer
 factory  func() (io.Closer, error)
 closed   bool
}

var ErrPoolClosed = errors.New("連接池已關閉")


func New(fn func() (io.Closer, error), size uint) (*Pool, error) {
 //判斷size大小是否合法
 if size <= 0 {
  return nil, errors.New("size不合法")
 }
 //基于工廠方法和size創(chuàng)建資源池
 return &Pool{
  resource: make(chan io.Closer, size),
  factory:  fn,
 }, nil

}

func (p *Pool) Acquire() (io.Closer, error) {
 select {
 //如果channel有足夠的資源分配則直接返回
 case r, ok := <-p.resource:
  if !ok {
   log.Println("連接池已關閉")
   return nil, ErrPoolClosed
  }
  log.Println("拿到連接池共享資源")
  return r, nil
  //基于工廠方法創(chuàng)建全新的資源返回出去
 default:
  log.Println("資源不足，創(chuàng)建新的連接資源")
  return p.factory()
 }

}

// Release 上鎖 設置方法退出后解鎖 查看當前連接池是否已關閉，若關閉則直接將資源關閉 ，反之select查看能否將其存入緩沖區(qū)，若可以輸出入隊成功，反之輸出隊列已滿
func (p *Pool) Release(r io.Closer) {
 //上鎖確保關閉和歸還資源操作互斥
 p.m.Lock()
 //函數(shù)退出時解鎖
 defer p.m.Unlock()
 //如果資源池關閉則直接將當前資源關閉銷毀
 if p.closed {
  log.Println("連接池已關閉,直接銷毀當前資源")
  r.Close()
 }
 //將連接歸還，如果滿了則直接關閉銷毀
 select {
 case p.resource <- r:
  log.Println("連接歸還成功")
 default:
  log.Println("連接池已滿，資源直接銷毀")
  r.Close()
 }

}

// Close 方法 上鎖 設置方法退出后解鎖 遍歷所有資源將其關閉 然后再關閉連接池
func (p *Pool) Close() {
 p.m.Lock()

 defer p.m.Unlock()

 if p.closed {
  log.Println("連接池已關閉,直接銷毀當前資源")
  return
 }
 //設置為關閉
 p.closed = true
 //關閉資源
 close(p.resource)
 //遍歷資源池資源
 for r := range p.resource {
  r.Close()
 }

}

//設置最大協(xié)程數(shù)與資源池數(shù)為24
const maxGoroutines = 24
const poolResources = 24

//創(chuàng)建可關閉的數(shù)據(jù)庫連接
type dbConnection struct {
 ID int32
}
//對應的關閉方法
func (d *dbConnection) Close() error {
 log.Println("當前數(shù)據(jù)庫連接", d.ID, "已關閉")
 return nil
}

var idCounter int32

func createConnection() (io.Closer, error) {
 id := atomic.AddInt32(&idCounter, 1)
 return &dbConnection{ID: id}, nil
}

func main() {
 //創(chuàng)建maxGoroutines個WaitGroup
 var wg sync.WaitGroup
 wg.Add(maxGoroutines)
 //傳入createConnection方法和連接池大小poolResources創(chuàng)建數(shù)據(jù)庫連接池
 p, err := pool.New(createConnection, poolResources)
 if err != nil {
  log.Println(err)
 }
 //創(chuàng)建24個協(xié)程獲取資源
 for i := 0; i < maxGoroutines; i++ {
  go func(queryParam int) {
   queryData(queryParam, p)
   defer wg.Done()
  }(i)
 }
 //等待操作完成關閉連接池
 wg.Wait()
 log.Println("查詢完成")
 p.Close()

}
//queryData 基于連接池Acquire獲取資源，完成后通過Release歸還資源
func queryData(queryParam int, p *pool.Pool) {

 r, e := p.Acquire()
 if e != nil {
  log.Println(e)
  return
 }

 defer p.Release(r)

 time.Sleep(time.Duration(rand.Intn(1000)) * time.Millisecond)
 log.Println("查詢", queryParam, "使用連接", r.(*dbConnection).ID)
}