diff --git a/util/sync/MemPool.go b/util/sync/MemPool.go index ace1748..bee47b0 100644 --- a/util/sync/MemPool.go +++ b/util/sync/MemPool.go @@ -51,7 +51,6 @@ func NewPool(C chan interface{},New func()interface{}) *Pool{ func NewPoolEx(C chan IPoolData,New func()IPoolData) *PoolEx{ var pool PoolEx pool.C = C - //pool.New = New pool.syncPool.New = func() interface{} { return New() } @@ -61,10 +60,18 @@ func NewPoolEx(C chan IPoolData,New func()IPoolData) *PoolEx{ func (pool *PoolEx) Get() IPoolData{ select { case d := <-pool.C: + if d.IsRef() { + panic("Pool data is in use.") + } + d.Ref() return d default: data := pool.syncPool.Get().(IPoolData) + if data.IsRef() { + panic("Pool data is in use.") + } + data.Ref() return data } @@ -76,7 +83,10 @@ func (pool *PoolEx) Put(data IPoolData){ if data.IsRef() == false { panic("Repeatedly freeing memory") } + //提前解引用,防止递归释放 + data.UnRef() data.Reset() + //再次解引用,防止Rest时错误标记 data.UnRef() select { case pool.C <- data: