fix: Remove sync task after finished (milvus-io#38681)

Related to milvus-io#38680 Signed-off-by: Congqi Xia <[email protected]>
bigsheeper · Dec 24, 2024 · 636e107 · 636e107
1 parent 877cd80
commit 636e107
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/flushcommon/syncmgr/sync_manager.go b/internal/flushcommon/syncmgr/sync_manager.go
@@ -124,6 +124,10 @@ func (mgr *syncManager) safeSubmitTask(ctx context.Context, task Task, callbacks
 
 func (mgr *syncManager) submit(ctx context.Context, 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
 		}