mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-06 05:49:24 +08:00
26f06dd732
Signed-off-by: SimFG <bang.fu@zilliz.com>
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package datanode
|
|
|
|
import (
|
|
"runtime"
|
|
"sync"
|
|
|
|
"github.com/milvus-io/milvus/pkg/util/conc"
|
|
)
|
|
|
|
var (
|
|
ioPool *conc.Pool[any]
|
|
ioPoolInitOnce sync.Once
|
|
)
|
|
|
|
var (
|
|
statsPool *conc.Pool[any]
|
|
statsPoolInitOnce sync.Once
|
|
)
|
|
|
|
func initIOPool() {
|
|
capacity := Params.DataNodeCfg.IOConcurrency.GetAsInt()
|
|
if capacity > 32 {
|
|
capacity = 32
|
|
}
|
|
// error only happens with negative expiry duration or with negative pre-alloc size.
|
|
ioPool = conc.NewPool[any](capacity)
|
|
}
|
|
|
|
func getOrCreateIOPool() *conc.Pool[any] {
|
|
ioPoolInitOnce.Do(initIOPool)
|
|
return ioPool
|
|
}
|
|
|
|
func initStatsPool() {
|
|
statsPool = conc.NewPool[any](runtime.GOMAXPROCS(0), conc.WithPreAlloc(false), conc.WithNonBlocking(false))
|
|
}
|
|
|
|
func getOrCreateStatsPool() *conc.Pool[any] {
|
|
statsPoolInitOnce.Do(initStatsPool)
|
|
return statsPool
|
|
}
|
|
|
|
func initMultiReadPool() {
|
|
capacity := Params.DataNodeCfg.FileReadConcurrency.GetAsInt()
|
|
if capacity > runtime.GOMAXPROCS(0) {
|
|
capacity = runtime.GOMAXPROCS(0)
|
|
}
|
|
// error only happens with negative expiry duration or with negative pre-alloc size.
|
|
ioPool = conc.NewPool[any](capacity)
|
|
}
|
|
|
|
func getMultiReadPool() *conc.Pool[any] {
|
|
ioPoolInitOnce.Do(initMultiReadPool)
|
|
return ioPool
|
|
}
|