absolutebi/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-12-08 12:19:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add hooks to the main loop

Browse Source 
Lets the Ethereum processor track the first key set as soon as it's set.
This commit is contained in:
Luke Parker
2024-09-19 01:31:52 -04:00
parent a691be21c8
commit 1367e41510
9 changed files with 67 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
processor/bin/src/lib.rs
View File

@@ -157,8 +157,18 @@ async fn first_block_after_time<S: ScannerFeed>(feed: &S, serai_time: u64) -> u6
}
}
/// Hooks to run during the main loop.
pub trait Hooks {
/// A hook to run upon receiving a message.
fn on_message(txn: &mut impl DbTxn, msg: &messages::CoordinatorMessage);
}
impl Hooks for () {
fn on_message(_: &mut impl DbTxn, _: &messages::CoordinatorMessage) {}
}
/// The main loop of a Processor, interacting with the Coordinator.
pub async fn main_loop<
H: Hooks,
S: ScannerFeed,
K: KeyGenParams<ExternalNetworkCiphersuite: Ciphersuite<G = KeyFor<S>>>,
Sch: Clone
@@ -183,6 +193,7 @@ pub async fn main_loop<
let db_clone = db.clone();
let mut txn = db.txn();
let msg = coordinator.next_message(&mut txn).await;
H::on_message(&mut txn, &msg);
let mut txn = Some(txn);
match msg {
messages::CoordinatorMessage::KeyGen(msg) => {