Gate databases behind features #28

Author: SudoDios

Cargo.toml

@@ -5,9 +5,16 @@ edition = "2021"
 authors = ["Sudo Dios"]
 description = "Rust backend for LibreSpeed"
 
+[features]
+default = ["sqlite","mysql","postgres"]
+
+sqlite = ["dep:rusqlite"]
+mysql = ["dep:mysql"]
+postgres = ["dep:postgres"]
+
 [dependencies]
 #async net
-tokio = {version = "1.48.0", features = ["net","io-util","rt","macros","rt-multi-thread"]}
+tokio = {version = "1.48.0", features = ["net","io-util","rt","macros","rt-multi-thread","sync"]}
 tokio-rustls = {version = "0.26.4", features = ["tls12","ring"], default-features = false}
 webpki-roots = "1.0.3"
 rustls-pemfile = "2.2.0"
@@ -25,9 +32,9 @@ serde = { version = "1.0.228", features = ["derive"] }
 #databases
 uuid = { version = "1.18.1", features = ["v4"] }
 chrono = "0.4.42"
-mysql = { version = "26.0.1",default-features = false }
-postgres = "0.19.12"
-rusqlite = { version = "0.37.0",features = ["bundled"] }
+mysql = { version = "26.0.1",default-features = false, optional = true }
+postgres = { version = "0.19.12", optional = true }
+rusqlite = { version = "0.37.0",features = ["bundled"], optional = true }
 #conf
 clap = { version = "4.5.49",features = ["std","color","help","usage"],default-features = false }
 toml = "0.9.8"

src/database/mod.rs

@@ -5,15 +5,21 @@ use tokio::sync::Mutex;
 use uuid::Uuid;
 use crate::config::SERVER_CONFIG;
 use crate::database::memory::MemoryDB;
+#[cfg(feature = "mysql")]
 use crate::database::mysql::MySql;
 use crate::database::none::NoneDB;
+#[cfg(feature = "postgres")]
 use crate::database::postgres::Postgres;
+#[cfg(feature = "sqlite")]
 use crate::database::sqlite::SQLite;
 use crate::results::TelemetryData;
 
+#[cfg(feature = "mysql")]
 mod mysql;
 mod none;
+#[cfg(feature = "postgres")]
 mod postgres;
+#[cfg(feature = "sqlite")]
 mod sqlite;
 mod memory;
 
@@ -34,16 +40,19 @@ pub fn generate_uuid () -> String {
 pub fn init () -> std::io::Result<Arc<Mutex<dyn Database + Send>>> {
     let config = SERVER_CONFIG.get().unwrap();
     match config.database_type.as_str() {
+        #[cfg(feature = "mysql")]
         "mysql" => {
             let mysql_setup = mysql::init(&config.database_username,&config.database_password,&config.database_hostname,&config.database_name)?;
             info!("Database {} initialized successfully","Mysql");
             Ok(Arc::new(Mutex::new(MySql{connection : mysql_setup})))
         }
+        #[cfg(feature = "postgres")]
         "postgres" => {
             let postgres_setup = postgres::init(&config.database_username,&config.database_password,&config.database_hostname,&config.database_name)?;
             info!("Database {} initialized successfully","Postgres");
             Ok(Arc::new(Mutex::new(Postgres {connection : postgres_setup})))
         }
+        #[cfg(feature = "sqlite")]
         "sqlite" => {
             let sqlite_setup = sqlite::init(&config.database_file)?;
             info!("Database {} initialized successfully","Sqlite");