diff options
Diffstat (limited to 'src/process_mail.rs')
| -rw-r--r-- | src/process_mail.rs | 23 |
1 files changed, 20 insertions, 3 deletions
diff --git a/src/process_mail.rs b/src/process_mail.rs index 5a22ce1..90fdd2a 100644 --- a/src/process_mail.rs +++ b/src/process_mail.rs @@ -1,11 +1,20 @@ +use std::sync::Arc; + use anyhow::Result; -use tokio::sync::mpsc::{Receiver, Sender}; +use tokio::sync::{ + Mutex, + mpsc::{Receiver, Sender}, +}; -use crate::model::Mail; +use crate::{ + database::Database, + model::{Mail, MailQuery}, +}; struct MailProcessor { rx: Receiver<Mail>, tx: Sender<Mail>, + db: Arc<Mutex<Database>>, } // TODO: Store first then forward. On complete forward remove from db, otherwise save db with the @@ -15,9 +24,17 @@ impl MailProcessor { // TODO: Check againts self.rx.is_closed() instead to stop the program before consuming all // of the queue. Store the emails not yet processed and restore the queue upon startup. while let Some(mail) = self.rx.recv().await { - todo!() + let mut db_guard = self.db.lock().await; + db_guard.execute(MailQuery::Insert(&mail))?; + drop(db_guard); + + todo!(); } Ok(()) } } + +mod tests { + // TODO: Think of a way to test +} |
