Split make_historical_event out of get_entries

Libera-Chat · Oct 28, 2024 · b87c630 · b87c630
1 parent 71ae9c1
commit b87c630
Showing 1 changed file with 25 additions and 31 deletions.
diff --git a/sable_history/src/pg_history_service.rs b/sable_history/src/pg_history_service.rs
@@ -139,37 +139,7 @@ impl<'a> HistoryService for PgHistoryService<'a> {
                 }
                 .await
                 .expect("could not query messages")
-                .map(|row| {
-                    row.map(
-                        |(
-                            id,
-                            timestamp,
-                            message_type,
-                            text,
-                            source_nick,
-                            source_ident,
-                            source_vhost,
-                            source_account,
-                        ): (
-                            uuid::Uuid,
-                            NaiveDateTime,
-                            crate::types::MessageType,
-                            String,
-                            String,
-                            String,
-                            String,
-                            _,
-                        )| HistoricalEvent::Message {
-                            id: MessageId::new(id.try_into().expect("Message id is a non-v7 UUID")),
-                            timestamp: timestamp.and_utc().timestamp(),
-                            source: format!("{}!{}@{}", source_nick, source_ident, source_vhost),
-                            source_account,
-                            message_type: message_type.into(),
-                            target: channel.name.clone(), // assume it's the same
-                            text,
-                        },
-                    )
-                })
+                .map_ok(|row| make_historical_event(&channel, row))
                 .try_collect::<Vec<_>>()
                 .await
                 .expect("could not parse all records")
@@ -196,3 +166,27 @@ impl<'a> HistoryService for PgHistoryService<'a> {
         }
     }
 }
+
+fn make_historical_event(
+    channel: &crate::models::Channel,
+    (id, timestamp, message_type, text, source_nick, source_ident, source_vhost, source_account): (
+        uuid::Uuid,
+        NaiveDateTime,
+        crate::types::MessageType,
+        String,
+        String,
+        String,
+        String,
+        Option<String>,
+    ),
+) -> HistoricalEvent {
+    HistoricalEvent::Message {
+        id: MessageId::new(id.try_into().expect("Message id is a non-v7 UUID")),
+        timestamp: timestamp.and_utc().timestamp(),
+        source: format!("{}!{}@{}", source_nick, source_ident, source_vhost),
+        source_account,
+        message_type: message_type.into(),
+        target: channel.name.clone(), // assume it's the same
+        text,
+    }
+}