Enhanced config file

src/config/config.toml

@@ -3,4 +3,8 @@ host = "localhost"
 port = 5432
 username = "postgres"
 password = "root"
-database = "RustyDB"
+database = "RustyDB"
+
+[server]
+maxUser = 30
+timeout = 10000 #(ms)

src/config/mod.rs

@@ -7,17 +7,18 @@ use std::env;
 use crate::utils::Logs::UtilsData;
 
 #[derive(Serialize, Deserialize, Debug)]
-struct configToml {
-    database: Option<configTomlDatabase>
+pub(crate) struct configToml {
+    pub database: Option<configTomlDatabase>,
+    pub server: Option<configTomlServer>
 }
 
 #[derive(Serialize, Deserialize, Debug)]
-struct configTomlDatabase {
-    host: String,
-    port: u16,
-    username: String,
-    password: String,
-    database: String
+pub struct configTomlDatabase {
+    pub host: String,
+    pub port: u16,
+    pub username: String,
+    pub password: String,
+    pub database: String
 }
 
 #[derive(Debug)]
@@ -29,7 +30,20 @@ pub struct configData {
     pub database: String
 }
 
-impl configData {
+#[derive(Serialize, Deserialize, Debug)]
+pub struct configTomlServer {
+    pub maxUser: u16,
+    pub timeout: u32
+}
+
+#[derive(Debug)]
+pub struct configServer {
+    pub maxUser: u16,
+    pub timeout: u32
+}
+
+
+impl configToml {
     pub fn new() -> Self {
 
         let mut fp: String = "".to_owned();
@@ -52,7 +66,8 @@ impl configData {
             let logs: UtilsData = utils::Logs::initLog(None, format!("Config file was found : {}", fp), None);
             utils::Logs::info(logs);
             configToml {
-                database: None
+                database: None,
+                server: None,
             }
         });
 
@@ -76,12 +91,32 @@ impl configData {
             )
         };
 
-        configData {
-            host,
-            port,
-            username,
-            password,
-            database,
+        let (maxUser, timeout):
+            (u16, u32) = match configToml.server {
+            Some(server) => {
+                (
+                    server.maxUser,
+                    server.timeout
+                )
+            }
+            None => (
+                10.to_owned(),
+                10000.to_owned(),
+            )
+        };
+
+        configToml {
+            database: Some(configTomlDatabase {
+                        host,
+                        port,
+                        username,
+                        password,
+                        database,
+                        }),
+            server: Some(configTomlServer{
+                maxUser,
+                timeout
+            }),
         }
     }
 }

src/database/database.rs

@@ -1,19 +1,21 @@
 use std::panic::resume_unwind;
-use crate::config::configData;
+use crate::config::{configToml, configTomlDatabase};
 use tokio_postgres::{NoTls, Row};
 use crate::server::users::userData;
 use crate::utils;
 
+const CONFIG: configToml = configToml::new();
+
 pub async fn connectToDB() -> Result<tokio_postgres::Client, tokio_postgres::Error> {
 
-    let config: configData = configData::new();
+    let db: configTomlDatabase = CONFIG.database.unwrap();
 
     let connection_string = format!("host={} port={} user={} password={}", // dbname={}",
-                                    config.host.trim(),
-                                    config.port,
-                                    config.username.trim(),
-                                    config.password.trim()/*,
-                                    config.database.trim()*/);
+                                    db.host,
+                                    db.port,
+                                    db.username.trim(),
+                                    db.password.trim(),
+                                    );
 
     match tokio_postgres::connect(&connection_string, NoTls).await {
         Ok((client, connection)) => {
@@ -23,7 +25,7 @@ pub async fn connectToDB() -> Result<tokio_postgres::Client, tokio_postgres::Err
                     utils::Logs::error(logs);
                 }
             });
-            let logs = utils::Logs::initLog(None, format!("Connected to the database ({}:{})", config.host.trim(), config.port), None);
+            let logs = utils::Logs::initLog(None, format!("Connected to the database ({}:{})", db.host, db.port), None);
             utils::Logs::success(logs);
             Ok(client)
         }

src/todo.txt

@@ -2,6 +2,7 @@
 ========== TO DO ==========
 2. FAIRE LE SENDER POUR ENVOYER LES MERDES DANS CHECKPROTOCOL !!!!
 3. Fix le problème du nom du sender quand on receive un message
+4. Use les variables server du fichier de config
 
 ========== IN PROGRESS ==========