Garbage collect expired clips

Cargo.toml

@@ -19,6 +19,7 @@ fern = "0.5"
 dotenv = "0.15.0"
 diesel = { version = "2.1.4", features = ["postgres", "chrono"] }
 regex = "1.10"
+clokwerk = "0.3.5"
 
 ## file things
 aws-config = "0.14.0"

src/main.rs

@@ -2,6 +2,7 @@ mod models;
 mod schema;
 mod utils;
 
+use clokwerk::{Scheduler, TimeUnits};
 use regex::Regex;
 use rocket::http::{Header, Status};
 use rocket::response::status::Custom;
@@ -22,7 +23,7 @@ use std::time::Duration;
 use rocket::form::Form;
 use rocket::serde::json::Json;
 
-use utils::db;
+use utils::db::{self, collect_garbage};
 
 use crate::utils::rate_limit::RateLimiter;
 use crate::utils::structs::{APIResponse, APIStatus};
@@ -379,6 +380,12 @@ async fn rocket() -> _ {
 
     let s3_client = create_storage_client().await.unwrap();
 
+    let mut db_connection = db::initialize().expect("Failed to connect to the database");
+    let mut scheduler = Scheduler::with_tz(chrono::Utc);
+    scheduler.every(1.hour()).run(move || {
+        let _ = collect_garbage(&mut db_connection);
+    });
+
     rocket::build()
         .mount(
             "/api",

src/utils/db.rs

@@ -69,3 +69,10 @@ pub fn insert_clip(
         .values(&new_clip)
         .get_result::<Clip>(connection)
 }
+
+/// Deletes expired clips from the database
+pub fn collect_garbage(connection: &mut PgConnection) -> Result<usize, diesel::result::Error> {
+    use crate::schema::clips::dsl::*;
+
+    diesel::delete(clips.filter(expires_at.is_not_null().and(expires_at.lt(chrono::Local::now().naive_local())))).execute(connection)
+}