From 9817e8313d1c25b2c03b4ab7c67f606b08a10cef Mon Sep 17 00:00:00 2001
From: Congqi Xia <congqi.xia@zilliz.com>
Date: Mon, 1 Jul 2024 15:22:46 +0800
Subject: [PATCH] enhance: Use channel name map finding channel watcher

See also #33235

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
---
 internal/datacoord/channel_manager_v2.go | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/internal/datacoord/channel_manager_v2.go b/internal/datacoord/channel_manager_v2.go
index ed70038741a6..f9c11beff393 100644
--- a/internal/datacoord/channel_manager_v2.go
+++ b/internal/datacoord/channel_manager_v2.go
@@ -384,20 +384,19 @@ func (m *ChannelManagerImplV2) FindWatcher(channel string) (UniqueID, error) {
 
 	infos := m.store.GetNodesChannels()
 	for _, info := range infos {
-		for _, channelInfo := range info.Channels {
-			if channelInfo.GetName() == channel {
-				return info.NodeID, nil
-			}
+		_, ok := info.Channels[channel]
+		if ok {
+			return info.NodeID, nil
 		}
 	}
 
 	// channel in buffer
 	bufferInfo := m.store.GetBufferChannelInfo()
-	for _, channelInfo := range bufferInfo.Channels {
-		if channelInfo.GetName() == channel {
-			return bufferID, errChannelInBuffer
-		}
+	_, ok := bufferInfo.Channels[channel]
+	if ok {
+		return bufferID, errChannelInBuffer
 	}
+
 	return 0, errChannelNotWatched
 }