fix: run_on_start command configuring SQLite pragmas

Author: ryands17

src/config.rs

@@ -219,9 +219,22 @@ pub struct Database {
     #[serde(default)]
     pub dangerously_recreate: bool,
 
-    // sqlite configuration
-    /// set the foreign key pragma to be enabled or disabled
-    pub enable_foreign_keys: Option<bool>,
+    // sqlite run on start commands
+    /// this can be used to confiure PRAGMAs for SQLite where you can pass all values as a string.
+    /// Default values are:
+    ///
+    /// PRAGMA foreign_keys = ON;
+    ///
+    /// PRAGMA journal_mode = WAL;
+    ///
+    /// PRAGMA synchronous = NORMAL;
+    ///
+    /// PRAGMA mmap_size = 134217728;
+    ///
+    /// PRAGMA journal_size_limit = 67108864;
+    ///
+    /// PRAGMA cache_size = 2000;
+    pub run_on_start: Option<String>,
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize)]

src/db.rs

@@ -150,24 +150,18 @@ pub async fn connect(config: &config::Database) -> Result<DbConn, sea_orm::DbErr
     let db = Database::connect(opt).await?;
 
     if db.get_database_backend() == DatabaseBackend::Sqlite {
-        // default set to "ON" to prevent breaking changes
-        let foreign_keys = config
-            .enable_foreign_keys
-            .map(|fk| if fk { "ON" } else { "OFF" })
-            .unwrap_or("ON");
-
         db.execute(Statement::from_string(
             DatabaseBackend::Sqlite,
-            format!(
+            config.run_on_start.clone().unwrap_or(
                 "
-            PRAGMA foreign_keys = {foreign_keys};
+            PRAGMA foreign_keys = ON;
             PRAGMA journal_mode = WAL;
             PRAGMA synchronous = NORMAL;
             PRAGMA mmap_size = 134217728;
             PRAGMA journal_size_limit = 67108864;
             PRAGMA cache_size = 2000;
-            PRAGMA busy_timeout = 5000;
             "
+                .to_string(),
             ),
         ))
         .await?;

src/tests_cfg/config.rs

@@ -36,7 +36,7 @@ pub fn test_config() -> Config {
             auto_migrate: false,
             dangerously_truncate: false,
             dangerously_recreate: false,
-            enable_foreign_keys: None,
+            run_on_start: None,
         },
         queue: None,
         auth: None,