fix: Use correct release func when datanode unwatch a channel (milvus…

…-io#35655) See also milvus-io#35654 Signed-off-by: Congqi Xia <[email protected]>
bigsheeper · Aug 23, 2024 · 59387f0 · 59387f0
1 parent e8e3544
commit 59387f0
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/internal/datanode/channel/channel_manager.go b/internal/datanode/channel/channel_manager.go
@@ -71,7 +71,15 @@ func NewChannelManager(pipelineParams *util.PipelineParams, fgManager pipeline.F
 		opRunners:     typeutil.NewConcurrentMap[string, *opRunner](),
 		abnormals:     typeutil.NewConcurrentMap[int64, string](),
 
-		releaseFunc: fgManager.RemoveFlowgraph,
+		releaseFunc: func(channelName string) {
+			if pipelineParams.CompactionExecutor != nil {
+				pipelineParams.CompactionExecutor.DiscardPlan(channelName)
+			}
+			if pipelineParams.WriteBufferManager != nil {
+				pipelineParams.WriteBufferManager.RemoveChannel(channelName)
+			}
+			fgManager.RemoveFlowgraph(channelName)
+		},
 
 		closeCh: lifetime.NewSafeChan(),
 	}