From 9e0a6dc3d2e6d402a2f985e47a876156d42c198b Mon Sep 17 00:00:00 2001
From: Zixuan Chen <remch183@outlook.com>
Date: Sun, 22 Oct 2023 16:01:45 +0800
Subject: [PATCH] feat: reserve (#44)

* feat: reserve

* fix: take currently free slots into account
---
 src/arena.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/arena.rs b/src/arena.rs
index 9f5eee0..24991c1 100644
--- a/src/arena.rs
+++ b/src/arena.rs
@@ -166,6 +166,13 @@ impl<T> Arena<T> {
         self.storage.capacity()
     }
 
+    /// Reserve capacity for at least `additional` more elements to be inserted
+    pub fn reserve(&mut self, additional: usize) {
+        let currently_free = self.storage.len().saturating_sub(self.len as usize);
+        let to_reserve = additional.saturating_sub(currently_free);
+        self.storage.reserve(to_reserve);
+    }
+
     /// Returns whether the arena is empty.
     pub const fn is_empty(&self) -> bool {
         self.len == 0