mirror of https://github.com/gobwas/glob.git
use sync.pool
This commit is contained in:
parent
0964106bb3
commit
ee3b8204de
|
@ -9,9 +9,7 @@ type SomePool interface {
|
||||||
Put([]int)
|
Put([]int)
|
||||||
}
|
}
|
||||||
|
|
||||||
var segmentsPools [1024]SomePool
|
var segmentsPools [1024]sync.Pool
|
||||||
|
|
||||||
//var segmentsPools [1024]*sync.Pool
|
|
||||||
|
|
||||||
func toPowerOfTwo(v int) int {
|
func toPowerOfTwo(v int) int {
|
||||||
v--
|
v--
|
||||||
|
@ -56,15 +54,9 @@ const (
|
||||||
func init() {
|
func init() {
|
||||||
for i := cacheToAndHigher; i >= cacheFrom; i >>= 1 {
|
for i := cacheToAndHigher; i >= cacheFrom; i >>= 1 {
|
||||||
func(i int) {
|
func(i int) {
|
||||||
// segmentsPools[i-1] = &sync.Pool{New: func() interface{} {
|
segmentsPools[i-1] = sync.Pool{New: func() interface{} {
|
||||||
// return make([]int, 0, i)
|
|
||||||
// }}
|
|
||||||
// segmentsPools[i-1] = newChanPool(func() []int {
|
|
||||||
// return make([]int, 0, i)
|
|
||||||
// })
|
|
||||||
segmentsPools[i-1] = newSyncPool(func() []int {
|
|
||||||
return make([]int, 0, i)
|
return make([]int, 0, i)
|
||||||
})
|
}}
|
||||||
}(i)
|
}(i)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -88,8 +80,7 @@ func acquireSegments(c int) []int {
|
||||||
return make([]int, 0, c)
|
return make([]int, 0, c)
|
||||||
}
|
}
|
||||||
|
|
||||||
// return segmentsPools[getTableIndex(c)].Get().([]int)[:0]
|
return segmentsPools[getTableIndex(c)].Get().([]int)[:0]
|
||||||
return segmentsPools[getTableIndex(c)].Get()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func releaseSegments(s []int) {
|
func releaseSegments(s []int) {
|
||||||
|
@ -103,59 +94,3 @@ func releaseSegments(s []int) {
|
||||||
|
|
||||||
segmentsPools[getTableIndex(c)].Put(s)
|
segmentsPools[getTableIndex(c)].Put(s)
|
||||||
}
|
}
|
||||||
|
|
||||||
type maker func() []int
|
|
||||||
|
|
||||||
type syncPool struct {
|
|
||||||
new maker
|
|
||||||
pool sync.Pool
|
|
||||||
}
|
|
||||||
|
|
||||||
func newSyncPool(m maker) *syncPool {
|
|
||||||
return &syncPool{
|
|
||||||
new: m,
|
|
||||||
pool: sync.Pool{New: func() interface{} {
|
|
||||||
return m()
|
|
||||||
}},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *syncPool) Get() []int {
|
|
||||||
return s.pool.Get().([]int)[:0]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *syncPool) Put(x []int) {
|
|
||||||
s.pool.Put(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
type chanPool struct {
|
|
||||||
pool chan []int
|
|
||||||
new maker
|
|
||||||
index int
|
|
||||||
}
|
|
||||||
|
|
||||||
func newChanPool(m maker) *chanPool {
|
|
||||||
return &chanPool{
|
|
||||||
pool: make(chan []int, 16),
|
|
||||||
new: m,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *chanPool) Get() []int {
|
|
||||||
select {
|
|
||||||
case s := <-c.pool:
|
|
||||||
return s[:0]
|
|
||||||
default:
|
|
||||||
// pool is empty
|
|
||||||
return c.new()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *chanPool) Put(s []int) {
|
|
||||||
select {
|
|
||||||
case c.pool <- s:
|
|
||||||
// ok
|
|
||||||
default:
|
|
||||||
// pool is full
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
Loading…
Reference in New Issue