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 pool memory allocation if too large

Browse source
This commit is contained in:
chrislu 2022-03-02 13:50:28 -08:00
parent ba14307319
commit 784583afc6
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -33,17 +33,18 @@ func init() {
} }
} }
func getSlotPool(size int) *sync.Pool { func getSlotPool(size int) (*sync.Pool, bool) {
index := bitCount(size) index := bitCount(size)
return pools[index] if index >= len(pools) {
return nil, false
}
return pools[index], true
} }
var total int64 var total int64
func Allocate(size int) []byte { func Allocate(size int) []byte {
pool := getSlotPool(size) if pool, found := getSlotPool(size); found {
if pool != nil {
newVal := atomic.AddInt64(&total, 1) newVal := atomic.AddInt64(&total, 1)
glog.V(4).Infof("++> %d", newVal) glog.V(4).Infof("++> %d", newVal)
@ -54,8 +55,7 @@ func Allocate(size int) []byte {
} }
func Free(buf []byte) { func Free(buf []byte) {
pool := getSlotPool(cap(buf)) if pool, found := getSlotPool(cap(buf)); found {
if pool != nil {
newVal := atomic.AddInt64(&total, -1) newVal := atomic.AddInt64(&total, -1)
glog.V(4).Infof("--> %d", newVal) glog.V(4).Infof("--> %d", newVal)
pool.Put(&buf) pool.Put(&buf)