blockworks-foundation
/
mangorpc-latency-tester
mirror of https://github.com/blockworks-foundation/mangorpc-latency-tester.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

better logging

This commit is contained in:
GroovieGermanikus 2024-06-06 15:23:55 +02:00
parent 13601c478c
commit f7a22558dc
No known key found for this signature in database
GPG Key ID: 5B6EB831A5CD2015
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/rpcnode_check_alive.rs
View File

@ -23,7 +23,7 @@ use tokio::task::{JoinError, JoinHandle, JoinSet};
use tokio::time::{Instant, timeout}; use tokio::time::{Instant, timeout};
use tokio_stream::{Stream, StreamExt}; use tokio_stream::{Stream, StreamExt};
use tokio_stream::wrappers::{BroadcastStream, ReceiverStream}; use tokio_stream::wrappers::{BroadcastStream, ReceiverStream};
use tracing::{debug, info, trace, warn}; use tracing::{debug, error, info, trace, warn};
use websocket_tungstenite_retry::websocket_stable::{StableWebSocket, WsMessage}; use websocket_tungstenite_retry::websocket_stable::{StableWebSocket, WsMessage};
use url::Url; use url::Url;
use yellowstone_grpc_proto::geyser::{SubscribeRequest, SubscribeRequestFilterAccounts, SubscribeRequestFilterSlots, SubscribeUpdate}; use yellowstone_grpc_proto::geyser::{SubscribeRequest, SubscribeRequestFilterAccounts, SubscribeRequestFilterSlots, SubscribeUpdate};
@ -84,21 +84,34 @@ async fn main() {
info!("all tasks started..."); info!("all tasks started...");
let mut tasks_successful = 0;
let mut tasks_timeout = 0;
let mut tasks_failed = 0;
while let Some(res) = all_check_tasks.join_next().await { while let Some(res) = all_check_tasks.join_next().await {
match res { match res {
Ok(CheckResult::Success(check)) => { Ok(CheckResult::Success(check)) => {
info!("one more Task completed: {:?}, {} left", check, all_check_tasks.len()); tasks_successful += 1;
info!("one more task completed <{:?}>, {} left", check, all_check_tasks.len());
} }
Ok(CheckResult::Timeout(check)) => { Ok(CheckResult::Timeout(check)) => {
info!("timeout running Task {:?}", check); tasks_timeout += 1;
warn!("timeout running task <{:?}>", check);
} }
Err(_) => { Err(_) => {
warn!("Task failed"); tasks_failed += 1;
warn!("Task execution failed");
} }
} }
} }
let tasks_total = tasks_successful + tasks_failed + tasks_timeout;
info!("all tasks completed..."); assert!(tasks_total > 0, "no results");
if tasks_failed + tasks_timeout > 0 {
warn!("tasks failed ({}) or timed out ({}) of {} total", tasks_failed, tasks_timeout, tasks_total);
} else {
info!("all {} tasks completed...", tasks_total);
}
} }
fn add_task(check: Check, task: impl Future<Output = ()> + Send + 'static, all_check_tasks: &mut JoinSet<CheckResult>) { fn add_task(check: Check, task: impl Future<Output = ()> + Send + 'static, all_check_tasks: &mut JoinSet<CheckResult>) {