Method SearchWithRequest

src/algorithm/ivf.cpp:943–982

}

DatasetPtr
IVF::SearchWithRequest(const SearchRequest& request) const {
    SearchStatistics stats;
    QueryContext ctx{.alloc = request.search_allocator_, .stats = &stats};

    auto param = this->create_search_param(request.params_str_, request.filter_);
    param.search_mode = KNN_SEARCH;
    param.topk = request.topk_;
    if (use_reorder_) {
        CHECK_ARGUMENT(
            param.factor > 0.0F,
            fmt::format("factor must be positive when use_reorder is true, got {}", param.factor));
        param.topk = static_cast<int64_t>(param.factor * static_cast<float>(request.topk_));
    }
    auto query = request.query_;
    if (request.enable_attribute_filter_ and this->attr_filter_index_ != nullptr) {
        auto& schema = this->attr_filter_index_->field_type_map_;
        auto expr = AstParse(request.attribute_filter_str_, &schema);
        for (int64_t i = 0; i < param.parallel_search_thread_count; ++i) {
            auto executor =
                Executor::MakeInstance(this->allocator_, expr, this->attr_filter_index_);
            executor->Init();
            param.executors.emplace_back(executor);
        }
    }
    auto search_result = this->search<KNN_SEARCH>(query, param, ctx);
    if (use_reorder_) {
        return reorder(request.topk_, search_result, query->GetFloat32Vectors(), param, ctx);
    }
    auto count = static_cast<const int64_t>(search_result->Size());
    auto [dataset_results, dists, labels] = create_fast_dataset(count, allocator_);
    for (int64_t j = count - 1; j >= 0; --j) {
        dists[j] = search_result->Top().first;
        labels[j] = label_table_->GetLabelById(search_result->Top().second);
        search_result->Pop();
    }

    dataset_results->Statistics(stats.Dump());
    return std::move(dataset_results);
}

void
IVF::fill_location_map() {