fix review comment

Signed-off-by: Wei Liu <[email protected]>
milvus-io · Jun 26, 2024 · 6388d23 · 6388d23
1 parent 49315d5
commit 6388d23
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/querycoordv2/task/scheduler.go b/internal/querycoordv2/task/scheduler.go
@@ -305,8 +305,14 @@ func (scheduler *taskScheduler) updateTaskDelta(task Task) {
 	var deltaMap map[int64]map[int64]int
 	switch task := task.(type) {
 	case *SegmentTask:
-		segment := scheduler.targetMgr.GetSealedSegment(task.CollectionID(), task.SegmentID(), meta.NextTargetFirst)
-		delta = int(segment.GetNumOfRows())
+		// skip growing segment's count, cause doesn't know realtime row number of growing segment
+		if task.Actions()[0].(*SegmentAction).Scope() == querypb.DataScope_Historical {
+			segment := scheduler.targetMgr.GetSealedSegment(task.CollectionID(), task.SegmentID(), meta.NextTargetFirst)
+			if segment != nil {
+				delta = int(segment.GetNumOfRows())
+			}
+		}
+
 		deltaMap = scheduler.segmentExecutingTaskDelta
 
 	case *ChannelTask: