Noratrieb/uptime
1
0
Fork 0
mirror of https://github.com/Noratrieb/uptime.git synced 2026-01-14 16:45:06 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

write website

Browse source
This commit is contained in:
nora 2023-08-31 20:46:44 +02:00
parent 24abf6d98c
commit 7ff9a67148
15 changed files with 1580 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

19
src/client.rs
View file

@ -1,16 +1,16 @@
use chrono::Utc;
use eyre::{bail, Context, Result};
use std::collections::HashMap;
use std::collections::BTreeMap;
use tracing::info;
use crate::config::WebsiteConfig;
pub struct Client {
websites: Vec<WebsiteConfig>,
req: reqwest::Client,
pub websites: Vec<WebsiteConfig>,
pub req: reqwest::Client,
}
pub struct Results {
pub states: HashMap<String, CheckResult>,
pub states: BTreeMap<String, CheckResult>,
}
pub struct CheckResult {
@ -18,13 +18,15 @@ pub struct CheckResult {
pub state: CheckState,
}
#[derive(sqlx::Type)]
#[sqlx(rename_all = "snake_case")]
pub enum CheckState {
Ok,
NotOk,
}
pub async fn do_checks(client: &Client) -> Results {
let mut states = HashMap::new();
let mut states = BTreeMap::new();
for website in &client.websites {
let check_result = make_request(&client.req, website).await;
states.insert(website.name.clone(), check_result);
@ -33,10 +35,13 @@ pub async fn do_checks(client: &Client) -> Results {
Results { states }
}
#[tracing::instrument(skip(client))]
async fn make_request(client: &reqwest::Client, website: &WebsiteConfig) -> CheckResult {
let time = Utc::now();
let result = client.get(website.url.clone()).send().await;
info!(?result, ?website.url, "Made health request");
match result {
Ok(res) => CheckResult {
time,
@ -46,7 +51,7 @@ async fn make_request(client: &reqwest::Client, website: &WebsiteConfig) -> Chec
CheckState::NotOk
},
},
Err(err) => CheckResult {
Err(_) => CheckResult {
time,
state: CheckState::NotOk,
},