valid (nearly) db export and so on

.env

@@ -1,3 +1,8 @@
 CONFIG_SERVER_CREDS = "ws://127.0.0.1:8080"
 API_GRUBBER_SOCKET = "api-grub.sock"
 PREPROC_SOCKET = "preproc.sock"
+
+DB_HOST = "192.168.2.37"
+DB_USER = "vlad"
+DB_PASSWORD = "vlad"
+DB_DBNAME = "vks"

config_api.json

@@ -1,13 +1,31 @@
 { 
-    "endpoints" : [
+	"id" : 1 ,
-        {
+	"template" : 	
-            "url" : "http://127.0.0.1:8081/ping",
+		[{
-            "method" : "GET"
+			"id" :"mock_api_1", 
+			"name" : "Mock / ",
+			"url" : "http://127.0.0.1:8081/",
+			"method" : "GET",
+			"measure" : 
+				[
+					"operation", "response"
+				]
 		},
 		{
-            "url" : "http://127.0.0.1:8081/",
+			"id" :"mock_api_2", 
-            "method" : "GET"
+			"name" : "Mock /ping ",
+			"url" : "http://127.0.0.1:8081/ping",
+			"method" : "GET",
+			"measure" : 
+				[
+					"operation", "response", "empty_field"
+				]
 		}
 		],
-    "delay" : 5
+	"ip_address" : 	"127.0.0.1:8081",
+	"login" : "",
+	"pass" : "" ,
+	"api_key" : 	"908c709827bd40n98r7209837x98273",
+	"period" : 		10,
+	"timeout" : 10
 }

crates/api-grub/Cargo.toml

@@ -13,3 +13,6 @@ log = "0.4.25"
 anyhow = "1.0.95"
 chrono = "0.4.39"
 reqwest = { version = "0.12.12", features = ["rustls-tls", "json"] }
+deadpool-postgres = "0.14.1"
+tokio-postgres = "0.7.12"
+dotenv = "0.15.0"

crates/api-grub/src/config.rs

@@ -1,7 +1,7 @@
 //  mod to communicate with api-grub config file
 //  1) check changes in unix-socket  
 //  2) save changes in local config file
-use integr_structs::api::ApiConfig;
+use integr_structs::api::{ApiConfig, ApiConfigV2};
 use anyhow::{Error, Ok, Result};
 use log::{info, warn, error}; 
 use std::{fs, path::Path};
@@ -15,7 +15,7 @@ const CONFIG_PATH: &str = "config_api.json";
 const SOCKET_PATH: &str = "api-grub.sock";
 
 // todo! rewrite to use current_exe
-pub async fn pull_local_config() -> Result<ApiConfig> {
+pub async fn pull_local_config() -> Result<ApiConfigV2> {
     // let conf_path = std::env::current_exe()?;
     let path = Path::new(CONFIG_PATH);
     // return match conf_path.parent() {
@@ -28,7 +28,7 @@ pub async fn pull_local_config() -> Result<ApiConfig> {
     //     None => Err(Error::msg("No local conf was found")) 
     // }
     if path.exists() && path.is_file() {
-        let config: ApiConfig = from_str(
+        let config: ApiConfigV2 = from_str(
             &fs::read_to_string(CONFIG_PATH)?
         )?;
         Ok(config)
@@ -39,7 +39,7 @@ pub async fn pull_local_config() -> Result<ApiConfig> {
 
 // for config pulling
 // ++++ reader to channel 
-pub async fn init_config_grub_mechanism(tx: &Sender<ApiConfig>) -> Result<()> {
+pub async fn init_config_grub_mechanism(tx: &Sender<ApiConfigV2>) -> Result<()> {
     info!("Initializing Unix-Socket listening for pulling new configs...");
     let server = init_unix_listener().await?;
     // 
@@ -51,7 +51,7 @@ pub async fn init_config_grub_mechanism(tx: &Sender<ApiConfig>) -> Result<()> {
             if let Err(er) =  stream.read_to_string(&mut buffer).await {
                 warn!("Cannot read config from stream due to {}", er);
             } else {
-                let config: Result<ApiConfig, serde_json::Error> = from_str(&buffer);
+                let config: Result<ApiConfigV2, serde_json::Error> = from_str(&buffer);
                 if let stdOk(conf) = config {
                     info!("New config was pulled from Unix-Stream. Saving it locally and sharing with API-grub module...");
                     if let Err(er) = save_new_config(&buffer).await {
@@ -97,13 +97,13 @@ mod config_unittests {
     #[test]
     async fn check_save_new_config() {
         use std::fs;
-        use integr_structs::api::ApiConfig;
+        use integr_structs::api::ApiConfigV2;
         use serde_json::to_string;
         
         let test_config_path  = "test_config_api.json";
 
         // config gen  
-        let config = to_string::<ApiConfig>(&ApiConfig::default());
+        let config = to_string::<ApiConfigV2>(&ApiConfigV2::default());
         assert!(config.is_ok());
         let config = config.unwrap();
 

crates/api-grub/src/main.rs

@@ -3,7 +3,7 @@ mod net;
 mod logger;
 
 use anyhow::Result;
-use integr_structs::api::ApiConfig;
+use integr_structs::api::ApiConfigV2;
 use logger::setup_logger;
 use log::{info, warn};
 use config::{pull_local_config, init_config_grub_mechanism};
@@ -19,7 +19,7 @@ async fn main() -> Result<()>{
     setup_logger().await?;
     let config = get_config().await;
     // config update channel
-    let (tx, mut rx) = mpsc::channel::<ApiConfig>(1);
+    let (tx, mut rx) = mpsc::channel::<ApiConfigV2>(1);
     // futures
     // todo : rewrite with spawn
     let config_fut = init_config_grub_mechanism(&tx);
@@ -30,7 +30,7 @@ async fn main() -> Result<()>{
     Ok(())
 }
 
-async fn get_config() -> ApiConfig {
+async fn get_config() -> ApiConfigV2 {
     return match pull_local_config().await {
         Ok(conf) => {
             info!("Local config was loaded");
@@ -38,7 +38,7 @@ async fn get_config() -> ApiConfig {
         },
         Err(er) => {
             warn!("Cannot get local config due to {}", er);
-            ApiConfig::default()
+            ApiConfigV2::default()
         }
     }
 }

crates/api-grub/src/net.rs

@@ -1,10 +1,70 @@
 // module to handle unix-socket connection + pulling info from api 
 use anyhow::{Error, Result};
-use integr_structs::api::ApiConfig;
+use integr_structs::api::{ApiConfigV2, ProcessedEndpoint};
 use log::{error, info};
+use serde_json::Value;
 use tokio::sync::mpsc::Receiver;
 use tokio::time::{sleep, Duration};
 use reqwest::{Client, Method};
+use std::sync::Arc;
+use tokio::task::JoinHandle;
+// use tokio::sync::Mutex;
+use tokio_postgres::types::ToSql;
+use deadpool_postgres::{Config, Pool, Runtime, Client as PgClient};
+use tokio_postgres::NoTls;
+use dotenv::dotenv;
+use std::env;
+
+// type BufferType = Arc<Mutex<Vec<String>>>; 
+
+struct Exporter {
+    pool : Option<Pool>,
+}
+
+impl Exporter {
+    fn config_construct() -> Result<Config> {
+        let mut cfg = Config::new();
+        cfg.host = Some(env::var("DB_HOST")?);
+        cfg.dbname = Some(env::var("DB_DBNAME")?);
+        cfg.user = Some(env::var("DB_USER")?);
+        cfg.password = Some(env::var("DB_PASSWORD")?);
+        Ok(cfg)
+    }
+    fn pool_construct() -> Option<Pool> {
+        return match Self::config_construct() {
+            Ok(config) => {
+                if let Ok(pool) = config.create_pool(Some(Runtime::Tokio1), NoTls) {
+                    info!("Connected to PostgreSQL");
+                    return Some(pool);
+                }
+                None
+            },
+            Err(_) => {
+                error!("Bad DB credentials or it's unreachable");
+                None
+            },
+        } 
+    } 
+    pub fn is_no_connection(&self) -> bool { self.pool.is_none() }
+    pub fn init() -> Self {
+        Self {
+            pool : Self::pool_construct()
+        }
+    }
+    pub async fn get_connection_from_pool(&self) -> Option<PgClient> {
+        if let Some(pool) = &self.pool {
+            return Some(pool.get().await.ok()?);
+        }
+        None
+    } 
+    pub async fn export_data(client: PgClient, metrics: &str) -> Result<()> {
+        // client.
+        let query = "INSERT INTO metrics (body) VALUES ($1);"; 
+        let _ = client.execute(query, &[&query]).await?;
+        Ok(())
+    }
+
+}
 
 struct RestMethod;
 
@@ -15,69 +75,113 @@ impl RestMethod {
             "patch" => Method::PATCH,
             "put" => Method::PUT,
             "delete" => Method::DELETE,
+            "head" => Method::HEAD,
+            "trace" => Method::TRACE,
+            "options" => Method::OPTIONS,
+            "connect" => Method::CONNECT,
             "get" | _ => Method::GET
         }
     } 
 }
 struct ApiPoll<'a> {
-    config : &'a mut ApiConfig,
+    config : &'a mut ApiConfigV2,
     client : Client,
 }
 
 impl<'a> ApiPoll<'a> {
-    pub async fn new(poll_cfg : &'a mut ApiConfig) -> Self {
+    pub async fn new(poll_cfg : &'a mut ApiConfigV2) -> Self {
         Self {
             config : poll_cfg,
             client : Client::new(),
         }
     }
     // can be weak and with bug test needed
-    pub async fn change_config(&mut self, conf: ApiConfig) {
+    pub async fn change_config(&mut self, conf: ApiConfigV2) {
         *self.config = conf;
     }
     pub async fn is_default(&self) -> bool {
-        self.config.endpoints.len() == 0
+        self.config.template.len() == 0
     }
-    pub async fn process_polling(&self) -> Result<Vec<String>> {
+    pub async fn process_polling(&self, exporter: Arc<Exporter>) -> Result<()> {
-        let mut buffer: Vec<String> = vec![];
+        // let buffer: BufferType = Arc::new(Mutex::new(vec![]));
+        let mut join_handles: Vec<JoinHandle<Result<()>>> = vec![];
+        let client = Arc::new(self.client.clone());
+        let template = Arc::new(self.config.template.clone());
+
         // TODO: rewrite nextly to async 
-        for point in &self.config.endpoints {
+        for point in template.iter() {
-            // let a = self.client.get(&point.url).send().await.unwrap();
+            let point = Arc::new(point.clone());
-            // a.text().await.unwrap();
+            // let buffer = buffer.clone();
-            match self.client.request(RestMethod::from_str(&point.method).await, &point.url).send().await {
+            let client = client.clone();
+            let exporter = exporter.clone();
+            let endpoint_processer = tokio::spawn(async move {
+                let point = point.clone();
+                match client.request(RestMethod::from_str(&point.method).await, &point.url).send().await {
                     Ok(resp) => {
                         if !resp.status().is_success() {
                             error!("ErrorCode in Response from API. Check configuration");
-                        continue;
+                            return Err(Error::msg("Error during sending request"));
                         }
                         if let Ok(text) = resp.text().await {
-                        info!("{}: {} - Successfull grubbing info", &point.method.to_uppercase(), &point.url);
+                            // 
-                        buffer.push(text);
+                            let metrics = ProcessedEndpoint::from_target_response(&text, &point)?;
+                            // 
+                            if let Some(conn) = exporter.get_connection_from_pool().await {
+                                if let Err(er) =  Exporter::export_data(conn, &metrics).await {
+                                    error!("Cannot export data to DB during to: {}", er);
+                                    return Err(Error::msg("Error during exporting data to DB"));
+                                }
+                            } else {
+                                if !exporter.is_no_connection() {
+                                    return Err(Error::msg("Error during getting connection from pool"));
+                                }
+                            }
+
+                            // let mut buffer = buffer.lock().await;
+                            // buffer.push(text);
                         } else {
                             error!("{}: {} - Error with extracting text field from Response", &point.method.to_uppercase(), &point.url);
                         }
                     },
-                Err(er) => {
+                    Err(_) => {
-                    error!("{}: {} - Query crushed due to {}", &point.method.to_uppercase(), &point.url, er);
+                        error!("{}: {} endpoint is unreachable", &point.method.to_uppercase(), &point.url);
                     },
                 }
+                Ok(())
+            });
+            join_handles.push(endpoint_processer);
         }
-        match &buffer.len() {
+
-            0 => Err(Error::msg("Error due to API grubbing. Check config" )),
+        for i in join_handles {
-            _ => Ok(buffer),
+            let _ = i.await;
         }
+
+        // let buffer = buffer.lock().await;
+        // match &buffer.len() {
+        //     0 => Err(Error::msg("Error due to API grubbing. Check config" )),
+        //     _ => {
+        //         Ok(())
+        //     },
+        // }
+        Ok(())
     }
     pub async fn get_delay(&self) -> u32 {
-        self.config.delay
+        self.config.timeout
     }
 }
 
 // for api info pulling 
-pub async fn init_api_grub_mechanism(config: ApiConfig, rx: &mut Receiver<ApiConfig>) -> Result<()> {
+pub async fn init_api_grub_mechanism(config: ApiConfigV2, rx: &mut Receiver<ApiConfigV2>) -> Result<()> {
     info!("Initializing API-info grubbing mechanism...");
+    info!("Loading vars from .env file if exists...");
+    let _ = dotenv().ok();
+
     let mut config = config;
     let mut poller = ApiPoll::new(&mut config).await;
+    let client = Exporter::init(); 
+    let shared_pool = Arc::new(client);
     loop {
+        let shared_pool = shared_pool.clone();
         if poller.is_default().await {
             sleep(Duration::from_secs(5)).await;
         } else {
@@ -87,7 +191,7 @@ pub async fn init_api_grub_mechanism(config: ApiConfig, rx: &mut Receiver<ApiCon
                     info!("Config changed");
                 }
             }
-            info!("Data from API: {:?}", poller.process_polling().await);
+            info!("Data from API: {:?}", poller.process_polling(shared_pool).await);
             sleep(Duration::from_secs(poller.get_delay().await as u64)).await;
         }
     }
@@ -98,7 +202,6 @@ pub async fn init_api_grub_mechanism(config: ApiConfig, rx: &mut Receiver<ApiCon
 mod net_unittests {
     use super::*;
     use tokio::test;
-    use integr_structs::api::{ApiConfig, ApiEndpoint};
     
     #[test]
     async fn check_str_to_rest_method() {
@@ -111,16 +214,16 @@ mod net_unittests {
     } 
     #[test]
     async fn check_api_poll_change_config() {
-        let mut conf1 = ApiConfig::default();
+        let mut conf1 = ApiConfigV2::default();
-        let conf2 = ApiConfig { endpoints : vec![], delay : 10, };
+        let conf2 = ApiConfigV2::pattern();
         let mut poll = ApiPoll::new(&mut conf1).await;
         poll.change_config(conf2).await;
-        assert_eq!(poll.config.delay, 10)
+        assert_eq!(poll.config.timeout, 1)
     }
 
     #[test]
     async fn check_api_poll_is_default() {
-        let mut conf1 = ApiConfig::default();
+        let mut conf1 = ApiConfigV2::default();
         let poll = ApiPoll::new(&mut conf1).await;
         assert!(poll.is_default().await)
     }
@@ -130,20 +233,14 @@ mod net_unittests {
         use log::{set_max_level, LevelFilter};
 
         set_max_level(LevelFilter::Off);
-        let mut conf1 = ApiConfig { 
+        let mut conf1 = ApiConfigV2::pattern();
-            endpoints : vec![ 
+        let conf2 = ApiConfigV2::default(); 
-                ApiEndpoint { 
+        let exporter = Arc::new(Exporter::init());
-                    url : String::from("https://dummy-json.mock.beeceptor.com/countries"), 
-                    method: String::from("get"),
-            }], 
-            delay : 10, 
-        };
-        let conf2 = ApiConfig::default(); 
 
         let mut poll = ApiPoll::new(&mut conf1).await;
-        assert!(poll.process_polling().await.is_ok());
+        assert!(poll.process_polling(exporter.clone()).await.is_ok());
 
         poll.change_config(conf2).await;
-        assert!(poll.process_polling().await.is_err());
+        assert!(poll.process_polling(exporter.clone()).await.is_err());
     }
 }

crates/config-delivery/src/delivery.rs

@@ -2,7 +2,7 @@
 // using Unix-Socket Client
 
 use anyhow::{Error, Result};
-use integr_structs::api::ApiConfig;
+use integr_structs::api::ApiConfigV2;
 use tokio::time::{sleep, Duration};
 use tokio::net::UnixStream;
 use std::env;
@@ -18,7 +18,7 @@ enum UnixSocketConsumer {
 // to create us-client
 struct UnixSocketClient;  
 // to catch new configs from integr submod
-type ConfigGateway = Receiver<ApiConfig>;
+type ConfigGateway = Receiver<ApiConfigV2>;
 
 pub async fn init_delivery_mech(rx: ConfigGateway) -> Result<()> {
     let mut rx = rx; 

crates/config-delivery/src/integration.rs

@@ -1,18 +1,18 @@
 use std::sync::Arc;
 
 use tokio::sync::mpsc::Sender;
-use integr_structs::api::ApiConfig;
+use integr_structs::api::ApiConfigV2;
 use anyhow::Result;
 use tokio::time::{sleep, Duration};
 // mock 
 
-type Worker = Arc<Sender<ApiConfig>>;
+type Worker = Arc<Sender<ApiConfigV2>>;
 
 #[allow(dead_code, unused_variables)]
 pub async fn init_integration_mech(tx: Worker) -> Result<()> {
     loop {
         sleep(Duration::from_secs(5)).await;
-        let conf = ApiConfig::default();
+        let conf = ApiConfigV2::default();
         tx.send(conf).await?;
     }
     // Ok(())

crates/config-delivery/src/main.rs

@@ -2,7 +2,7 @@ mod delivery;
 mod integration;
 mod logger;
 
-use integr_structs::api::ApiConfig;
+use integr_structs::api::ApiConfigV2;
 use logger::setup_logger;
 use dotenv::dotenv;
 use anyhow::Result;
@@ -23,7 +23,7 @@ async fn main() -> Result<()> {
     info!("Pulling env vars from .env file if exists ...");
     dotenv().ok();
 
-    let (tx, rx) = mpsc::channel::<ApiConfig>(1);
+    let (tx, rx) = mpsc::channel::<ApiConfigV2>(1);
     let tx = Arc::new(tx);
     let integr_jh = tokio::spawn(
         async move {

crates/integr-structs/Cargo.toml

@@ -4,5 +4,6 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+anyhow = "1.0.95"
 serde = { version = "1.0.217", features = ["derive"] }
 serde_json = "1.0.135"

crates/integr-structs/src/api.rs

@@ -1,7 +1,7 @@
 use std::collections::HashMap;
-
 use serde::{Serialize, Deserialize};
-use serde_json::Value;
+use serde_json::{ to_string_pretty, Value };
+use anyhow::Result;
 
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -29,24 +29,87 @@ impl Default for ApiConfig {
 // v2
 #[derive(Serialize, Deserialize, Debug)]
 pub struct ApiConfigV2 {
-    id : u64,
+    pub id : u64,
     #[serde(default)]
-    template : Vec<Template>,
+    pub template : Vec<Template>,
-    ip_address : String,
+    pub ip_address : String,
-    login : Option<String>,
+    pub login : Option<String>,
-    pass : Option<String>,
+    pub pass : Option<String>,
-    api_key : Option<String>,
+    pub api_key : Option<String>,
-    period : u32,               // if "0" -> inf
+    pub period : u32,               // if "0" -> inf
-    timeout :  u32,             // if "0" -> no-delay
+    pub timeout :  u32,             // if "0" -> no-delay
 }
 
-#[derive(Serialize, Deserialize, Debug)]
+impl Default for ApiConfigV2 {
+    fn default() -> Self {
+        ApiConfigV2 {
+            id : 0,
+            template : Vec::new(),
+            ip_address : String::from("no_ip"),
+            login : None,
+            pass : None,
+            api_key : None,
+            period : 0,               
+            timeout :  0,  
+        }
+    }
+}
+impl ApiConfigV2 {
+    pub fn template() -> Self {
+        ApiConfigV2 {
+            id : 1111,
+            template : Vec::new(),
+            ip_address : String::from("ip"),
+            login : None,
+            pass : None,
+            api_key : None,
+            period : 1111,               
+            timeout :  1111,  
+        }
+    }
+    pub fn pattern() -> Self {
+        ApiConfigV2 {
+            id : 1111,
+            template : vec![
+                Template {
+                    id : String::from("no id"),
+                    name : String::from("open api"),
+                    url : String::from("https://dummy-json.mock.beeceptor.com/countries"),
+                    method : String::from("GET"),
+                    measure : Vec::new(),
+                }
+            ],
+            ip_address : String::from("ip"),
+            login : None,
+            pass : None,
+            api_key : None,
+            period : 1,               
+            timeout :  1,  
+        }
+    }
+}
+
+
+#[derive(Serialize, Deserialize, Debug, Clone)]
 pub struct Template {
-    id : String,
+    pub id : String,
-    name : String,
+    pub name : String,
-    url : String,
+    pub url : String,
+    pub method : String,
     #[serde(default)]
-    measure : Vec<String>,
+    pub measure : Vec<String>,
+}
+
+impl Default for Template {
+    fn default() -> Self {
+        Template {
+            id : String::from("no-id"),
+            name : String::from("no-name"),
+            url : String::from("no-url"),
+            method : String::from("no-method"),
+            measure : Vec::new(),
+        }
+    }
 }
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -54,6 +117,31 @@ pub struct ProcessedEndpoint {
     id : String,
     name : String,
     url : String,
+    method : String,
     #[serde(default)]
-    metrics : HashMap<String, Value>
+    metrics : HashMap<String, Value>,
+}
+
+impl ProcessedEndpoint {
+    pub fn new(id: &str, name: &str, url: &str, method: &str, metrics: HashMap<String, Value>) -> Self {
+        ProcessedEndpoint {
+            id : id.to_owned(),
+            name : name.to_owned(),
+            url : url.to_owned(),
+            method : method.to_owned(), 
+            metrics : metrics,
+        }
+    }
+    pub fn from_target_response(response: &str, keys: &Template) -> Result<String> {
+        let mut hm: HashMap<String, Value> = HashMap::new();
+        let mut response: Value = serde_json::from_str(response)?;
+
+        let _ = keys.measure.iter()
+            .map(|key| (key, response[key].take()))
+            .for_each(|(key, value)| {
+                hm.insert(key.clone(), value);
+            });
+        let val = ProcessedEndpoint::new(&keys.id, &keys.name, &keys.url, &keys.method,hm);
+        Ok(to_string_pretty(&val)?)
+    }
 }

template_global_config.json

@@ -2,20 +2,30 @@
 	"id" : 1 ,
 	"template" : 	
 		[{
-			"id" :"conference", 
+			"id" :"mock_api_1", 
-			"name" : "Conference",
+			"name" : "Mock / ",
-			"url" : "https://$ip_address/api/v1/conferences",
+			"url" : "http://127.0.0.1:8081/",
+			"method" : "GET",
 			"measure" : 
 				[
-					"total", "number", "description", 
+					"operation", "response", "empty_field"
-					"shutdown", "startTime", "stopTime",
-					"participants_total", "participants_online"
 				]
-		}],
+		},
+		{
+			"id" :"mock_api_2", 
+			"name" : "Mock /ping ",
+			"url" : "http://127.0.0.1:8081/ping",
+			"method" : "GET",
+			"measure" : 
+				[
+					"operation", "response", "empty_field"
+				]
+		}
+		],
 	"ip_address" : 	"127.0.0.1:8081",
 	"login" : "",
 	"pass" : "" ,
 	"api_key" : 	"908c709827bd40n98r7209837x98273",
-	"period" : 		"10",
+	"period" : 		10,
-	"timeout" : "2"
+	"timeout" : 2
 }