estym/Obsessed-Yanqing
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

1.4.5 - Added Panic on event loop error

Browse source
This commit is contained in:
Evann Regnault 2023-12-04 16:30:54 +01:00
parent 446c8e4369
commit bb1e61b20d
2 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Cargo.toml
View file

@ -1,6 +1,6 @@
[package] [package]
name = "obsessed-yanqing" name = "obsessed-yanqing"
version = "1.4.4" version = "1.4.5"
edition = "2021" edition = "2021"
authors = ["Evann Regnault"] authors = ["Evann Regnault"]
license = "MIT" license = "MIT"

13
src/main.rs
View file

@ -18,8 +18,13 @@ use crate::metrics::core::{AllRegistries, create_registries, Empty, setup_server
use crate::mongo::core::get_all_status_messages; use crate::mongo::core::get_all_status_messages;
fn update_daily(ctx: Context, all_registries: Arc<AllRegistries>) { fn update_daily(ctx: Context, all_registries: Arc<AllRegistries>) {
let default_panic = std::panic::take_hook();
tokio::task::spawn(async move { tokio::task::spawn(async move {
let mut interval = tokio::time::interval(Duration::from_secs(60*60)); std::panic::set_hook(Box::new(move |info| {
default_panic(info);
std::process::exit(1);
}));
let mut interval = tokio::time::interval(Duration::from_secs(60 * 60));
loop { loop {
get_total_guilds_count(&ctx, &all_registries).await; get_total_guilds_count(&ctx, &all_registries).await;
let status_messages = get_all_status_messages().await; let status_messages = get_all_status_messages().await;
@ -29,10 +34,10 @@ fn update_daily(ctx: Context, all_registries: Arc<AllRegistries>) {
let ctx = ctx.clone(); let ctx = ctx.clone();
let embeds = embeds.clone(); let embeds = embeds.clone();
tokio::spawn(async move { tokio::spawn(async move {
if sm.channel_id == 0 {return} if sm.channel_id == 0 { return }
let msg = ChannelId::from(sm.channel_id as u64).message(ctx.clone().http, sm.message_id as u64).await; let msg = ChannelId::from(sm.channel_id as u64).message(ctx.clone().http, sm.message_id as u64).await;
match msg { match msg {
Ok (mut m) => { Ok(mut m) => {
match m.edit(&ctx.http, |f| { match m.edit(&ctx.http, |f| {
f.set_embeds(embeds) f.set_embeds(embeds)
}).await { }).await {
@ -40,7 +45,7 @@ fn update_daily(ctx: Context, all_registries: Arc<AllRegistries>) {
Err(e) => println!("Error while editing message {}", e) Err(e) => println!("Error while editing message {}", e)
} }
}, },
Err(_) => {println!("Cannot update guild : {}", sm.channel_id);} Err(_) => { println!("Cannot update guild : {}", sm.channel_id); }
} }
}); });
} }