From 887d9d7af6dcc7b5f5ffebe579013ccd1d052645 Mon Sep 17 00:00:00 2001
From: FluorineDog <guilin.gou@zilliz.com>
Date: Tue, 22 Sep 2020 02:09:49 +0800
Subject: [PATCH] Refactor default index_meta

Signed-off-by: FluorineDog <guilin.gou@zilliz.com>
---
 core/src/dog_segment/SegmentNaive.cpp | 1 +
 core/unittest/test_c_api.cpp          | 8 ++++----
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/core/src/dog_segment/SegmentNaive.cpp b/core/src/dog_segment/SegmentNaive.cpp
index 0500793f0..64ef13290 100644
--- a/core/src/dog_segment/SegmentNaive.cpp
+++ b/core/src/dog_segment/SegmentNaive.cpp
@@ -587,6 +587,7 @@ SegmentNaive::BuildIndex(IndexMetaPtr remote_index_meta) {
     if(record_.ack_responder_.GetAck() < 1024 * 4) {
         return Status(SERVER_BUILD_INDEX_ERROR, "too few elements");
     }
+    index_meta_ = remote_index_meta;
     for (auto&[index_name, entry]: index_meta_->get_entries()) {
         assert(entry.index_name == index_name);
         const auto &field = (*schema_)[entry.field_name];
diff --git a/core/unittest/test_c_api.cpp b/core/unittest/test_c_api.cpp
index dcb2beb4e..d2618ac2f 100644
--- a/core/unittest/test_c_api.cpp
+++ b/core/unittest/test_c_api.cpp
@@ -188,7 +188,7 @@ TEST(CApiTest, BuildIndexTest) {
     std::vector<uint64_t> timestamps;
     std::vector<int64_t> uids;
 
-    int N = 3;
+    int N = 10000;
     int DIM = 16;
 
     std::vector<float> vec(DIM);
@@ -202,8 +202,8 @@ TEST(CApiTest, BuildIndexTest) {
         // append vec
 
         raw_data.insert(raw_data.end(),
-                        (const char*)&vec,
-                        ((const char*)&vec) + sizeof(float) * vec.size());
+                        (const char*)&vec[0],
+                        ((const char*)&vec[0]) + sizeof(float) * vec.size());
         int age = i;
         raw_data.insert(
             raw_data.end(), (const char*)&age, ((const char*)&age) + sizeof(age));
@@ -238,7 +238,7 @@ TEST(CApiTest, BuildIndexTest) {
     CQueryInfo queryInfo{1, 10, "fakevec"};
 
     auto sea_res = Search(
-        segment, queryInfo, 1, query_raw_data.data(), DIM, result_ids, result_distances);
+        segment, queryInfo, 20, query_raw_data.data(), DIM, result_ids, result_distances);
     assert(sea_res == 0);
 
     DeleteCollection(collection);
-- 
GitLab