fixed closest_set memory leak

bosonnetwork · Sep 20, 2023 · 8d7e04c · 8d7e04c
1 parent 3e34dba
commit 8d7e04c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/core/task/closest_set.h b/src/core/task/closest_set.h
@@ -52,13 +52,13 @@ class ClosestSet {
         closest[cn->getId()] = cn;
         if (closest.size() > capacity) {
             auto last = std::prev(closest.cend());
-            closest.erase(last);
-
             if (last->second == cn) {
                 insertAttemptsSinceTailModification++;
             } else {
                 insertAttemptsSinceTailModification = 0;
             }
+
+            closest.erase(last->first);
         }
 
         const auto head = closest.cbegin()->second;
@@ -102,7 +102,7 @@ class ClosestSet {
 
 private:
     const Id& target;
-    int capacity;
+    int capacity {0};
 
     std::map<Id, Sp<CandidateNode>> closest {};