From ec644995361aa96b4263bdb1f1ef505c8a70f8f1 Mon Sep 17 00:00:00 2001
From: congqixia <congqi.xia@zilliz.com>
Date: Tue, 18 Jun 2024 14:11:58 +0800
Subject: [PATCH] fix: Check nodeID wildcard when removing pkOracle (#33895)

See also #33894

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
---
 internal/querynodev2/pkoracle/candidate.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/internal/querynodev2/pkoracle/candidate.go b/internal/querynodev2/pkoracle/candidate.go
index bb2479702b..9f8a8b7daf 100644
--- a/internal/querynodev2/pkoracle/candidate.go
+++ b/internal/querynodev2/pkoracle/candidate.go
@@ -52,7 +52,8 @@ func WithSegmentType(typ commonpb.SegmentState) CandidateFilter {
 // WithWorkerID returns CandidateFilter with provided worker id.
 func WithWorkerID(workerID int64) CandidateFilter {
 	return func(candidate candidateWithWorker) bool {
-		return candidate.workerID == workerID
+		return candidate.workerID == workerID ||
+			workerID == -1 // wildcard for offline node
 	}
 }