mirrors/buzzrelay
1
0
Fork 0
mirror of https://github.com/astro/buzzrelay.git synced 2024-11-21 11:41:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

relay: add backoff on error

Browse source
This commit is contained in:
Astro 2023-05-15 01:14:42 +02:00
parent 7b93419a91
commit fc659b57a1
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/relay.rs
View file

@ -1,4 +1,4 @@
use std::{sync::Arc, collections::{HashSet, HashMap}, time::Instant};
use std::{sync::Arc, collections::{HashSet, HashMap}, time::{Duration, Instant}};
use futures::{channel::mpsc::{channel, Sender}, StreamExt};
use metrics::{increment_counter, histogram};
use serde::Deserialize;
@ -73,15 +73,30 @@ fn spawn_worker(client: Arc<reqwest::Client>, inbox: String) -> Sender<Job> {
let (tx, mut rx) = channel(1024);
tokio::spawn(async move {
let mut errors = 0u32;
let mut last_request = None;
while let Some(Job { post_url, actor_id, key_id, private_key, body }) = rx.next().await {
if errors > 0 && last_request.map_or(false, |last_request|
Instant::now() - last_request < Duration::from_secs(10) * errors
) {
// there have been errors, skip for time proportional
// to the number of subsequent errors
tracing::trace!("skip {} from {} to {}", post_url, actor_id, inbox);
continue;
}
tracing::debug!("relay {} from {} to {}", post_url, actor_id, inbox);
last_request = Some(Instant::now());
if let Err(e) = send::send_raw(
&client, &inbox,
&key_id, &private_key, body
).await {
tracing::error!("relay::send {:?}", e);
errors = errors.saturating_add(1);
} else {
// success
errors = 0;
systemd::daemon::notify(
false, [
(systemd::daemon::STATE_WATCHDOG, "1")