mandlm/VectorSearch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: reduce mutex waits in parallel finder

Browse source
This commit is contained in:
Michael Mandl 2024-03-20 13:33:47 +01:00
parent ba460cc00a
commit 1ab711f81c
Signed by: mandlm
GPG key ID: 4AA25D647AA54CC7
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lib_vector_search/src/parallel_finder.cpp
View file

@ -27,13 +27,17 @@ ParallelFinder::find_prefix(string_view search_term) const {
[](const vector<string> &word_list, const string_view &search_term, [](const vector<string> &word_list, const string_view &search_term,
forward_list<const string *> &result, size_t start_index, forward_list<const string *> &result, size_t start_index,
size_t end_index, mutex &result_mutex) { size_t end_index, mutex &result_mutex) {
forward_list<const string *> thread_results;
for (size_t index = start_index; index < end_index; ++index) { for (size_t index = start_index; index < end_index; ++index) {
const auto &current_word = word_list[index]; const auto &current_word = word_list[index];
if (current_word.starts_with(search_term)) { if (current_word.starts_with(search_term)) {
const lock_guard<mutex> lock(result_mutex); thread_results.push_front(&current_word);
result.push_front(&current_word);
} }
} }
if (!thread_results.empty()) {
const lock_guard<mutex> lock(result_mutex);
result.merge(thread_results);
}
}, },
cref(word_list_), cref(search_term), ref(result), first_word_index, cref(word_list_), cref(search_term), ref(result), first_word_index,
last_word_index, ref(result_mutex)); last_word_index, ref(result_mutex));