Move add transaction to db

will-lynas · Oct 13, 2024 · 1fc8d2c · 1fc8d2c
1 parent 99bb26f
commit 1fc8d2c
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 13 deletions.
diff --git a/src/db.rs b/src/db.rs
@@ -43,4 +43,20 @@ impl<'a> DB<'a> {
         .await
         .unwrap();
     }
+
+    pub async fn add_transaction(&self, transaction: Transaction) {
+        sqlx::query!(
+            r#"
+                INSERT INTO transactions (chatID, userID, description, amount)
+                VALUES (?1, ?2, ?3, ?4)
+                "#,
+            self.chat_id,
+            transaction.user_id,
+            transaction.description,
+            transaction.amount
+        )
+        .execute(self.pool)
+        .await
+        .unwrap();
+    }
 }
diff --git a/src/main.rs b/src/main.rs
@@ -3,7 +3,10 @@ use std::{
     path::Path,
 };
 
-use db::DB;
+use db::{
+    Transaction,
+    DB,
+};
 use dotenv::from_path;
 use dptree::case;
 use sqlx::sqlite::SqlitePool;
@@ -180,19 +183,13 @@ async fn receive_title(
             let user = msg.from.unwrap();
             let user_id = user.id.0 as i64;
 
-            sqlx::query!(
-                r#"
-                INSERT INTO transactions (chatID, userID, description, amount)
-                VALUES (?1, ?2, ?3, ?4)
-                "#,
-                chat_id,
+            let transaction = Transaction {
                 user_id,
-                title,
-                amount
-            )
-            .execute(&pool)
-            .await
-            .unwrap();
+                description: Some(title.into()),
+                amount: Some(amount),
+            };
+
+            DB::new(&pool, chat_id).add_transaction(transaction).await;
 
             bot.send_message(
                 msg.chat.id,