also log message_id in DB

ibotty · Oct 17, 2023 · b7a1bb0 · b7a1bb0
1 parent 9293e34
commit b7a1bb0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/db.rs b/src/db.rs
@@ -4,6 +4,7 @@ use sqlx::postgres::PgPool;
 
 pub async fn insert_mail(
     pool: &PgPool,
+    message_id: &str,
     rcpt: &str,
     from: &str,
     body_text: &str,
@@ -13,8 +14,9 @@ pub async fn insert_mail(
 ) -> Result<()> {
     let query = sqlx::query!(
         r#"INSERT INTO data_gateways.smtp_gateway
-            ("to", "from", body_text, body_html, headers, attachments)
-            VALUES ($1, $2, $3, $4, $5, $6);"#,
+            (message_id, "to", "from", body_text, body_html, headers, attachments)
+            VALUES ($1, $2, $3, $4, $5, $6, $7);"#,
+        message_id,
         rcpt,
         from,
         body_text,

diff --git a/src/s3.rs b/src/s3.rs
@@ -88,6 +88,7 @@ pub async fn upload_message(
     // afterwards, when complete, insert into DB
     db::insert_mail(
         pg_pool,
+        message_id,
         rcpt,
         from,
         body_text