database: add Events::force_insert

Signed-off-by: Yuki Kishimoto <[email protected]>

Author: yukibtc

CHANGELOG.md

@@ -92,6 +92,7 @@
 * nostr: impl `FromIterator<Tag>` for `Tags` ([Yuki Kishimoto])
 * nostr: add `EventDeletionRequest` struct ([Yuki Kishimoto])
 * nostr: add `notifications` field to NIP47 `GetInfoResponse` ([Yuki Kishimoto])
+* database: add `Events::force_insert` ([Yuki Kishimoto])
 * pool: event verification cache ([Yuki Kishimoto])
 * pool: add `AdmitPolicy` trait ([Yuki Kishimoto])
 * ffi: add Mac Catalyst support in Swift package ([Yuki Kishimoto])

crates/nostr-database/src/collections/events.rs

@@ -70,11 +70,22 @@ impl Events {
     /// Insert [`Event`]
     ///
     /// If the set did not previously contain an equal value, `true` is returned.
+    /// If the collection is full, the older events will be discarded.
+    /// Use [`Events::force_insert`] to always make sure the event is inserted.
     #[inline]
     pub fn insert(&mut self, event: Event) -> bool {
         self.set.insert(event).inserted
     }
 
+    /// Force insert [`Event`]
+    ///
+    /// Use [`Events::insert`] to respect the max collection capacity (if any).
+    /// If the collection capacity is full, this method will increase it.
+    #[inline]
+    pub fn force_insert(&mut self, event: Event) -> bool {
+        self.set.force_insert(event).inserted
+    }
+
     /// Insert events
     #[inline]
     pub fn extend<I>(&mut self, events: I)

crates/nostr-database/src/collections/tree.rs

@@ -237,6 +237,28 @@ where
         }
     }
 
+    /// Force insert the value
+    ///
+    /// If the capacity is full, automatically increases the capacity.
+    pub fn force_insert(&mut self, value: T) -> InsertResult<T> {
+        let inserted: bool = self.set.insert(value);
+
+        // If successfully inserted, check if the capacity must be increased
+        if inserted {
+            if let Capacity::Bounded { max, .. } = self.capacity {
+                if self.set.len() >= max {
+                    self.capacity = Capacity::bounded(max + 1);
+                }
+            }
+        }
+
+        // Insert value
+        InsertResult {
+            inserted,
+            pop: None,
+        }
+    }
+
     /// Extend with values
     pub fn extend<I>(&mut self, values: I)
     where