From 0b8ad185ab7de9be62e0a9f48137734831fc62e9 Mon Sep 17 00:00:00 2001
From: Alexey Velikiy <cg@corpglory.com>
Date: Thu, 14 Oct 2021 12:32:59 +0300
Subject: [PATCH] some commit

---
 src/grafana_service.rs | 54 ++++++++++++++++++++++++------------------
 src/main.rs            |  1 +
 2 files changed, 32 insertions(+), 23 deletions(-)

diff --git a/src/grafana_service.rs b/src/grafana_service.rs
index 9588e5b..5aaaada 100644
--- a/src/grafana_service.rs
+++ b/src/grafana_service.rs
@@ -1,4 +1,4 @@
-use std::io::BufRead;
+use std::{io::BufRead};
 
 use hyper::{Body, Client, Method, Request};
 use tokio::io::{stdout, AsyncWriteExt as _};
@@ -18,55 +18,63 @@ impl GrafanaService {
     }
 
     pub async fn test_connection(&self) -> types::Result<()> {
-        let req = Request::builder()
-            .method(Method::GET)
-            .uri(self.url.clone())
-            .header("Authorization", format!("Bearer {}", self.api_key))
-            .body(Default::default())
-            .unwrap();
+        println!("Test connection response");
+        self.get("/api").await?;
+        Ok(())
+    }
 
-        let client = Client::new();
-        let resp = client.request(req).await?;
-        println!("Response: {}", resp.status());
+    pub async fn get_datasources(&self) -> types::Result<()> {
+        self.get("/api/datasources").await?;
         Ok(())
     }
 
     pub async fn extract_metrics(&self) -> types::Result<()> {
         let req = Request::builder()
-            .method(Method::POST)
             .uri(self.url.clone() + "/api/datasources/proxy/1/api/v1/query_range")
-            .header("content-type", "application/json")
             .header("Authorization", format!("Bearer {}", self.api_key))
+            .method(Method::POST)
+            .header("content-type", "application/json")
             .body(Body::from(json!({ 
                 "query":"go_memstats_alloc_bytes_total", 
-                "start": "1634163645",
-                "end": "1634163945", 
+                "from": "1634163645",
+                "to": "1634163945", 
                 "step": "15"
             }).to_string())
             )
                 
             .unwrap();
 
+        Ok(())
+    }
+
+    async fn get(&self, suburl:&str) -> types::Result<()> {
+        let req = Request::builder()
+            .method(Method::GET)
+            .uri(self.url.to_owned() + suburl)
+            .header("Authorization", format!("Bearer {}", self.api_key))
+            .body(Body::empty())
+            .unwrap();
+
         let client = Client::new();
         let res = client.request(req).await?;
         println!("Response: {}", res.status());
-            
-        // while let Some(next) = resp.body().data().await {
-        //     let chunk = next?;
-        //     io::stdout().write_all(&chunk).await?;
-        // }
-        
+        println!("");
+
         let body = hyper::body::aggregate(res).await?;
         let mut reader = body.reader();
         let mut line = String::new();
         loop {
             match reader.read_line(&mut line) {
-                Ok(_) => println!("{}", line),
+                Ok(s) => {
+                    if s == 0 {
+                        break;
+                    }
+                    println!("{}", line);
+                    line.clear();
+                },
                 Err(_) => break
             }
         }
-        
-
         Ok(())
     }
 }
diff --git a/src/main.rs b/src/main.rs
index 5b43804..b9d19a2 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -28,6 +28,7 @@ async fn main() -> types::Result<()> {
     let gs = grafana_service::GrafanaService::new(url.to_string(), key.to_string());
 
     gs.test_connection().await?;
+    gs.get_datasources().await?;
     gs.extract_metrics().await?;
 
     Ok(())