You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

59 lines
1.4 KiB

// Copyright 2020 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.
//go:build !race
// +build !race
package localpool
// Get gets an object from the pool.
func (p *LocalPool) Get() interface{} {
pid := procPin()
slot := p.slots[pid]
objLen := len(slot.objs)
var result interface{}
if objLen > 0 {
lastIdx := objLen - 1
result = slot.objs[lastIdx]
slot.objs[lastIdx] = nil
slot.objs = slot.objs[:lastIdx]
slot.getHit++
} else {
slot.getMiss++
}
procUnpin()
if result == nil {
result = p.newFn()
}
return result
}
// Put puts an object back to the pool.
// It returns true if the pool is not full and the obj is successfully put into the pool.
func (p *LocalPool) Put(obj interface{}) bool {
if p.resetFn != nil {
p.resetFn(obj)
}
var ok bool
pid := procPin()
slot := p.slots[pid]
if len(slot.objs) < p.sizePerProc {
slot.objs = append(slot.objs, obj)
slot.putHit++
ok = true
} else {
slot.putMiss++
}
procUnpin()
return ok
}