From 1da459596ddc59d9fd9815a5775f20f4d85ef09f Mon Sep 17 00:00:00 2001
From: groot <yihua.mo@zilliz.com>
Date: Sun, 21 Jul 2019 18:34:59 +0800
Subject: [PATCH] invalid topk

Former-commit-id: a353f66e8f0503030713d9834a0bbaf726ca0001
---
 cpp/src/server/RequestTask.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cpp/src/server/RequestTask.cpp b/cpp/src/server/RequestTask.cpp
index e553c036..d6c1a2fb 100644
--- a/cpp/src/server/RequestTask.cpp
+++ b/cpp/src/server/RequestTask.cpp
@@ -513,7 +513,7 @@ ServerError SearchVectorTaskBase::OnExecute() {
             return SetError(res, "Invalid table name: " + table_name_);
         }
 
-        if(top_k_ <= 0) {
+        if(top_k_ <= 0 || top_k_ > 1024) {
             return SetError(SERVER_INVALID_TOPK, "Invalid topk: " + std::to_string(top_k_));
         }
         if(record_array_.empty()) {
-- 
GitLab