78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
|
package pool
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
"io"
|
|||
|
|
"log"
|
|||
|
|
"sync"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
//一个安全的资源池，被管理的资源必须都实现io.Close接口
|
|||
|
|
type Pool struct {
|
|||
|
|
m sync.Mutex
|
|||
|
|
res chan io.Closer
|
|||
|
|
factory func() (io.Closer, error)
|
|||
|
|
closed bool
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var ErrPoolClosed = errors.New("资源池已经被关闭。")
|
|||
|
|
|
|||
|
|
//创建一个资源池
|
|||
|
|
func NewPool(fn func() (io.Closer, error), size uint) (*Pool, error) {
|
|||
|
|
if size <= 0 {
|
|||
|
|
return nil, errors.New("size的值太小了。")
|
|||
|
|
}
|
|||
|
|
return &Pool{
|
|||
|
|
factory: fn,
|
|||
|
|
res: make(chan io.Closer, size),
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//从资源池里获取一个资源
|
|||
|
|
func (p *Pool) Acquire() (io.Closer, error) {
|
|||
|
|
select {
|
|||
|
|
case r, ok := <-p.res:
|
|||
|
|
log.Println("Acquire:共享资源")
|
|||
|
|
if !ok {
|
|||
|
|
return nil, ErrPoolClosed
|
|||
|
|
}
|
|||
|
|
return r, nil
|
|||
|
|
default:
|
|||
|
|
log.Println("Acquire:新生成资源")
|
|||
|
|
return p.factory()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//关闭资源池，释放资源
|
|||
|
|
func (p *Pool) Close() {
|
|||
|
|
p.m.Lock()
|
|||
|
|
defer p.m.Unlock()
|
|||
|
|
if p.closed {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
p.closed = true
|
|||
|
|
//关闭通道，不让写入了
|
|||
|
|
close(p.res)
|
|||
|
|
//关闭通道里的资源
|
|||
|
|
for r := range p.res {
|
|||
|
|
r.Close()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
func (p *Pool) Release(r io.Closer) {
|
|||
|
|
//保证该操作和Close方法的操作是安全的
|
|||
|
|
p.m.Lock()
|
|||
|
|
defer p.m.Unlock()
|
|||
|
|
//资源池都关闭了，就省这一个没有释放的资源了，释放即可
|
|||
|
|
if p.closed {
|
|||
|
|
r.Close()
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
select {
|
|||
|
|
case p.res <- r:
|
|||
|
|
log.Println("资源释放到池子里了")
|
|||
|
|
default:
|
|||
|
|
log.Println("资源池满了，释放这个资源吧")
|
|||
|
|
r.Close()
|
|||
|
|
}
|
|||
|
|
}
|