mirror of https://github.com/gobwas/glob.git
98 lines
1.6 KiB
Go
98 lines
1.6 KiB
Go
package match
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
var segmentsPools [1024]*sync.Pool
|
|
|
|
func toPowerOfTwo(v int) int {
|
|
v--
|
|
v |= v >> 1
|
|
v |= v >> 2
|
|
v |= v >> 4
|
|
v |= v >> 8
|
|
v |= v >> 16
|
|
v++
|
|
|
|
return v
|
|
}
|
|
|
|
const (
|
|
cacheFrom = 16
|
|
cacheToAndHigher = 1024
|
|
cacheFromIndex = 15
|
|
cacheToAndHigherIndex = 1023
|
|
)
|
|
|
|
var asciiTable [128]byte
|
|
|
|
var segmentsByRuneLength [5][]int
|
|
|
|
var (
|
|
segments0 = []int{0}
|
|
segments1 = []int{1}
|
|
segments2 = []int{2}
|
|
segments3 = []int{3}
|
|
segments4 = []int{4}
|
|
)
|
|
|
|
const (
|
|
asciiLo = 0
|
|
asciiHi = 127
|
|
)
|
|
|
|
func init() {
|
|
for i := cacheToAndHigher; i >= cacheFrom; i >>= 1 {
|
|
func(i int) {
|
|
segmentsPools[i-1] = &sync.Pool{New: func() interface{} {
|
|
return make([]int, 0, i)
|
|
}}
|
|
}(i)
|
|
}
|
|
|
|
segmentsByRuneLength[0] = segments0
|
|
segmentsByRuneLength[1] = segments1
|
|
segmentsByRuneLength[2] = segments2
|
|
segmentsByRuneLength[3] = segments3
|
|
segmentsByRuneLength[4] = segments4
|
|
|
|
for i := 0; i <= 127; i++ {
|
|
asciiTable[i] = 1
|
|
}
|
|
}
|
|
|
|
func getTableIndex(c int) int {
|
|
p := toPowerOfTwo(c)
|
|
switch {
|
|
case p >= cacheToAndHigher:
|
|
return cacheToAndHigherIndex
|
|
case p <= cacheFrom:
|
|
return cacheFromIndex
|
|
default:
|
|
return p - 1
|
|
}
|
|
}
|
|
|
|
func acquireSegments(c int) []int {
|
|
// make []int with less capacity than cacheFrom
|
|
// is faster than acquiring it from pool
|
|
if c < cacheFrom {
|
|
return make([]int, 0, c)
|
|
}
|
|
|
|
return segmentsPools[getTableIndex(c)].Get().([]int)[:0]
|
|
}
|
|
|
|
func releaseSegments(s []int) {
|
|
c := cap(s)
|
|
|
|
// make []int with less capacity than cacheFrom
|
|
// is faster than acquiring it from pool
|
|
if c < cacheFrom {
|
|
return
|
|
}
|
|
|
|
segmentsPools[getTableIndex(cap(s))].Put(s)
|
|
}
|