From ea8e1623cdc4c0231bd9c13d332a00ac38280810 Mon Sep 17 00:00:00 2001
From: "zhenshan.cao" <zhenshan.cao@zilliz.com>
Date: Tue, 24 May 2022 10:05:58 +0800
Subject: [PATCH] Acquire nq from placeholdergroup incase of nq is zero
 (#17168)

Signed-off-by: zhenshan.cao <zhenshan.cao@zilliz.com>
---
 internal/proxy/task_search.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/internal/proxy/task_search.go b/internal/proxy/task_search.go
index 9c1892009c..c483549203 100644
--- a/internal/proxy/task_search.go
+++ b/internal/proxy/task_search.go
@@ -251,6 +251,16 @@ func (t *searchTask) PreExecute(ctx context.Context) error {
 	t.DbID = 0 // todo
 	t.SearchRequest.Dsl = t.request.Dsl
 	t.SearchRequest.PlaceholderGroup = t.request.PlaceholderGroup
+	if t.request.GetNq() == 0 {
+		x := &commonpb.PlaceholderGroup{}
+		err := proto.Unmarshal(t.request.GetPlaceholderGroup(), x)
+		if err != nil {
+			return err
+		}
+		for _, h := range x.GetPlaceholders() {
+			t.request.Nq += int64(len(h.Values))
+		}
+	}
 	t.SearchRequest.Nq = t.request.GetNq()
 	log.Info("search PreExecute done.",
 		zap.Any("requestID", t.Base.MsgID), zap.Any("requestType", "search"))