From 11052951b74b9ad4b6a9412ae490086235f9154b Mon Sep 17 00:00:00 2001
From: Joris
Date: Sun, 3 Jan 2021 13:40:40 +0100
Subject: Rewrite in Rust

---
 src/db/jobs.rs | 56 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 src/db/jobs.rs

(limited to 'src/db/jobs.rs')

diff --git a/src/db/jobs.rs b/src/db/jobs.rs
new file mode 100644
index 0000000..88c2005
--- /dev/null
+++ b/src/db/jobs.rs
@@ -0,0 +1,56 @@
+use sqlx::error::Error;
+use sqlx::sqlite::SqlitePool;
+
+use crate::model::job::Job;
+
+pub async fn should_run(pool: &SqlitePool, job: Job) -> bool {
+    let run_from = match job {
+        Job::WeeklyReport => "date('now', 'weekday 0', '-6 days')",
+        Job::MonthlyPayment => "date('now', 'start of month')",
+    };
+
+    let query = format!(
+        r#"
+SELECT
+    1
+FROM
+    jobs
+WHERE
+    name = ?
+    AND last_execution < {}
+    "#,
+        run_from
+    );
+
+    let res = sqlx::query(&query).bind(job).fetch_one(pool).await;
+
+    match res {
+        Ok(_) => true,
+        Err(Error::RowNotFound) => false,
+        Err(err) => {
+            error!("Error looking if job should run: {:?}", err);
+            false
+        }
+    }
+}
+
+pub async fn actualize_last_execution(pool: &SqlitePool, job: Job) -> () {
+    let query = r#"
+UPDATE
+    jobs
+SET
+    last_execution = datetime()
+WHERE
+    name = ?
+    "#;
+
+    let res = sqlx::query(query).bind(job).execute(pool).await;
+
+    match res {
+        Ok(_) => (),
+        Err(err) => {
+            error!("Error actualizing job last execution: {:?}", err);
+            ()
+        }
+    }
+}
-- 
cgit v1.2.3