From e18e7d3b32f9e20222e8443b7634455d4bb2126f Mon Sep 17 00:00:00 2001
From: 1mmortal <lmzzzzz1@163.com>
Date: Thu, 16 Oct 2025 11:18:00 +0800
Subject: [PATCH] fix: Pingpong load balancing issue when segment has only 1
 row(#44840) (#44841)

Use math.Ceil to calculate Priority uniformly
issue: https://github.com/milvus-io/milvus/issues/44840

Signed-off-by: 1mmortal <lmzzzzz1@163.com>
---
 internal/querycoordv2/balance/rowcount_based_balancer.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internal/querycoordv2/balance/rowcount_based_balancer.go b/internal/querycoordv2/balance/rowcount_based_balancer.go
index 01cd9647fb..bd4b686292 100644
--- 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 {