Fix load timeout in chaos (#17241)

Signed-off-by: xige-16 <xi.ge@zilliz.com>
2022-05-26 22:28:03 +08:00 · 2022-05-26 22:28:03 +08:00 · 3a63d6c98e
parent d76a22cf98
commit 3a63d6c98e
1 changed files with 6 additions and 1 deletions
--- a/internal/querycoord/channel_allocator.go
+++ b/internal/querycoord/channel_allocator.go
@ -48,7 +48,12 @@ func shuffleChannelsToQueryNode(ctx context.Context, reqs []*querypb.WatchDmChan
 			if err != nil {
 				return err
 			}
-			onlineNodeIDs = replica.GetNodeIds()
+			replicaNodes := replica.GetNodeIds()
+			for _, nodeID := range replicaNodes {
+				if ok, err := cluster.isOnline(nodeID); err == nil && ok {
+					onlineNodeIDs = append(onlineNodeIDs, nodeID)
+				}
+			}
 		}
 		if len(onlineNodeIDs) == 0 {
 			err := errors.New("no online QueryNode to allocate")