Fix delegator leaked (#25347)

Signed-off-by: N yah01 <yang.cen@zilliz.com>

Fix delegator leaked (#25347)
Signed-off-by: N yah01 <yang.cen@zilliz.com>
3d228341 · yah01 · GitHub · dd5f896d · 3d228341
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

internal/querynodev2/services.go internal/querynodev2/services.go +1 -2

未找到文件。
--- a/internal/querynodev2/services.go
+++ b/internal/querynodev2/services.go
@@ -358,7 +358,7 @@ func (node *QueryNode) UnsubDmChannel(ctx context.Context, req *querypb.UnsubDmC
 		return status, nil
 	}

-	delegator, ok := node.delegators.Get(req.GetChannelName())
+	delegator, ok := node.delegators.GetAndRemove(req.GetChannelName())
 	if ok {
 		// close the delegator first to block all coming query/search requests
 		delegator.Close()
@@ -366,7 +366,6 @@ func (node *QueryNode) UnsubDmChannel(ctx context.Context, req *querypb.UnsubDmC
 		node.pipelineManager.Remove(req.GetChannelName())
 		node.manager.Segment.RemoveBy(segments.WithChannel(req.GetChannelName()))
 		node.tSafeManager.Remove(req.GetChannelName())
-		node.delegators.GetAndRemove(req.GetChannelName())
 	}

 	log.Info("unsubscribed channel")