gitlab-com-seaweedfs/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2024-01-19 02:48:24 +00:00
Code Issues Projects Releases Wiki Activity

avoid possible too big memory allocation

Browse source
This commit is contained in:
chrislu 2022-02-26 03:22:41 -08:00
parent 3345a50d9b
commit 708e14fcfa
1 changed files with 16 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
weed/util/mem/slot_pool.go
View file

@ -41,14 +41,23 @@ func getSlotPool(size int) *sync.Pool {
var total int64
func Allocate(size int) []byte {
newVal := atomic.AddInt64(&total, 1)
glog.V(4).Infof("++> %d", newVal)
slab := *getSlotPool(size).Get().(*[]byte)
return slab[:size]
pool := getSlotPool(size)
if pool != nil {
newVal := atomic.AddInt64(&total, 1)
glog.V(4).Infof("++> %d", newVal)
slab := *pool.Get().(*[]byte)
return slab[:size]
}
return make([]byte, size)
}
func Free(buf []byte) {
newVal := atomic.AddInt64(&total, -1)
glog.V(4).Infof("--> %d", newVal)
getSlotPool(cap(buf)).Put(&buf)
pool := getSlotPool(cap(buf))
if pool != nil {
newVal := atomic.AddInt64(&total, -1)
glog.V(4).Infof("--> %d", newVal)
pool.Put(&buf)
}
}