|
|
|
@ -2,6 +2,7 @@ package buf
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"os"
|
|
|
|
|
"runtime"
|
|
|
|
|
"strconv"
|
|
|
|
|
"sync"
|
|
|
|
|
)
|
|
|
|
@ -107,8 +108,17 @@ var (
|
|
|
|
|
smallPool = NewSyncPool(SizeSmall)
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func getDefaultPoolSize() uint32 {
|
|
|
|
|
switch runtime.GOARCH {
|
|
|
|
|
case "amd64", "386":
|
|
|
|
|
return 20
|
|
|
|
|
default:
|
|
|
|
|
return 5
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
|
var size uint32 = 20
|
|
|
|
|
var size uint32 = getDefaultPoolSize()
|
|
|
|
|
sizeStr := os.Getenv(poolSizeEnvKey)
|
|
|
|
|
if len(sizeStr) > 0 {
|
|
|
|
|
customSize, err := strconv.ParseUint(sizeStr, 10, 32)
|
|
|
|
|