fix: Pingpong load balancing issue when segment has only 1 row(#44840) (#44841)

lmzzzzz1 · web-flow · commit e18e7d3b32f9 · 2025-10-16T11:18:00.000+08:00
Use math.Ceil to calculate Priority uniformly issue: #44840 Signed-off-by: 1mmortal <lmzzzzz1@163.com>
diff --git a/internal/querycoordv2/balance/rowcount_based_balancer.go b/internal/querycoordv2/balance/rowcount_based_balancer.go
@@ -433,7 +433,7 @@ func (b *nodeItem) setPriority(priority int) {
 }
 
 func (b *nodeItem) getPriorityWithCurrentScoreDelta(delta float64) int {
-	return int((b.currentScore + delta) - b.assignedScore)
+	return int(math.Ceil((b.currentScore + delta) - b.assignedScore))
 }
 
 func (b *nodeItem) getCurrentScore() float64 {

Original file line number	Diff line number	Diff line change
`@@ -433,7 +433,7 @@ func (b *nodeItem) setPriority(priority int) {`
`433`	`433`	`}`
`434`	`434`
`435`	`435`	`func (b *nodeItem) getPriorityWithCurrentScoreDelta(delta float64) int {`
`436`		`- return int((b.currentScore + delta) - b.assignedScore)`
	`436`	`+ return int(math.Ceil((b.currentScore + delta) - b.assignedScore))`
`437`	`437`	`}`
`438`	`438`
`439`	`439`	`func (b *nodeItem) getCurrentScore() float64 {`