update JetBrains License Server

common/pool/pool.go

@@ -0,0 +1,77 @@
+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()
+	}
+}

common/pool/pool_test.go

@@ -0,0 +1,72 @@
+package pool
+
+import (
+	"io"
+	"log"
+	"math/rand"
+	"sync"
+	"sync/atomic"
+	"testing"
+	"time"
+)
+
+const (
+	//模拟的最大goroutine
+	maxGoroutine = 5
+	//资源池的大小
+	poolRes = 2
+)
+
+func TestPool(t *testing.T) {
+	//等待任务完成
+	var wg sync.WaitGroup
+	wg.Add(maxGoroutine)
+	p, err := NewPool(createConnection, poolRes)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	//模拟好几个goroutine同时使用资源池查询数据
+	for query := 0; query < maxGoroutine; query++ {
+		go func(q int) {
+			dbQuery(q, p)
+			wg.Done()
+		}(query)
+	}
+	wg.Wait()
+	log.Println("开始关闭资源池")
+	p.Close()
+}
+
+//模拟数据库查询
+func dbQuery(query int, pool *Pool) {
+	conn, err := pool.Acquire()
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer pool.Release(conn)
+	//模拟查询
+	time.Sleep(time.Duration(rand.Intn(1000)) * time.Millisecond)
+	log.Printf("第%d个查询，使用的是ID为%d的数据库连接", query, conn.(*dbConnection).ID)
+}
+
+//数据库连接
+type dbConnection struct {
+	ID int32 //连接的标志
+}
+
+//实现io.Closer接口
+func (db *dbConnection) Close() error {
+	log.Println("关闭连接", db.ID)
+	return nil
+}
+
+var idCounter int32
+
+//生成数据库连接的方法，以供资源池使用
+func createConnection() (io.Closer, error) {
+	//并发安全，给数据库连接生成唯一标志
+	id := atomic.AddInt32(&idCounter, 1)
+	return &dbConnection{id}, nil
+}