Ecosystem/gupaxx
1
0
Fork 0
mirror of https://github.com/Cyrix126/gupaxx.git synced 2024-11-16 23:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

node: check get_info sync progress

Browse source
This commit is contained in:
hinto.janai 2023-11-24 09:12:03 -05:00
parent 846b6265ea
commit ff4b0181ad
No known key found for this signature in database
GPG key ID: D47CE05FA175A499
1 changed files with 33 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
src/node.rs
View file

@ -239,8 +239,8 @@ impl NodeData {
// A struct repr of the JSON-RPC we're
// expecting back from the pinged nodes.
//
// This structs leave out most fields on purpose,
// we only need a few to verify it works.
// This struct leaves out most fields on purpose,
// we only need a few to verify the node is ok.
#[derive(Debug, serde::Deserialize)]
struct GetInfo<'a> {
id: &'a str,
@ -379,22 +379,37 @@ impl Ping {
Ok(fastest_info)
}
async fn response(client: Client<HttpConnector>, request: Request<Body>, ip: &'static str, ping: Arc<Mutex<Self>>, percent: f32, node_vec: Arc<Mutex<Vec<NodeData>>>) {
async fn response(
client: Client<HttpConnector>,
request: Request<Body>,
ip: &'static str,
ping: Arc<Mutex<Self>>,
percent: f32,
node_vec: Arc<Mutex<Vec<NodeData>>>
) {
const DEAD_NODE_PING: u128 = 5000;
let ms;
let now = Instant::now();
const DEAD_NODE_PING: u128 = 5000;
match tokio::time::timeout(Duration::from_secs(5), client.request(request)).await {
Ok(Ok(json_rpc)) => {
// Attempt to convert to JSON-RPC.
match hyper::body::to_bytes(json_rpc.into_body()).await {
Ok(b) => {
if serde_json::from_slice::<GetInfo<'_>>(&b).is_ok() {
ms = now.elapsed().as_millis();
} else {
ms = DEAD_NODE_PING;
warn!("Ping | {ip} responded but with invalid get_info, remove this node!");
match serde_json::from_slice::<GetInfo<'_>>(&b) {
Ok(rpc) => {
if rpc.result.mainnet && rpc.result.synchronized {
ms = now.elapsed().as_millis();
} else {
ms = DEAD_NODE_PING;
warn!("Ping | {ip} responded with valid get_info but is not in sync, remove this node!");
}
}
_ => {
ms = DEAD_NODE_PING;
warn!("Ping | {ip} responded but with invalid get_info, remove this node!");
}
}
},
_ => ms = DEAD_NODE_PING,
@ -406,21 +421,21 @@ impl Ping {
let info = format!("{ms}ms ... {ip}");
info!("Ping | {ms}ms ... {ip}");
let color;
if ms < 300 {
color = GREEN;
let color = if ms < 300 {
GREEN
} else if ms < 500 {
color = YELLOW;
YELLOW
} else if ms < DEAD_NODE_PING {
color = RED;
RED
} else {
color = BLACK;
}
BLACK
};
let mut ping = lock!(ping);
ping.msg = info;
ping.prog += percent;
drop(ping);
lock!(node_vec).push(NodeData { ip, ms, color, });
lock!(node_vec).push(NodeData { ip, ms, color });
}
}