feat: conditionally init postgres/mariadb connection

src/db.rs

@@ -0,0 +1,60 @@
+/*
+ * Copyright (C) 2022  Aravinth Manivannan <realaravinth@batsense.net>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+use crate::settings::Settings;
+use db_core::prelude::*;
+
+pub type BoxDB = Box<dyn MCDatabase>;
+
+pub mod pg {
+    use super::*;
+    use db_sqlx_postgres::{ConnectionOptions, Fresh};
+    use sqlx::postgres::PgPoolOptions;
+
+    pub async fn get_data(settings: Option<Settings>) -> BoxDB {
+        let settings = settings.unwrap_or_else(|| Settings::new().unwrap());
+        let pool = settings.database.pool;
+        let pool_options = PgPoolOptions::new().max_connections(pool);
+        let connection_options = ConnectionOptions::Fresh(Fresh {
+            pool_options,
+            url: settings.database.url.clone(),
+            disable_logging: !settings.debug,
+        });
+        let db = connection_options.connect().await.unwrap();
+        db.migrate().await.unwrap();
+        Box::new(db)
+    }
+}
+
+pub mod maria {
+    use super::*;
+    use db_sqlx_maria::{ConnectionOptions, Fresh};
+    use sqlx::mysql::MySqlPoolOptions;
+
+    pub async fn get_data(settings: Option<Settings>) -> BoxDB {
+        let settings = settings.unwrap_or_else(|| Settings::new().unwrap());
+        let pool = settings.database.pool;
+        let pool_options = MySqlPoolOptions::new().max_connections(pool);
+        let connection_options = ConnectionOptions::Fresh(Fresh {
+            pool_options,
+            url: settings.database.url.clone(),
+            disable_logging: !settings.debug,
+        });
+        let db = connection_options.connect().await.unwrap();
+        db.migrate().await.unwrap();
+        Box::new(db)
+    }
+}