nki/youmubot
1
0
Fork 0
mirror of https://github.com/natsukagami/youmubot.git synced 2025-04-18 16:28:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add a timeout to the reqwest client

Browse source
This commit is contained in:
Natsu Kagami 2023-11-21 13:27:22 +01:00
parent 67a7eb9aec
commit 2be38a3759
Signed by: nki
GPG key ID: 55A032EB38B49ADB
2 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
youmubot-osu/src/lib.rs
View file

@ -9,7 +9,7 @@ use models::*;
use request::builders::*;
use request::*;
use reqwest::Client as HTTPClient;
use std::convert::TryInto;
use std::{convert::TryInto, time::Duration};
use youmubot_prelude::{ratelimit::Ratelimit, *};
/// The number of requests per minute to the osu! server.
@ -35,7 +35,10 @@ impl Client {
/// Create a new client from the given API key.
pub fn new(key: String) -> Client {
let client = Ratelimit::new(
HTTPClient::new(),
HTTPClient::builder()
.timeout(Duration::from_secs(2))
.build()
.expect("Client should be built"),
REQUESTS_PER_MINUTE,
std::time::Duration::from_secs(60),
);

11
youmubot-prelude/src/announcer.rs
View file

@ -163,15 +163,14 @@ impl AnnouncerHandler {
async move {
loop {
eprintln!(" - scanning key `{}`", key);
Self::announce(data.clone(), cache.clone(), key, announcer)
.map(move |v| {
if let Err(e) = v {
match Self::announce(data.clone(), cache.clone(), key, announcer).await {
Err(e) => {
eprintln!(" - key `{}`: {:?}", *key, e)
} else {
}
Ok(()) => {
eprintln!(" - key `{}`: complete", *key)
}
})
.await;
};
looper.tick().await;
}
}