fix: [2.4] Remove sync task after finished (#38681) (#38686)

Cherry-pick from master pr: #38681 Related to #38680 Signed-off-by: Congqi Xia <[email protected]>
milvus-io · Dec 24, 2024 · 5fd69a0 · 5fd69a0
1 parent afaabc2
commit 5fd69a0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/datanode/syncmgr/sync_manager.go b/internal/datanode/syncmgr/sync_manager.go
@@ -127,6 +127,10 @@ func (mgr *syncManager) safeSubmitTask(task Task, callbacks ...func(error) error
 
 func (mgr *syncManager) submit(key int64, task Task, callbacks ...func(error) error) *conc.Future[struct{}] {
 	handler := func(err error) error {
+		taskKey := fmt.Sprintf("%d-%d", task.SegmentID(), task.Checkpoint().GetTimestamp())
+		defer func() {
+			mgr.tasks.Remove(taskKey)
+		}()
 		if err == nil {
 			return nil
 		}