From 7032f8ac7ca14e582c393a6da4fc81b631bad8aa Mon Sep 17 00:00:00 2001 From: groot Date: Fri, 21 Jun 2019 20:10:43 +0800 Subject: [PATCH] fix a bug about quert by file interface Former-commit-id: 801726772d7555b84f36a828bded58c45692fe14 --- cpp/src/db/DBImpl.cpp | 4 ++++ cpp/src/db/DBMetaImpl.cpp | 1 + 2 files changed, 5 insertions(+) diff --git a/cpp/src/db/DBImpl.cpp b/cpp/src/db/DBImpl.cpp index 01ea9d29bb..e66471d5c3 100644 --- a/cpp/src/db/DBImpl.cpp +++ b/cpp/src/db/DBImpl.cpp @@ -241,6 +241,10 @@ Status DBImpl::Query(const std::string& table_id, const std::vector return status; } + if(files_array.empty()) { + return Status::Error("Invalid file id"); + } + return QueryAsync(table_id, files_array, k, nq, vectors, dates, results); } diff --git a/cpp/src/db/DBMetaImpl.cpp b/cpp/src/db/DBMetaImpl.cpp index c6c80ab9dd..9174663395 100644 --- a/cpp/src/db/DBMetaImpl.cpp +++ b/cpp/src/db/DBMetaImpl.cpp @@ -627,6 +627,7 @@ Status DBMetaImpl::GetTableFiles(const std::string& table_id, for (auto &file : files) { TableFileSchema file_schema; + file_schema.table_id_ = table_id; file_schema.file_id_ = std::get<0>(file); file_schema.file_type_ = std::get<1>(file); file_schema.size_ = std::get<2>(file);