[fix][client] Fix SendBatchVectorAdd with no file input.

Signed-off-by: Ketor <[email protected]>
yuhaijun999 · Mar 7, 2024 · 9099b2e · 9099b2e
1 parent 925ee76
commit 9099b2e
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/client/store_client_function.cc b/src/client/store_client_function.cc
@@ -1683,14 +1683,20 @@ int SendBatchVectorAdd(int64_t region_id, uint32_t dimension, std::vector<int64_
   dingodb::pb::index::VectorAddResponse response;
 
   uint32_t max_size = 0;
-  if (vector_datas.size() > vector_datas_offset + vector_ids.size()) {
+  if (vector_datas.empty()) {
     max_size = vector_ids.size();
   } else {
-    max_size = vector_datas.size() > vector_datas_offset ? vector_datas.size() - vector_datas_offset : 0;
+    if (vector_datas.size() > vector_datas_offset + vector_ids.size()) {
+      max_size = vector_ids.size();
+    } else {
+      max_size = vector_datas.size() > vector_datas_offset ? vector_datas.size() - vector_datas_offset : 0;
+    }
   }
 
   if (max_size == 0) {
-    DINGO_LOG(INFO) << "vector_datas.size() - vector_datas_offset <= vector_ids.size(), max_size: " << max_size;
+    DINGO_LOG(INFO) << "vector_datas.size() - vector_datas_offset <= vector_ids.size(), max_size: " << max_size
+                    << ", vector_datas.size: " << vector_datas.size()
+                    << ", vector_datas_offset: " << vector_datas_offset << ", vector_ids.size: " << vector_ids.size();
     return 0;
   } else {
     DINGO_LOG(DEBUG) << "vector_datas.size(): " << vector_datas.size()