problem and admin initialization is only done if needed

1 files changed, 12 insertions, 0 deletions

diff --git a/src/database/database.rs b/src/database/database.rs
index ee9b6b9..90a54bc 100644
--- a/src/database/database.rs
+++ b/src/database/database.rs
@@ -284,6 +284,18 @@ impl Database {
         
         Ok(())
     }
+
+    pub fn initialize_problems(&self) -> Result<(), DatabaseError> {
+        static QUERY: &str = include_str!("sql/initialize_problems.sql");
+        let conn = self.pool
+            .get()
+            .map_err(|e| DatabaseError::Connection(e.to_string()))?;
+        
+        conn.execute_batch(QUERY)
+            .map_err(|e| DatabaseError::Query(e.to_string()))?;
+        
+        Ok(())
+    }
 }
 
 #[cfg(test)]