mirror of
https://github.com/serai-dex/serai.git
synced 2025-12-08 12:19:24 +00:00
b5a6b0693e
This isn't necessary. Because we just log the error, we never match off of it,
we don't need any structure beyond String (or now Debug, which still gives us
a way to print the error). This is for the ergonomics of not having to
constantly write `.map_err(|e| format!("{e:?}"))`.
58 lines
1.7 KiB
Rust
58 lines
1.7 KiB
Rust
use core::future::Future;
|
|
use std::time::{Duration, SystemTime};
|
|
|
|
use serai_db::*;
|
|
use serai_task::{DoesNotError, ContinuallyRan};
|
|
|
|
use crate::evaluator::CosignedBlocks;
|
|
|
|
/// How often callers should broadcast the cosigns flagged for rebroadcasting.
|
|
pub const BROADCAST_FREQUENCY: Duration = Duration::from_secs(60);
|
|
const SYNCHRONY_EXPECTATION: Duration = Duration::from_secs(10);
|
|
const ACKNOWLEDGEMENT_DELAY: Duration =
|
|
Duration::from_secs(BROADCAST_FREQUENCY.as_secs() + SYNCHRONY_EXPECTATION.as_secs());
|
|
|
|
create_db!(
|
|
SubstrateCosignDelay {
|
|
// The latest cosigned block number.
|
|
LatestCosignedBlockNumber: () -> u64,
|
|
}
|
|
);
|
|
|
|
/// A task to delay acknowledgement of cosigns.
|
|
pub(crate) struct CosignDelayTask<D: Db> {
|
|
pub(crate) db: D,
|
|
}
|
|
|
|
impl<D: Db> ContinuallyRan for CosignDelayTask<D> {
|
|
type Error = DoesNotError;
|
|
|
|
fn run_iteration(&mut self) -> impl Send + Future<Output = Result<bool, Self::Error>> {
|
|
async move {
|
|
let mut made_progress = false;
|
|
loop {
|
|
let mut txn = self.db.txn();
|
|
|
|
// Receive the next block to mark as cosigned
|
|
let Some((block_number, time_evaluated)) = CosignedBlocks::try_recv(&mut txn) else {
|
|
break;
|
|
};
|
|
// Calculate when we should mark it as valid
|
|
let time_valid =
|
|
SystemTime::UNIX_EPOCH + Duration::from_secs(time_evaluated) + ACKNOWLEDGEMENT_DELAY;
|
|
// Sleep until then
|
|
tokio::time::sleep(SystemTime::now().duration_since(time_valid).unwrap_or(Duration::ZERO))
|
|
.await;
|
|
|
|
// Set the cosigned block
|
|
LatestCosignedBlockNumber::set(&mut txn, &block_number);
|
|
txn.commit();
|
|
|
|
made_progress = true;
|
|
}
|
|
|
|
Ok(made_progress)
|
|
}
|
|
}
|
|
}
|