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

Start handling P2P messages

Browse Source 
This defines the tart of a very complex series of locks I'm really unhappy
with. At the same time, there's not immediately a better solution. This also
should work without issue.
This commit is contained in:
Luke Parker
2023-04-23 16:56:23 -04:00
parent f2d9d70068
commit ad5522d854
12 changed files with 199 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
coordinator/src/tests/tributary/chain.rs
View File

@@ -1,8 +1,8 @@
use std::time::{Duration, SystemTime};
use zeroize::Zeroizing;
use rand_core::{RngCore, CryptoRng, OsRng};
use futures::{task::Poll, poll};
use ciphersuite::{
group::{ff::Field, GroupEncoding},
@@ -95,11 +95,12 @@ pub async fn run_tributaries(
) {
loop {
for (p2p, tributary) in tributaries.iter_mut() {
while let Some(msg) = p2p.receive().await {
match msg.0 {
P2pMessageKind::Tributary => {
if tributary.handle_message(&msg.1).await {
p2p.broadcast(msg.0, msg.1).await;
while let Poll::Ready(msg) = poll!(p2p.receive()) {
match msg.kind {
P2pMessageKind::Tributary(genesis) => {
assert_eq!(genesis, tributary.genesis());
if tributary.handle_message(&msg.msg).await {
p2p.broadcast(msg.kind, msg.msg).await;
}
}
}
@@ -163,10 +164,11 @@ async fn tributary_test() {
let timeout = SystemTime::now() + Duration::from_secs(65);
while (blocks < 10) && (SystemTime::now().duration_since(timeout).is_err()) {
for (p2p, tributary) in tributaries.iter_mut() {
while let Some(msg) = p2p.receive().await {
match msg.0 {
P2pMessageKind::Tributary => {
tributary.handle_message(&msg.1).await;
while let Poll::Ready(msg) = poll!(p2p.receive()) {
match msg.kind {
P2pMessageKind::Tributary(genesis) => {
assert_eq!(genesis, tributary.genesis());
tributary.handle_message(&msg.msg).await;
}
}
}
@@ -187,10 +189,11 @@ async fn tributary_test() {
// Handle all existing messages
for (p2p, tributary) in tributaries.iter_mut() {
while let Some(msg) = p2p.receive().await {
match msg.0 {
P2pMessageKind::Tributary => {
tributary.handle_message(&msg.1).await;
while let Poll::Ready(msg) = poll!(p2p.receive()) {
match msg.kind {
P2pMessageKind::Tributary(genesis) => {
assert_eq!(genesis, tributary.genesis());
tributary.handle_message(&msg.msg).await;
}
}
}

2
coordinator/src/tests/tributary/dkg.rs
View File

@@ -30,7 +30,7 @@ async fn dkg_test() {
let keys = new_keys(&mut OsRng);
let spec = new_spec(&mut OsRng, &keys);
let mut tributaries = new_tributaries(&keys, &spec).await;
let tributaries = new_tributaries(&keys, &spec).await;
// Run the tributaries in the background
tokio::spawn(run_tributaries(tributaries.clone()));

2
coordinator/src/tests/tributary/tx.rs
View File

@@ -19,7 +19,7 @@ async fn tx_test() {
let keys = new_keys(&mut OsRng);
let spec = new_spec(&mut OsRng, &keys);
let mut tributaries = new_tributaries(&keys, &spec).await;
let tributaries = new_tributaries(&keys, &spec).await;
// Run the tributaries in the background
tokio::spawn(run_tributaries(tributaries.clone()));