Merge pull request #148 from MathisBurger/feature/delete-20day-old

Deletion scheduler

Author: MathisBurger

tasky/Cargo.lock

@@ -32,6 +32,7 @@ awc = "3.5.1"
 actix-http = "3.9.0"
 serial_test = "0.5"
 diesel_full_text_search = "2.2.0"
+actix-rt = "2.10.0"
 
 [build-dependencies]
 tonic-build = "0.12.2"

tasky/Cargo.toml

@@ -0,0 +1,30 @@
+use crate::{
+    models::DB, schema::group_join_requests::dsl as join_request_dsl,
+    schema::notifications::dsl as notification_dsl,
+};
+use actix_rt::time;
+use diesel::dsl::IntervalDsl;
+use diesel::prelude::*;
+use std::time::Duration;
+
+/// Schedules deletion of entities after specific amount of time
+pub async fn scheduler(conn: &mut DB) {
+    let mut interval = time::interval(Duration::from_secs(3600 * 24));
+    loop {
+        diesel::delete(
+            join_request_dsl::group_join_requests
+                .filter(join_request_dsl::updated_at.lt(diesel::dsl::now - 20_i32.days())),
+        )
+        .execute(conn)
+        .expect("Cannot delete join requests");
+
+        diesel::delete(
+            notification_dsl::notifications
+                .filter(notification_dsl::created_at.lt(diesel::dsl::now - 20_i32.days())),
+        )
+        .execute(conn)
+        .expect("Cannot delete notifications");
+
+        interval.tick().await;
+    }
+}

tasky/src/deletion_scheduler.rs

@@ -13,6 +13,7 @@ pub struct AppState {
 }
 
 pub mod auth_middleware;
+pub mod deletion_scheduler;
 pub mod error;
 pub mod grpc;
 pub mod handler;

tasky/src/lib.rs

@@ -6,6 +6,7 @@ use futures::future::join;
 use log::info;
 use std::net::SocketAddr;
 use tasky::auth_middleware::Auth;
+use tasky::deletion_scheduler;
 use tasky::routes::init_services;
 use tasky::spotlight;
 use tasky::tasky_grpc::tasky_api_server::TaskyApiServer;
@@ -31,6 +32,13 @@ async fn main() -> std::io::Result<()> {
         )
         .await
     };
+
+    let db_pool = state.clone().db.db.clone();
+    actix_rt::spawn(async move {
+        let db_conn = &mut db_pool.get().unwrap();
+        deletion_scheduler::scheduler(db_conn).await;
+    });
+
     let actix = HttpServer::new(move || {
         App::new()
             .wrap(middleware::Logger::default())