Browse Source

cli to struct

main
Alexey Velikiy 3 years ago
parent
commit
010850408c
  1. 1
      README.md
  2. 73
      src/cli.rs
  3. 61
      src/main.rs

1
README.md

@ -5,5 +5,4 @@ subbeat
``` ```
subbeat http://localhost:3000 eyJrIjoiWnRRMTNmcGpvTHNPb3UzNzdUNUphRm53Rk9tMTNzOTQiLCJuIjoic3ViYmVhdC10ZXN0IiwiaWQiOjF9 "/api/datasources/proxy/1/api/v1/query_range" "rate(go_memstats_alloc_bytes_total[5m])" 1634672070 1634672970 subbeat http://localhost:3000 eyJrIjoiWnRRMTNmcGpvTHNPb3UzNzdUNUphRm53Rk9tMTNzOTQiLCJuIjoic3ViYmVhdC10ZXN0IiwiaWQiOjF9 "/api/datasources/proxy/1/api/v1/query_range" "rate(go_memstats_alloc_bytes_total[5m])" 1634672070 1634672970
``` ```

73
src/cli.rs

@ -0,0 +1,73 @@
use clap::{App, Arg, SubCommand};
pub struct CLI {
pub url: String,
pub key: String,
pub datasource_url: String,
pub query: String,
pub from: u64,
pub to: u64
}
impl CLI {
pub fn new() -> CLI {
let matches = App::new("subbeat")
.version("0.0.2")
.about("Timeseries toolkit")
.arg(
Arg::with_name("GRAFANA_URL")
.help("URL to your Grafana instance")
.required(true)
.index(1),
)
.arg(
Arg::with_name("GRAFANA_API_KEY")
.help("Grafna API Key. Go to http://<grafana-url>/org/apikeys to get one")
.required(true)
.index(2),
)
.arg(
Arg::with_name("datasource_url")
.help("relative path to datasource")
.required(true)
.index(3),
)
.arg(
Arg::with_name("query")
.help("your query to datasource")
.required(true)
.index(4),
)
.arg(
Arg::with_name("from")
.help("timestamp")
.required(true)
.index(5),
)
.arg(
Arg::with_name("to")
.help("timestampt")
.required(true)
.index(6),
)
.get_matches();
let url = matches.value_of("GRAFANA_URL").unwrap();
let key = matches.value_of("GRAFANA_API_KEY").unwrap();
let datasource_url = matches.value_of("datasource_url").unwrap();
let query = matches.value_of("query").unwrap();
let from = matches.value_of("from").unwrap().parse().unwrap();
let to = matches.value_of("to").unwrap().parse().unwrap();
CLI{
url: url.to_owned(),
key: key.to_owned(),
datasource_url: datasource_url.to_owned(),
query: query.to_owned(),
from,
to
}
}
}

61
src/main.rs

@ -1,70 +1,25 @@
use clap::{App, Arg, SubCommand};
use subbeat::grafana_service; use subbeat::grafana_service;
mod types; mod types;
mod cli;
#[tokio::main] #[tokio::main]
async fn main() -> types::Result<()> { async fn main() -> types::Result<()> {
let matches = App::new("subbeat")
.version("0.0.2")
.about("Timeseries toolkit")
.arg(
Arg::with_name("GRAFANA_URL")
.help("URL to your Grafana instance")
.required(true)
.index(1),
)
.arg(
Arg::with_name("GRAFANA_API_KEY")
.help("Grafna API Key. Go to http://<grafana-url>/org/apikeys to get one")
.required(true)
.index(2),
)
.arg(
Arg::with_name("datasource_url")
.help("relative path to datasource")
.required(true)
.index(3),
)
.arg(
Arg::with_name("query")
.help("your query to datasource")
.required(true)
.index(4),
)
.arg(
Arg::with_name("from")
.help("timestamp")
.required(true)
.index(5),
)
.arg(
Arg::with_name("to")
.help("timestampt")
.required(true)
.index(6),
)
.get_matches();
let url = matches.value_of("GRAFANA_URL").unwrap();
let key = matches.value_of("GRAFANA_API_KEY").unwrap();
let datasource_url = matches.value_of("datasource_url").unwrap();
let query = matches.value_of("query").unwrap();
let from = matches.value_of("from").unwrap().parse().unwrap();
let to = matches.value_of("to").unwrap().parse().unwrap();
let cli = cli::CLI::new();
let gs = grafana_service::GrafanaService::new(url.to_string(), key.to_string()); let gs = grafana_service::GrafanaService::new(cli.url.to_string(), cli.key.to_string());
// gs.test_connection().await?; // gs.test_connection().await?;
// gs.get_datasources().await?; // gs.get_datasources().await?;
// "http://localhost:3000/d/YeBxHjzWz/starter-app-stats?editPanel=2&orgId=1" // "http://localhost:3000/d/YeBxHjzWz/starter-app-stats?editPanel=2&orgId=1"
let r = gs let r = gs
.extract_metrics( .extract_metrics(
datasource_url, &cli.datasource_url,
query, &cli.query,
from, cli.from,
to, cli.to,
15, 15,
) )
.await?; .await?;

Loading…
Cancel
Save