config support for DB_CONNECT_PARAMS_JSON

Author: CodeWarriorr

Cargo.lock

@@ -42,3 +42,4 @@ chrono = { version = "0.4.38", features = ["serde"] }
 regex = "1.11.1"
 moka = { version = "0.12.10", default-features = false, features = ["future"] }
 dotenvy = "0.15.7"
+urlencoding = "2.1.3"

url_finder/Cargo.toml

@@ -3,6 +3,8 @@ use std::env;
 use color_eyre::Result;
 use once_cell::sync::Lazy;
 
+use crate::types::DbConnectParams;
+
 pub static CONFIG: Lazy<Config> = Lazy::new(|| Config::new_from_env().unwrap());
 
 #[derive(Debug)]
@@ -13,8 +15,18 @@ pub struct Config {
 }
 impl Config {
     pub fn new_from_env() -> Result<Self> {
+        let db_url = env::var("DATABASE_URL").unwrap_or_else(|_| {
+            let json_params = env::var("DB_CONNECT_PARAMS_JSON")
+                .expect("DB_CONNECT_PARAMS_JSON environment variable not set");
+
+            let params: DbConnectParams =
+                serde_json::from_str(&json_params).expect("Invalid JSON in DB_CONNECT_PARAMS_JSON");
+
+            params.to_url()
+        });
+
         Ok(Self {
-            db_url: env::var("DATABASE_URL").expect("DATABASE_URL must be set"),
+            db_url,
             dmob_db_url: env::var("DMOB_DATABASE_URL").expect("DMOB_DATABASE_URL must be set"),
             log_level: env::var("LOG_LEVEL").unwrap_or("info".to_string()),
         })

url_finder/src/config.rs

@@ -38,6 +38,7 @@ mod provider_endpoints;
 mod repository;
 mod routes;
 mod services;
+mod types;
 mod url_tester;
 
 pub struct AppState {

url_finder/src/main.rs

@@ -0,0 +1,26 @@
+use serde::Deserialize;
+
+#[derive(Deserialize)]
+pub(super) struct DbConnectParams {
+    password: String,
+    dbname: String,
+    engine: String,
+    port: u16,
+    host: String,
+    username: String,
+}
+
+impl DbConnectParams {
+    pub fn to_url(&self) -> String {
+        format!(
+            "{}://{}:{}@{}:{}/{}?{}",
+            self.engine,
+            self.username,
+            urlencoding::encode(&self.password),
+            self.host,
+            self.port,
+            self.dbname,
+            std::env::var("DB_OPTIONS").unwrap_or_default(),
+        )
+    }
+}