basically everything is done uwu

services/tasks/Cargo.toml

@@ -6,3 +6,5 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+redis = "0.22.1"
+once_cell = "1.16.0"

services/tasks/src/main.rs

@@ -1,3 +1,48 @@
-fn main() {
-    println!("Hello, world!");
+use once_cell::sync::Lazy;
+use redis::Commands;
+use std::env::args;
+use std::ops::Sub;
+use std::time::{Duration, SystemTime};
+
+pub static REDIS_CLIENT: Lazy<redis::Client> = Lazy::new(|| {
+    redis::Client::open(std::env::var("REDIS_ADDR").unwrap_or("redis://localhost:6379".to_string()))
+        .unwrap()
+});
+
+fn cmd_prune() {
+    println!("Pruning old data...");
+    let mut con = REDIS_CLIENT.get_connection().unwrap();
+
+    let prune_after = SystemTime::now()
+        .sub(Duration::from_secs(60))
+        .duration_since(SystemTime::UNIX_EPOCH)
+        .unwrap()
+        .as_secs();
+
+    let keys: Vec<String> = con.keys("wp:*").unwrap();
+    for key in keys {
+        println!("-> Pruning {}", key);
+        let removed_items: u64 = con.zrembyscore(key, 0, prune_after).unwrap();
+        println!("==> Removed {} items", removed_items);
+    }
+}
+
+fn cmd_help() {
+    println!("Usage: {} [command]", args().nth(0).unwrap());
+    println!("Commands:");
+    println!("  help - Show this help message");
+    println!("  prune - Remove stale data from Redis");
+}
+
+fn main() {
+    let command = args().nth(1).unwrap_or("help".to_string());
+
+    match command.as_str() {
+        "help" => cmd_help(),
+        "prune" => cmd_prune(),
+        _ => {
+            println!("Unknown command: {}", command);
+            cmd_help();
+        }
+    }
 }