fix memory leaks when shrinking the tree

yahoojapan · Aug 31, 2023 · 10c8e23 · 10c8e23
1 parent eaee806
commit 10c8e23
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-2.1.2
+2.1.3
diff --git a/lib/NGT/Tree.h b/lib/NGT/Tree.h
@@ -248,9 +248,17 @@ namespace NGT {
       removeNode(Node::ID id) {
       size_t idx = id.getID();
       if (id.getType() == Node::ID::Leaf) {
+#ifndef NGT_SHARED_MEMORY_ALLOCATOR
+	LeafNode &n = *static_cast<LeafNode*>(getNode(id));
+	delete n.pivot;
+#endif
 	leafNodes.remove(idx);
       } else {
-	internalNodes.remove(idx);
+#ifndef NGT_SHARED_MEMORY_ALLOCATOR
+	InternalNode &n = *static_cast<InternalNode*>(getNode(id));
+	delete n.pivot;
+#endif
+	internalNodes.remove(idx);	  
       }
     }