fix the memory leak issue in VisitListPool allocation (#229)

Signed-off-by: jinjiabao.jjb <jinjiabao.jjb@antgroup.com>
antgroup · Dec 19, 2024 · 4549cd3 · 4549cd3
1 parent 3baa6e1
commit 4549cd3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/include/vsag/allocator.h b/include/vsag/allocator.h
@@ -40,7 +40,12 @@ class Allocator {
     T*
     New(Args&&... args) {
         void* p = Allocate(sizeof(T));
-        return (T*)::new (p) T(std::forward<Args>(args)...);
+        try {
+            return (T*)::new (p) T(std::forward<Args>(args)...);
+        } catch (std::exception& e) {
+            Deallocate(p);
+            throw e;
+        }
     }
 
     template <typename T>