Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-15 16:34:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化代码规范

Browse Source
This commit is contained in:
duanhf2012
2024-09-20 17:25:08 +08:00
parent 1cf071a444
commit 39116c4402
63 changed files with 1971 additions and 2043 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
util/sync/MemPool.go
View File

@@ -5,34 +5,32 @@ import (
)
type Pool struct {
C chan interface{} //最大缓存的数量
C chan interface{} //最大缓存的数量
syncPool sysSync.Pool
}
type IPoolData interface {
Reset()
IsRef()bool
IsRef() bool
Ref()
UnRef()
}
type PoolEx struct{
C chan IPoolData //最大缓存的数量
type PoolEx struct {
C chan IPoolData //最大缓存的数量
syncPool sysSync.Pool
}
func (pool *Pool) Get() interface{}{
func (pool *Pool) Get() interface{} {
select {
case d := <-pool.C:
return d
default:
return pool.syncPool.Get()
}
return nil
}
func (pool *Pool) Put(data interface{}){
func (pool *Pool) Put(data interface{}) {
select {
case pool.C <- data:
default:
@@ -41,14 +39,14 @@ func (pool *Pool) Put(data interface{}){
}
func NewPool(C chan interface{},New func()interface{}) *Pool{
func NewPool(C chan interface{}, New func() interface{}) *Pool {
var p Pool
p.C = C
p.syncPool.New = New
return &p
}
func NewPoolEx(C chan IPoolData,New func()IPoolData) *PoolEx{
func NewPoolEx(C chan IPoolData, New func() IPoolData) *PoolEx {
var pool PoolEx
pool.C = C
pool.syncPool.New = func() interface{} {
@@ -57,7 +55,7 @@ func NewPoolEx(C chan IPoolData,New func()IPoolData) *PoolEx{
return &pool
}
func (pool *PoolEx) Get() IPoolData{
func (pool *PoolEx) Get() IPoolData {
select {
case d := <-pool.C:
if d.IsRef() {
@@ -79,7 +77,7 @@ func (pool *PoolEx) Get() IPoolData{
return nil
}
func (pool *PoolEx) Put(data IPoolData){
func (pool *PoolEx) Put(data IPoolData) {
if data.IsRef() == false {
panic("Repeatedly freeing memory")
}
@@ -94,5 +92,3 @@ func (pool *PoolEx) Put(data IPoolData){
pool.syncPool.Put(data)
}
}