From 762045adff4d11df4f6402efdc3a75658c975ac9 Mon Sep 17 00:00:00 2001 From: Ten Thousand Leaves <69466447+soothing-rain@users.noreply.github.com> Date: Wed, 19 Oct 2022 13:23:28 +0800 Subject: [PATCH] Fix an issue where working task might get cleaned up (#19888) /kind improvement Signed-off-by: Yuchen Gao Signed-off-by: Yuchen Gao --- internal/rootcoord/import_manager.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/internal/rootcoord/import_manager.go b/internal/rootcoord/import_manager.go index d4ee6106c7..8ee2b4f597 100644 --- a/internal/rootcoord/import_manager.go +++ b/internal/rootcoord/import_manager.go @@ -889,7 +889,7 @@ func (m *importManager) expireOldTasksFromMem() { defer m.workingLock.Unlock() for _, v := range m.workingTasks { taskExpiredAndStateUpdated := false - if taskExpired(v) { + if v.GetState().GetStateCode() != commonpb.ImportState_ImportCompleted && taskExpired(v) { log.Info("a working task has expired", zap.Int64("task ID", v.GetId())) taskID := v.GetId() m.workingLock.Unlock()