milvus/internal/datanode/io_pool.go

25 lines
475 B
Go
Raw Normal View History

package datanode
import (
"sync"
"github.com/milvus-io/milvus/pkg/util/conc"
)
var ioPool *conc.Pool[any]
var ioPoolInitOnce 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
}