added async to zabbixlaunch

This commit is contained in:
Paul 2022-01-16 21:18:25 +01:00
parent 43cac4a821
commit 2dd5020963
8 changed files with 378 additions and 102 deletions

101
.drone.yml Normal file
View File

@ -0,0 +1,101 @@
---
kind: pipeline
type: docker
name: cleanup-before
steps:
- name: clean
image: alpine
commands:
- rm -rf /build/*
volumes:
- name: build
path: /build
when:
event: tag
volumes:
- name: build
host:
path: /tmp/ipbl/build
---
kind: pipeline
type: docker
name: default-linux-amd64
steps:
- name: Build
image: rust:latest
commands:
- cargo build --verbose --all
---
kind: pipeline
type: docker
name: gitea-release
steps:
- name: move
image: alpine
commands:
- mv target/release/* ./
volumes:
- name: build
path: /drone/src/build
when:
event: tag
- name: release
image: plugins/gitea-release
settings:
base_url: https://git.paulbsd.com
api_key:
from_secret: gitea_token
files: "*.tar.gz"
checksum:
- sha256
- sha512
title: VERSION
volumes:
- name: build
path: /drone/src/build
when:
event: tag
- name: ls
image: alpine
commands:
- find .
volumes:
- name: build
path: /drone/src/build
when:
event: tag
volumes:
- name: build
host:
path: /tmp/ipbl/build
depends_on:
- default-linux-amd64
---
kind: pipeline
type: docker
name: cleanup-after
steps:
- name: clean
image: alpine
commands:
- rm -rf /build/*
volumes:
- name: build
path: /build
when:
event: tag
volumes:
- name: build
host:
path: /tmp/ipbl/build

142
Cargo.lock generated
View File

@ -248,6 +248,21 @@ dependencies = [
"percent-encoding", "percent-encoding",
] ]
[[package]]
name = "futures"
version = "0.3.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a12aa0eb539080d55c3f2d45a67c3b58b6b0773c1a3ca2dfec66d58c97fd66ca"
dependencies = [
"futures-channel",
"futures-core",
"futures-executor",
"futures-io",
"futures-sink",
"futures-task",
"futures-util",
]
[[package]] [[package]]
name = "futures-channel" name = "futures-channel"
version = "0.3.17" version = "0.3.17"
@ -255,6 +270,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5da6ba8c3bb3c165d3c7319fc1cc8304facf1fb8db99c5de877183c08a273888" checksum = "5da6ba8c3bb3c165d3c7319fc1cc8304facf1fb8db99c5de877183c08a273888"
dependencies = [ dependencies = [
"futures-core", "futures-core",
"futures-sink",
] ]
[[package]] [[package]]
@ -263,12 +279,36 @@ version = "0.3.17"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "88d1c26957f23603395cd326b0ffe64124b818f4449552f960d815cfba83a53d" checksum = "88d1c26957f23603395cd326b0ffe64124b818f4449552f960d815cfba83a53d"
[[package]]
name = "futures-executor"
version = "0.3.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "45025be030969d763025784f7f355043dc6bc74093e4ecc5000ca4dc50d8745c"
dependencies = [
"futures-core",
"futures-task",
"futures-util",
]
[[package]] [[package]]
name = "futures-io" name = "futures-io"
version = "0.3.17" version = "0.3.17"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "522de2a0fe3e380f1bc577ba0474108faf3f6b18321dbf60b3b9c39a75073377" checksum = "522de2a0fe3e380f1bc577ba0474108faf3f6b18321dbf60b3b9c39a75073377"
[[package]]
name = "futures-macro"
version = "0.3.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "18e4a4b95cea4b4ccbcf1c5675ca7c4ee4e9e75eb79944d07defde18068f79bb"
dependencies = [
"autocfg",
"proc-macro-hack",
"proc-macro2",
"quote",
"syn",
]
[[package]] [[package]]
name = "futures-sink" name = "futures-sink"
version = "0.3.17" version = "0.3.17"
@ -288,12 +328,17 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "36568465210a3a6ee45e1f165136d68671471a501e632e9a98d96872222b5481" checksum = "36568465210a3a6ee45e1f165136d68671471a501e632e9a98d96872222b5481"
dependencies = [ dependencies = [
"autocfg", "autocfg",
"futures-channel",
"futures-core", "futures-core",
"futures-io", "futures-io",
"futures-macro",
"futures-sink",
"futures-task", "futures-task",
"memchr", "memchr",
"pin-project-lite", "pin-project-lite",
"pin-utils", "pin-utils",
"proc-macro-hack",
"proc-macro-nested",
"slab", "slab",
] ]
@ -434,6 +479,15 @@ dependencies = [
"hashbrown", "hashbrown",
] ]
[[package]]
name = "instant"
version = "0.1.12"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7a5bbe824c507c5da5956355e86a746d82e0e1464f65d862cc5e71da70e94b2c"
dependencies = [
"cfg-if 1.0.0",
]
[[package]] [[package]]
name = "ipnet" name = "ipnet"
version = "2.3.1" version = "2.3.1"
@ -475,6 +529,15 @@ version = "0.2.103"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "dd8f7255a17a627354f321ef0055d63b898c6fb27eff628af4d1b66b7331edf6" checksum = "dd8f7255a17a627354f321ef0055d63b898c6fb27eff628af4d1b66b7331edf6"
[[package]]
name = "lock_api"
version = "0.4.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "712a4d093c9976e24e7dbca41db895dabcbac38eb5f4045393d17a95bdfb1109"
dependencies = [
"scopeguard",
]
[[package]] [[package]]
name = "log" name = "log"
version = "0.4.14" version = "0.4.14"
@ -627,9 +690,9 @@ dependencies = [
[[package]] [[package]]
name = "num_cpus" name = "num_cpus"
version = "1.13.0" version = "1.13.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "05499f3756671c15885fee9034446956fff3f243d6077b91e5767df161f766b3" checksum = "19e64526ebdee182341572e50e9ad03965aa510cd94427a4549448f285e957a1"
dependencies = [ dependencies = [
"hermit-abi", "hermit-abi",
"libc", "libc",
@ -674,6 +737,31 @@ dependencies = [
"vcpkg", "vcpkg",
] ]
[[package]]
name = "parking_lot"
version = "0.11.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7d17b78036a60663b797adeaee46f5c9dfebb86948d1255007a1d6be0271ff99"
dependencies = [
"instant",
"lock_api",
"parking_lot_core",
]
[[package]]
name = "parking_lot_core"
version = "0.8.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d76e8e1493bcac0d2766c42737f34458f1c8c50c0d23bcb24ea953affb273216"
dependencies = [
"cfg-if 1.0.0",
"instant",
"libc",
"redox_syscall",
"smallvec",
"winapi",
]
[[package]] [[package]]
name = "percent-encoding" name = "percent-encoding"
version = "2.1.0" version = "2.1.0"
@ -704,6 +792,18 @@ version = "0.2.10"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ac74c624d6b2d21f425f752262f42188365d7b8ff1aff74c82e45136510a4857" checksum = "ac74c624d6b2d21f425f752262f42188365d7b8ff1aff74c82e45136510a4857"
[[package]]
name = "proc-macro-hack"
version = "0.5.19"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "dbf0c48bc1d91375ae5c3cd81e3722dff1abcf81a30960240640d223f59fe0e5"
[[package]]
name = "proc-macro-nested"
version = "0.1.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bc881b2c22681370c6a780e47af9840ef841837bc98118431d4e1868bd0c1086"
[[package]] [[package]]
name = "proc-macro2" name = "proc-macro2"
version = "1.0.29" version = "1.0.29"
@ -831,6 +931,12 @@ dependencies = [
"winapi", "winapi",
] ]
[[package]]
name = "scopeguard"
version = "1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d29ab0c6d3fc0ee92fe66e2d99f700eab17a8d57d1c1d3b748380fb20baa78cd"
[[package]] [[package]]
name = "security-framework" name = "security-framework"
version = "2.3.1" version = "2.3.1"
@ -897,12 +1003,27 @@ dependencies = [
"serde", "serde",
] ]
[[package]]
name = "signal-hook-registry"
version = "1.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e51e73328dc4ac0c7ccbda3a494dfa03df1de2f46018127f60c693f2648455b0"
dependencies = [
"libc",
]
[[package]] [[package]]
name = "slab" name = "slab"
version = "0.4.4" version = "0.4.4"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c307a32c1c5c437f38c7fd45d753050587732ba8628319fbdf12a7e289ccc590" checksum = "c307a32c1c5c437f38c7fd45d753050587732ba8628319fbdf12a7e289ccc590"
[[package]]
name = "smallvec"
version = "1.8.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f2dd574626839106c320a323308629dcb1acfc96e32a8cba364ddc61ac23ee83"
[[package]] [[package]]
name = "socket2" name = "socket2"
version = "0.4.1" version = "0.4.1"
@ -991,10 +1112,25 @@ dependencies = [
"memchr", "memchr",
"mio", "mio",
"num_cpus", "num_cpus",
"once_cell",
"parking_lot",
"pin-project-lite", "pin-project-lite",
"signal-hook-registry",
"tokio-macros",
"winapi", "winapi",
] ]
[[package]]
name = "tokio-macros"
version = "1.7.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b557f72f448c511a979e2564e55d74e6c4432fc96ff4f6241bc6bded342643b7"
dependencies = [
"proc-macro2",
"quote",
"syn",
]
[[package]] [[package]]
name = "tokio-native-tls" name = "tokio-native-tls"
version = "0.3.0" version = "0.3.0"
@ -1243,9 +1379,11 @@ version = "1.0.0"
dependencies = [ dependencies = [
"clap", "clap",
"embedded-graphics", "embedded-graphics",
"futures",
"launchy", "launchy",
"nix 0.23.0", "nix 0.23.0",
"reqwest", "reqwest",
"serde", "serde",
"serde_json", "serde_json",
"tokio",
] ]

View File

@ -6,10 +6,12 @@ edition = "2018"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies] [dependencies]
clap = { version = "2.33.3", features = ["yaml"] } clap = { version = "2", features = ["yaml"] }
embedded-graphics = { version = "0.7.1", optional = true } embedded-graphics = { version = "0.7", optional = true }
futures = "*"
launchy = { git = "https://github.com/kangalioo/launchy", branch = "master" } launchy = { git = "https://github.com/kangalioo/launchy", branch = "master" }
reqwest = { version = "0.11.5", features = ["blocking", "json"] } nix = "0.23"
serde = { version = "1.0.130", features = ["derive"] } reqwest = { version = "0.11", features = ["json"] }
serde_json = "1.0.68" serde = { version = "1.0", features = ["derive"] }
nix = "0.23.0" serde_json = "1.0"
tokio = { version = "1", features = ["full"] }

View File

@ -1,7 +1,7 @@
use crate::zabbix::api::get_zabbix_authtoken; use crate::zabbix::api::get_zabbix_authtoken;
use crate::zabbix::problems::DataMatrix; use crate::zabbix::problems::DataMatrix;
use clap::App; use clap::App;
use nix::sys::inotify::{AddWatchFlags, InitFlags, Inotify}; use nix::sys::inotify::{AddWatchFlags, InitFlags, Inotify, InotifyEvent};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use serde_json::Error as JsonError; use serde_json::Error as JsonError;
use serde_json::Value as JsonValue; use serde_json::Value as JsonValue;
@ -28,7 +28,7 @@ pub struct Context {
} }
impl Context { impl Context {
pub fn new() -> Self { pub async fn new() -> Self {
let yaml = load_yaml!("cli.yml"); let yaml = load_yaml!("cli.yml");
let cli = App::from_yaml(yaml).get_matches(); let cli = App::from_yaml(yaml).get_matches();
@ -43,7 +43,7 @@ impl Context {
}; };
println!("Loading {configfile} file ...", configfile = ctx.configfile); println!("Loading {configfile} file ...", configfile = ctx.configfile);
ctx.cfg.load(&ctx.configfile); ctx.cfg.load(&ctx.configfile).await;
println!( println!(
"Adding inotify watch on {configfile} file ...", "Adding inotify watch on {configfile} file ...",
@ -52,30 +52,21 @@ impl Context {
ctx.inotify ctx.inotify
.add_watch(ctx.configfile.as_str(), AddWatchFlags::IN_MODIFY) .add_watch(ctx.configfile.as_str(), AddWatchFlags::IN_MODIFY)
.unwrap(); .unwrap();
get_zabbix_authtoken(&mut ctx.cfg); let tokenrefresh = get_zabbix_authtoken(&mut ctx.cfg);
tokenrefresh.await;
ctx ctx
} }
pub fn hotreload(&mut self) {
let mut i = 0; pub async fn hotreload(&mut self) {
let waitmilli = self.cfg.refresh.unwrap_or(5) * 1000; let events = match self.inotify.read_events() {
while i < waitmilli { Ok(ev) => ev,
let waitinc = waitmilli / ReloadFrequency::Medium as u64; Err(_) => vec![],
let events = match self.inotify.read_events() { };
Ok(ev) => ev, if events.len() > 0 {
Err(_) => vec![], println!("Reloading {cfg}", cfg = self.configfile);
}; get_zabbix_authtoken(&mut self.cfg).await;
if events.len() > 0 { self.cfg.load(self.configfile.as_str());
self.cfg.load(self.configfile.as_str());
println!(
"Reloading {cfg}",
cfg = self.configfile
);
break;
} else {
sleep(Duration::from_millis(waitinc));
}
i += waitinc;
} }
} }
} }
@ -104,7 +95,7 @@ impl<'a> Config {
} }
} }
pub fn load(&'a mut self, configfile: &str) { async fn load(&'a mut self, configfile: &str) {
let fileopen: Result<File, std::io::Error>; let fileopen: Result<File, std::io::Error>;
let filemeta = std::fs::metadata(configfile); let filemeta = std::fs::metadata(configfile);
let fileexists = match filemeta { let fileexists = match filemeta {
@ -130,6 +121,7 @@ impl<'a> Config {
Ok(cfg) => { Ok(cfg) => {
let tmpcfg = Config::new(); let tmpcfg = Config::new();
Config { Config {
<<<<<<< Updated upstream
server: cfg["server"].as_str().unwrap_or(tmpcfg.server.as_str()).to_string(), server: cfg["server"].as_str().unwrap_or(tmpcfg.server.as_str()).to_string(),
username: cfg["username"].as_str().unwrap_or(tmpcfg.username.as_str()).to_string(), username: cfg["username"].as_str().unwrap_or(tmpcfg.username.as_str()).to_string(),
password: cfg["password"].as_str().unwrap_or(tmpcfg.password.as_str()).to_string(), password: cfg["password"].as_str().unwrap_or(tmpcfg.password.as_str()).to_string(),
@ -137,17 +129,51 @@ impl<'a> Config {
sloweffect: Some(cfg["sloweffect"].as_bool().unwrap_or(tmpcfg.sloweffect.unwrap())), sloweffect: Some(cfg["sloweffect"].as_bool().unwrap_or(tmpcfg.sloweffect.unwrap())),
refresh: Some(cfg["refresh"].as_u64().unwrap_or(tmpcfg.refresh.unwrap())), refresh: Some(cfg["refresh"].as_u64().unwrap_or(tmpcfg.refresh.unwrap())),
limit: Some(cfg["limit"].as_u64().unwrap_or(tmpcfg.limit.unwrap())), limit: Some(cfg["limit"].as_u64().unwrap_or(tmpcfg.limit.unwrap())),
=======
server: ncfg["server"]
.as_str()
.unwrap_or(&tmpcfg.server.clone())
.to_string(),
username: ncfg["username"]
.as_str()
.unwrap_or(&tmpcfg.username.clone())
.to_string(),
password: ncfg["password"]
.as_str()
.unwrap_or(&tmpcfg.password.clone())
.to_string(),
authtoken: Some(
ncfg["authtoken"]
.as_str()
.unwrap_or(self.authtoken.clone().unwrap().as_str())
.to_string(),
),
sloweffect: Some(
ncfg["sloweffect"]
.as_bool()
.unwrap_or(tmpcfg.sloweffect.unwrap()),
),
refresh: Some(ncfg["refresh"].as_u64().unwrap_or(tmpcfg.refresh.unwrap())),
limit: Some(ncfg["limit"].as_u64().unwrap_or(tmpcfg.limit.unwrap())),
>>>>>>> Stashed changes
} }
}, }
Err(err) => { Err(err) => {
println!("error occured: '{}'", err); println!("error occured: '{}'", err);
Config::new() Config::new()
} }
}; };
<<<<<<< Updated upstream
self.save(&configfile); self.save(&configfile);
=======
*self = newcfg;
if !fileexists || errorreading {
self.save(&configfile).await;
}
>>>>>>> Stashed changes
} }
pub fn save(&self, configfile: &str) { async fn save(&self, configfile: &str) {
let file = File::create(configfile).unwrap(); let file = File::create(configfile).unwrap();
serde_json::to_writer_pretty(file, &self).unwrap(); serde_json::to_writer_pretty(file, &self).unwrap();
} }

View File

@ -5,9 +5,7 @@ mod zabbix;
#[macro_use] #[macro_use]
extern crate clap; extern crate clap;
use config::Context; #[tokio::main]
async fn main() {
fn main() { padcontrol::run().await;
let mut context = Context::new();
padcontrol::run(&mut context);
} }

View File

@ -1,19 +1,32 @@
use crate::config::Context; use crate::config::Context;
use crate::zabbix::api::get_zabbix_problems; use crate::zabbix::api::*;
use launchy::Color; use launchy::Color;
use launchy::{self, Canvas, CanvasLayout, CanvasLayoutPoller, MsgPollingWrapper, Pad}; use launchy::{self, Canvas, CanvasLayout, CanvasLayoutPoller, MsgPollingWrapper, Pad};
//use std::process::exit;
//use std::sync::mpsc::{channel, Receiver, Sender};
use std::thread::sleep; use std::thread::sleep;
use std::time::Duration; use std::time::Duration;
pub const MATRIX_WIDTH: i32 = 8; const MATRIX_WIDTH: i32 = 8;
pub const MATRIX_SIZE: i32 = MATRIX_WIDTH * MATRIX_WIDTH; const MATRIX_SIZE: i32 = MATRIX_WIDTH * MATRIX_WIDTH;
pub const REQUEST_TIMEOUT: i32 = 10; const REQUEST_WAITFOR: i32 = 5;
const ZERO_COLOR: f32 = 0.;
const ONE_COLOR: f32 = 1.;
pub fn run(ctx: &mut Context) { pub async fn run() {
println!("test1 🦀");
let mut ctx = Context::new().await;
println!("test2 🦀");
loop {
ctx.hotreload().await;
test(&mut ctx).await;
}
/*
let (send, recv): (Sender<Context>, Receiver<Context>) = channel();
match ctx.mode.as_str() { match ctx.mode.as_str() {
"draw" => { /*"draw" => {
let (mut canvas, mut _poller) = initpad(); let (mut canvas, mut _poller) = initpad();
draw(&mut canvas, ctx); draw(&mut canvas, c);
} }
"input" => { "input" => {
let (mut canvas, mut poller) = initpad(); let (mut canvas, mut poller) = initpad();
@ -21,14 +34,20 @@ pub fn run(ctx: &mut Context) {
} }
"effect" => { "effect" => {
let (mut canvas, mut poller) = initpad(); let (mut canvas, mut poller) = initpad();
effect(&mut canvas, &mut poller) effect(&mut canvas, &mut poller);
} }*/
"test" => test(ctx), "test" => loop {
let mut data = cm.lock().unwrap();
let future1 = test(&mut data);
let mut data = cm.lock().unwrap();
let future2 = data.hotreload();
//futures::join!(future1, future2);
},
_ => { _ => {
println!("No valid option choosen for mode"); println!("No valid option choosen for mode");
std::process::exit(1) std::process::exit(1)
} }
} }*/
} }
fn initpad() -> (CanvasLayout<'static>, CanvasLayoutPoller) { fn initpad() -> (CanvasLayout<'static>, CanvasLayoutPoller) {
@ -57,35 +76,26 @@ fn input(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) {
} }
} }
fn test(ctx: &mut Context) { async fn test(ctx: &mut Context) {
println!( let sec = ctx.cfg.refresh.unwrap();
"Refresh rate is {sec} seconds", let zabbix_data = get_zabbix_problems(&ctx.cfg).await.unwrap();
sec = ctx.cfg.refresh.unwrap() println!("{}", zabbix_data);
); sleep(Duration::from_secs(sec));
loop {
let zabbix_data_result = get_zabbix_problems(&ctx.cfg);
let zabbix_data = match zabbix_data_result {
Ok(z) => z,
Err(_) => {
println!("Error requesting zabbix service, retrying in {}", REQUEST_TIMEOUT);
sleep(Duration::from_secs(REQUEST_TIMEOUT as u64));
continue;
}
};
println!("{}", zabbix_data);
ctx.hotreload();
}
} }
pub fn draw(canvas: &mut CanvasLayout, ctx: &mut Context) { /*
async fn draw(canvas: &mut CanvasLayout<'_>, ctx: &mut Context) {
println!("Refresh rate is {} seconds", ctx.cfg.refresh.unwrap()); println!("Refresh rate is {} seconds", ctx.cfg.refresh.unwrap());
loop { loop {
let zabbix_data_result = get_zabbix_problems(&ctx.cfg); let zabbix_data_result = get_zabbix_problems(&ctx.cfg);
let zabbix_data = match zabbix_data_result { let zabbix_data = match zabbix_data_result {
Ok(zabbix_data) => zabbix_data, Ok(zabbix_data) => zabbix_data,
Err(_) => { Err(_) => {
println!("Error requesting zabbix service, retrying in {}", REQUEST_TIMEOUT); println!(
sleep(Duration::from_secs(REQUEST_TIMEOUT as u64)); "Error requesting zabbix service, retrying in {}",
REQUEST_WAITFOR
);
sleep(Duration::from_secs(REQUEST_WAITFOR as u64));
continue; continue;
} }
}; };
@ -116,7 +126,7 @@ pub fn draw(canvas: &mut CanvasLayout, ctx: &mut Context) {
} }
ctx.hotreload(); ctx.hotreload();
} }
} }*/
fn effect(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) { fn effect(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) {
for color in (0u64..).map(|f| Color::red_green_color(f as f32 / 60.0 / 2.5)) { for color in (0u64..).map(|f| Color::red_green_color(f as f32 / 60.0 / 2.5)) {

View File

@ -1,53 +1,53 @@
use std::time::Duration;
use std::thread::sleep;
use crate::config::Config; use crate::config::Config;
use serde_json::{json, Value}; use serde_json::json;
use serde_json::Value as JsonValue;
use std::thread::sleep;
use std::time::Duration;
pub const ZABBIX_API_VERSION: &'static str = "2.0"; pub const ZABBIX_API_VERSION: &'static str = "2.0";
pub const ZABBIX_API_ID: i32 = 1; pub const ZABBIX_API_ID: i32 = 1;
/// Refresh the user token /// Refresh the user token
pub fn get_zabbix_authtoken(cfg: &mut Config) { pub async fn get_zabbix_authtoken(cfg: &mut Config) -> Result<(), reqwest::Error> {
let body = build_query_auth_token(&cfg.username, &cfg.password); let body = build_query_auth_token(&cfg.username, &cfg.password);
let resp = reqwest::blocking::Client::new() let resp = reqwest::Client::new()
.post(&cfg.server) .post(&cfg.server)
.json(&body) .json(&body)
.send() .send()
.unwrap(); .await?;
let values: JsonValue = resp.json().await?;
let values: Value = resp.json().unwrap();
cfg.authtoken = Some(values["result"].as_str().unwrap().to_string()); cfg.authtoken = Some(values["result"].as_str().unwrap().to_string());
Ok(())
} }
/// Fetch Zabbix problems /// Fetch Zabbix problems
pub fn get_zabbix_problems(cfg: &Config) -> Result<Value, reqwest::Error> { pub async fn get_zabbix_problems(cfg: &Config) -> Result<JsonValue, reqwest::Error> {
let body = build_query_triggers(&cfg.authtoken.as_ref().unwrap_or(&String::from(""))); let body = build_query_triggers(&cfg.authtoken.as_ref().unwrap_or(&String::from("")));
let resp = reqwest::blocking::Client::new() let resp = reqwest::Client::new()
.post(&cfg.server) .post(&cfg.server)
.json(&body) .json(&body)
.send(); .send()
.await?;
Ok(resp.unwrap().json().unwrap()) Ok(resp.json().await?)
} }
/// Check if Zabbix is operational /// Check if Zabbix is operational
/// Undefinitely check that Zabbix works /// Undefinitely check that Zabbix works
fn check_zabbix_connection(cfg: &Config) -> bool { async fn _async_check_zabbix_connection(cfg: &Config) -> Result<bool, reqwest::Error> {
let mut res: bool = false; let mut res: bool = false;
let delay = 5; let delay = 5;
let timeout = 10; let timeout = 10;
while !res { while !res {
let req = reqwest::blocking::Client::builder().timeout(Duration::from_secs(timeout)).build().unwrap(); let req = reqwest::Client::builder()
let resp = req.get(&cfg.server) .timeout(Duration::from_secs(timeout))
.send(); .build()
match resp { .unwrap();
Ok(_) => res = true, req.get(&cfg.server).send().await?;
Err(_) => res = false res = true;
} println!("Waiting for {delay}", delay = delay);
println!("Waiting for {delay}", delay=delay);
sleep(Duration::from_secs(delay)); sleep(Duration::from_secs(delay));
} }
res Ok(res)
} }
/// Build the query that fetchs the token /// Build the query that fetchs the token
@ -65,7 +65,7 @@ fn build_query_auth_token(zabbix_username: &String, zabbix_password: &String) ->
} }
/// Build the query that fetchs problems /// Build the query that fetchs problems
fn build_query_problems(zabbix_token: &String, zabbix_limit: i64) -> Value { fn _build_query_problems(zabbix_token: &String, zabbix_limit: i64) -> JsonValue {
let zabbix_api_function = "problem.get"; let zabbix_api_function = "problem.get";
json!({ json!({
"jsonrpc": ZABBIX_API_VERSION, "jsonrpc": ZABBIX_API_VERSION,

View File

@ -2,11 +2,12 @@ use serde_json::Value;
use std::fmt::{Display, Formatter, Result}; use std::fmt::{Display, Formatter, Result};
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub enum ZabbixStatus { pub enum _ZabbixStatus {
High = 4, _Disaster = 5,
Average = 3, _High = 4,
Warning = 2, _Average = 3,
Info = 1, _Warning = 2,
_Info = 1,
} }
#[derive(Debug, Clone)] #[derive(Debug, Clone)]