diff options
Diffstat (limited to 'src/database/database.rs')
| -rw-r--r-- | src/database/database.rs | 12 |
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)] |
