mirror of https://github.com/milvus-io/milvus.git
28 lines
645 B
Go
28 lines
645 B
Go
package querynode
|
|
|
|
import (
|
|
"container/list"
|
|
)
|
|
|
|
type scheduleReadTaskPolicy func(sqTasks *list.List, targetUsage int32, maxNum int32) ([]readTask, int32)
|
|
|
|
func defaultScheduleReadPolicy(sqTasks *list.List, targetUsage int32, maxNum int32) ([]readTask, int32) {
|
|
var ret []readTask
|
|
usage := int32(0)
|
|
var next *list.Element
|
|
for e := sqTasks.Front(); e != nil && maxNum > 0; e = next {
|
|
next = e.Next()
|
|
t, _ := e.Value.(readTask)
|
|
tUsage := t.CPUUsage()
|
|
if usage+tUsage > targetUsage {
|
|
break
|
|
}
|
|
usage += tUsage
|
|
sqTasks.Remove(e)
|
|
rateCol.rtCounter.sub(t, readyQueueType)
|
|
ret = append(ret, t)
|
|
maxNum--
|
|
}
|
|
return ret, usage
|
|
}
|