modify the data structure of VisitList for stack operations (#206)

Signed-off-by: jinjiabao.jjb <jinjiabao.jjb@antgroup.com>
antgroup · Dec 11, 2024 · 62312c0 · 62312c0
1 parent 2f67c39
commit 62312c0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/algorithm/hnswlib/visited_list_pool.h b/src/algorithm/hnswlib/visited_list_pool.h
@@ -88,7 +88,7 @@ class VisitedListPool {
             std::unique_lock<std::mutex> lock(poolguard_);
             if (not pool_.empty()) {
                 rez = pool_.front();
-                pool_.pop_back();
+                pool_.pop_front();
             } else {
                 rez = std::make_shared<VisitedList>(max_element_count_, allocator_);
             }
@@ -104,7 +104,7 @@ class VisitedListPool {
     }
 
 private:
-    vsag::Vector<std::shared_ptr<VisitedList>> pool_;
+    vsag::Deque<std::shared_ptr<VisitedList>> pool_;
     std::mutex poolguard_;
     uint64_t max_element_count_;
     vsag::Allocator* allocator_;

diff --git a/src/utils.h b/src/utils.h
@@ -17,6 +17,7 @@
 
 #include <chrono>
 #include <cstdint>
+#include <deque>
 #include <string>
 #include <vector>
 
@@ -34,6 +35,9 @@ using UnorderedSet =
 template <typename T>
 using Vector = std::vector<T, vsag::AllocatorWrapper<T>>;
 
+template <typename T>
+using Deque = std::deque<T, vsag::AllocatorWrapper<T>>;
+
 template <typename KeyType, typename ValType>
 using UnorderedMap = std::unordered_map<KeyType,
                                         ValType,