feat: init leaderboard for all users before server starts

src/app/init/mod.rs

@@ -0,0 +1,51 @@
+use log::{error, info};
+
+use crate::service::database::{
+    conn::DbPool,
+    models::{Leaderboard, User},
+};
+
+pub async fn initialize_leaderboards(pool: &DbPool) -> Result<(), Box<dyn std::error::Error>> {
+    let mut conn = pool.get()?;
+
+    // Get all users
+    let users = User::get_all_users(&mut conn)?;
+
+    if users.is_empty() {
+        info!("No users found in database - skipping leaderboard initialization");
+        return Ok(());
+    }
+
+    let mut initialized_count = 0;
+    for user in users {
+        // Check if user has a leaderboard entry
+        match Leaderboard::get_leaderboard(&mut conn, Some(&user.id)) {
+            Ok(leaderboard) => {
+                if leaderboard.is_empty() {
+                    // Create new leaderboard for user
+                    let new_leaderboard = Leaderboard::new(&user.id, None, 0, 0);
+                    Leaderboard::create(&mut conn, new_leaderboard)?;
+                    initialized_count += 1;
+                }
+            }
+            Err(e) => {
+                error!("Error checking leaderboard for user {}: {}", user.id, e);
+                // Create new leaderboard for user
+                let new_leaderboard = Leaderboard::new(&user.id, None, 0, 0);
+                Leaderboard::create(&mut conn, new_leaderboard)?;
+                initialized_count += 1;
+            }
+        }
+    }
+
+    if initialized_count > 0 {
+        info!(
+            "Initialized {} missing leaderboard records",
+            initialized_count
+        );
+    } else {
+        info!("All users have leaderboard records - no initialization needed");
+    }
+
+    Ok(())
+}

src/app/mod.rs

@@ -1,4 +1,5 @@
 pub mod auth;
+pub mod init;
 pub mod routes;
 pub mod progress;
 pub mod repo;

src/main.rs

@@ -2,11 +2,13 @@ use actix_cors::Cors;
 use actix_web::{middleware::Logger, web, App, HttpServer};
 use app::{
     auth::middleware::AuthMiddleware,
+    init::initialize_leaderboards,
     routes,
     websockets::{handler::websocket_handler, manager::WebSocketManagerHandle},
 };
 use dotenvy::dotenv;
 use env_logger::Env;
+use log::error;
 use service::{database::conn::get_connection_pool, queue::consume_queue};
 
 mod app;
@@ -37,6 +39,20 @@ async fn main() -> std::io::Result<()> {
     let connection_url =
         std::env::var("CONNECTION_URL").unwrap_or_else(|_| "127.0.0.1:4925".to_string());
     let pool = get_connection_pool();
+
+    // Initialize leaderboards before starting server
+    // This is done to ensure that leaderboards are created for all users
+    // since we have some users created before the leaderboard service was
+    // added to signup.
+    // TODO: Remove this once all users have leaderboard records
+    if let Err(e) = initialize_leaderboards(&pool).await {
+        error!("Failed to initialize leaderboards: {}", e);
+        return Err(std::io::Error::new(
+            std::io::ErrorKind::Other,
+            format!("Failed to initialize leaderboards: {}", e),
+        ));
+    }
+
     let manager_handle = WebSocketManagerHandle::new();
     let manager_handle_clone = manager_handle.clone();
 

src/service/repository/users.rs

@@ -132,4 +132,8 @@ impl User {
             _ => Err(anyhow::anyhow!("No input provided")),
         }
     }
+
+    pub fn get_all_users(conn: &mut PgConnection) -> QueryResult<Vec<User>> {
+        users::table.select(User::as_select()).load(conn)
+    }
 }