fix to replicate entity lost when pause the task

Signed-off-by: SimFG <bang.fu@zilliz.com>
zilliztech · Nov 28, 2024 · 7a3ca02 · 7a3ca02
1 parent 4bc59b9
commit 7a3ca02
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/server/cdc_impl.go b/server/cdc_impl.go
@@ -1228,9 +1228,13 @@ func (e *MetaCDC) pauseTaskWithReason(taskID, reason string, currentStates []met
 	if replicateEntity, ok := e.replicateEntityMap.data[uKey]; ok {
 		if quitFunc, ok := replicateEntity.taskQuitFuncs.GetAndRemove(taskID); ok {
 			quitFunc()
+			replicateEntity.refCnt.Dec()
+		}
+		if replicateEntity.refCnt.Load() == 0 {
+			replicateEntity.entityQuitFunc()
+			delete(e.replicateEntityMap.data, uKey)
 		}
 	}
-	delete(e.replicateEntityMap.data, uKey)
 	e.replicateEntityMap.Unlock()
 	return err
 }
@@ -1284,9 +1288,9 @@ func (e *MetaCDC) delete(taskID string) error {
 		}
 		if replicateEntity.refCnt.Load() == 0 {
 			replicateEntity.entityQuitFunc()
+			delete(e.replicateEntityMap.data, uKey)
 		}
 	}
-	delete(e.replicateEntityMap.data, uKey)
 	e.replicateEntityMap.Unlock()
 
 	return err