milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix Server crashed during random test (#3436) 

Signed-off-by: yinghao.zou <yinghao.zou@zilliz.com>
pull/3450/head
BossZou 2020-08-24 23:34:35 +08:00 committed by GitHub
parent 17fedaf698
commit 84114944a0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
core/src/segment/SegmentReader.cpp
View File

@ -259,12 +259,14 @@ SegmentReader::LoadVectorIndex(const std::string& field_name, knowhere::VecIndex
return Status(DB_ERROR, "Field is not vector type");
}
// load uids
std::vector<int64_t> uids;
STATUS_CHECK(LoadUids(uids));
// load deleted doc
auto& segment = segment_visitor_->GetSegment();
auto& snapshot = segment_visitor_->GetSnapshot();
auto segment_commit = snapshot->GetSegmentCommitBySegmentId(segment->GetID());
faiss::ConcurrentBitsetPtr concurrent_bitset_ptr =
std::make_shared<faiss::ConcurrentBitset>(segment_commit->GetRowCount());
faiss::ConcurrentBitsetPtr concurrent_bitset_ptr = std::make_shared<faiss::ConcurrentBitset>(uids.size());
segment::DeletedDocsPtr deleted_docs_ptr;
STATUS_CHECK(LoadDeletedDocs(deleted_docs_ptr));
@ -275,10 +277,6 @@ SegmentReader::LoadVectorIndex(const std::string& field_name, knowhere::VecIndex
}
}
// load uids
std::vector<int64_t> uids;
STATUS_CHECK(LoadUids(uids));
knowhere::BinarySet index_data;
knowhere::BinaryPtr raw_data, compress_data;
@ -304,7 +302,7 @@ SegmentReader::LoadVectorIndex(const std::string& field_name, knowhere::VecIndex
engine::BinaryDataPtr raw;
STATUS_CHECK(LoadField(field_name, raw, false));
auto dataset = knowhere::GenDataset(segment_commit->GetRowCount(), dimension, raw->data_.data());
auto dataset = knowhere::GenDataset(uids.size(), dimension, raw->data_.data());
// construct IDMAP index
knowhere::VecIndexFactory& vec_index_factory = knowhere::VecIndexFactory::GetInstance();