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

feat: support range search pagination retains order (#35738) 

issue: #35464

Signed-off-by: Patrick Weizhi Xu <weizhi.xu@zilliz.com>
pull/35815/head
Patrick Weizhi Xu 2024-08-29 14:09:00 +08:00 committed by GitHub
parent 73be0ba941
commit b3089b5bdc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
internal/core/src/common/Consts.h
View File

@ -45,6 +45,7 @@ const char OFFSET_MAPPING_NAME[] = "offset_mapping";
const char NUM_CLUSTERS[] = "num_clusters"; const char NUM_CLUSTERS[] = "num_clusters";
const char KMEANS_CLUSTER[] = "KMEANS"; const char KMEANS_CLUSTER[] = "KMEANS";
const char VEC_OPT_FIELDS[] = "opt_fields"; const char VEC_OPT_FIELDS[] = "opt_fields";
const char PAGE_RETAIN_ORDER[] = "page_retain_order";
const char DEFAULT_PLANNODE_ID[] = "0"; const char DEFAULT_PLANNODE_ID[] = "0";
const char DEAFULT_QUERY_ID[] = "0"; const char DEAFULT_QUERY_ID[] = "0";

8
internal/core/src/index/VectorDiskIndex.cpp
View File

@ -282,6 +282,14 @@ VectorDiskAnnIndex<T>::Query(const DatasetPtr dataset,
search_config[RANGE_FILTER], search_config[RANGE_FILTER],
GetMetricType()); GetMetricType());
} }
auto page_retain_order = GetValueFromConfig<bool>(
search_info.search_params_, PAGE_RETAIN_ORDER);
if (page_retain_order.has_value()) {
search_config[knowhere::meta::RETAIN_ITERATOR_ORDER] =
page_retain_order.value();
}
auto res = index_.RangeSearch(dataset, search_config, bitset); auto res = index_.RangeSearch(dataset, search_config, bitset);
if (!res.has_value()) { if (!res.has_value()) {