diff --git a/internal/querycoordv2/task/scheduler.go b/internal/querycoordv2/task/scheduler.go index f8e2dab0b2..a35be5ad3e 100644 --- a/internal/querycoordv2/task/scheduler.go +++ b/internal/querycoordv2/task/scheduler.go @@ -329,14 +329,7 @@ func (scheduler *taskScheduler) preAdd(task Task) error { taskType := GetTaskType(task) - if taskType == TaskTypeGrow { - leaderSegmentDist := scheduler.distMgr.LeaderViewManager.GetSealedSegmentDist(task.SegmentID()) - nodeSegmentDist := scheduler.distMgr.SegmentDistManager.GetSegmentDist(task.SegmentID()) - if lo.Contains(leaderSegmentDist, task.Actions()[0].Node()) && - lo.Contains(nodeSegmentDist, task.Actions()[0].Node()) { - return merr.WrapErrServiceInternal("segment loaded, it can be only balanced") - } - } else if taskType == TaskTypeMove { + if taskType == TaskTypeMove { leaderSegmentDist := scheduler.distMgr.LeaderViewManager.GetSealedSegmentDist(task.SegmentID()) nodeSegmentDist := scheduler.distMgr.SegmentDistManager.GetSegmentDist(task.SegmentID()) if !lo.Contains(leaderSegmentDist, task.Actions()[1].Node()) ||