summaryrefslogtreecommitdiff
path: root/src/database/database.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/database/database.rs')
-rw-r--r--src/database/database.rs12
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)]