eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-04 04:56:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix(scheduled_tasks): don't crash on missing data

Browse source
This commit is contained in:
alyssa 2025-04-08 17:34:50 +00:00
parent eca5cbff97
commit 4acc51195d
1 changed files with 5 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
crates/scheduled_tasks/src/tasks.rs
View file

@ -10,7 +10,6 @@ use metrics::gauge;
use num_format::{Locale, ToFormattedString};
use reqwest::ClientBuilder;
use sqlx::Executor;
use tracing::error;
use crate::AppCtx;
@ -190,28 +189,25 @@ pub async fn update_stats_api(ctx: AppCtx) -> anyhow::Result<()> {
let data = resp.json::<PrometheusResult>().await?;
let error_handler = || {
error!("missing data at {}", $q);
anyhow::anyhow!("missing data at {}", $q)
};
data.data
.result
.get(0)
.ok_or_else(error_handler)
.unwrap()
.ok_or_else(error_handler)?
.value
.clone()
.get(1)
.ok_or_else(error_handler)
.unwrap()
.ok_or_else(error_handler)?
.as_str()
.ok_or_else(error_handler)
.unwrap()
.ok_or_else(error_handler)?
.parse::<$t>()?
}};
($t:ty, $q:expr, $wrap:expr) => {{
let val = prom_instant_query!($t, $q);
let val = (val * $wrap).round() / $wrap;
format!("{:.2}", val).parse::<f64>().unwrap()
format!("{:.2}", val).parse::<f64>()?
}};
}