From 773c64ecbbd3d4e02d7d3183e8deec34e4774db4 Mon Sep 17 00:00:00 2001
From: congqixia <congqi.xia@zilliz.com>
Date: Thu, 14 Mar 2024 15:09:03 +0800
Subject: [PATCH] fix: Set nodeID when remove distribution (#31259)

See also #30930

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
---
 internal/querycoordv2/task/executor.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/internal/querycoordv2/task/executor.go b/internal/querycoordv2/task/executor.go
index 799875a5b7..ec69aa9ac6 100644
--- a/internal/querycoordv2/task/executor.go
+++ b/internal/querycoordv2/task/executor.go
@@ -526,6 +526,7 @@ func (ex *Executor) removeDistribution(task *LeaderTask, step int) error {
 			{
 				Type:      querypb.SyncType_Remove,
 				SegmentID: action.SegmentID(),
+				NodeID:    action.Node(),
 			},
 		},
 	}