sdk: save event in database if never seen before

rust-nostr · Oct 16, 2023 · 77ca8a3 · 77ca8a3
1 parent d7b16be
commit 77ca8a3
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 12 deletions.
diff --git a/crates/nostr-sdk-db/src/memory.rs b/crates/nostr-sdk-db/src/memory.rs
@@ -27,6 +27,7 @@ impl From<Error> for DatabaseError {
 #[derive(Debug, Default)]
 pub struct MemoryDatabase {
     seen_event_ids: Arc<RwLock<HashMap<EventId, HashSet<Url>>>>,
+    // TODO: add messages queue? (messages not sent)
 }
 
 impl MemoryDatabase {

diff --git a/crates/nostr-sdk/src/relay/mod.rs b/crates/nostr-sdk/src/relay/mod.rs
@@ -432,16 +432,6 @@ impl Relay {
         self.relay_sender.max_capacity() - self.relay_sender.capacity()
     }
 
-    /* fn is_auto_connect_loop_running(&self) -> bool {
-        self.auto_connect_loop_running.load(Ordering::SeqCst)
-    }
-
-    fn set_auto_connect_loop_running(&self, value: bool) {
-        let _ =
-            self.auto_connect_loop_running
-                .fetch_update(Ordering::SeqCst, Ordering::SeqCst, |_| Some(value));
-    } */
-
     fn is_scheduled_for_stop(&self) -> bool {
         self.scheduled_for_stop.load(Ordering::SeqCst)
     }
@@ -786,7 +776,11 @@ impl Relay {
                         if size <= max_size {
                             match RelayMessage::from_json(&data) {
                                 Ok(msg) => {
-                                    tracing::trace!("Received message to {}: {:?}", relay.url, msg);
+                                    tracing::trace!(
+                                        "Received message from {}: {:?}",
+                                        relay.url,
+                                        msg
+                                    );
                                     if let Err(err) = relay
                                         .pool_sender
                                         .send(RelayPoolMessage::ReceivedMsg {

diff --git a/crates/nostr-sdk/src/relay/pool.rs b/crates/nostr-sdk/src/relay/pool.rs
@@ -163,9 +163,18 @@ impl RelayPoolTask {
                                             if !seen {
                                                 // Verifies if the event is valid
                                                 if event.verify().is_ok() {
+                                                    if let Err(e) =
+                                                        this.database.save_event(&event).await
+                                                    {
+                                                        tracing::error!(
+                                                            "Impossible to save event {}: {e}",
+                                                            event.id
+                                                        );
+                                                    }
+
                                                     let notification = RelayPoolNotification::Event(
                                                         relay_url.clone(),
-                                                        event.as_ref().clone(),
+                                                        *event.clone(),
                                                     );
                                                     let _ =
                                                         this.notification_sender.send(notification);