Forgot to recalculate hash at strategic locations
This commit is contained in:
parent
a09f019cc5
commit
a7b85146fe
1 changed files with 21 additions and 7 deletions
|
@ -70,9 +70,12 @@ impl Members {
|
|||
nodes.sort_by_key(|(id, _status)| *id);
|
||||
|
||||
let mut hasher = Sha256::new();
|
||||
eprintln!("Current set of pingable nodes: --");
|
||||
for (id, status) in nodes {
|
||||
eprintln!("{} {}", hex::encode(id), status.addr);
|
||||
hasher.input(format!("{} {}\n", hex::encode(id), status.addr));
|
||||
}
|
||||
eprintln!("END --");
|
||||
self.status_hash.copy_from_slice(&hasher.result()[..]);
|
||||
}
|
||||
}
|
||||
|
@ -85,18 +88,20 @@ pub struct NodeStatus {
|
|||
|
||||
impl System {
|
||||
pub fn new(config: Config, id: UUID) -> Self {
|
||||
System{
|
||||
config,
|
||||
id,
|
||||
rpc_client: Client::new(),
|
||||
members: RwLock::new(Members{
|
||||
let mut members = Members{
|
||||
status: HashMap::new(),
|
||||
status_hash: [0u8; 32],
|
||||
config: NetworkConfig{
|
||||
members: HashMap::new(),
|
||||
version: 0,
|
||||
},
|
||||
}),
|
||||
};
|
||||
members.recalculate_status_hash();
|
||||
System{
|
||||
config,
|
||||
id,
|
||||
rpc_client: Client::new(),
|
||||
members: RwLock::new(members),
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -197,9 +202,9 @@ impl System {
|
|||
propagate.push(node.clone());
|
||||
}
|
||||
}
|
||||
drop(members);
|
||||
|
||||
if propagate.len() > 0 {
|
||||
members.recalculate_status_hash();
|
||||
tokio::spawn(self.clone().broadcast(Message::AdvertiseNodesUp(propagate), PING_TIMEOUT));
|
||||
}
|
||||
|
||||
|
@ -242,9 +247,14 @@ impl System {
|
|||
})).await;
|
||||
|
||||
let mut members = self.members.write().await;
|
||||
let mut has_changes = false;
|
||||
|
||||
for (id, addr, ping_resp) in ping_resps {
|
||||
if let Ok(Message::Ping(ping)) = ping_resp {
|
||||
let is_new = members.handle_ping(addr.ip(), &ping);
|
||||
if is_new {
|
||||
has_changes = true;
|
||||
}
|
||||
if is_new || members.status_hash != ping.status_hash {
|
||||
tokio::spawn(self.clone().pull_status(ping.id.clone()));
|
||||
}
|
||||
|
@ -256,6 +266,7 @@ impl System {
|
|||
if remaining_attempts == 0 {
|
||||
eprintln!("Removing node {} after too many failed pings", hex::encode(id));
|
||||
members.status.remove(id);
|
||||
has_changes = true;
|
||||
} else {
|
||||
if let Some(st) = members.status.get_mut(id) {
|
||||
st.remaining_ping_attempts = remaining_attempts - 1;
|
||||
|
@ -263,6 +274,9 @@ impl System {
|
|||
}
|
||||
}
|
||||
}
|
||||
if has_changes {
|
||||
members.recalculate_status_hash();
|
||||
}
|
||||
drop(members);
|
||||
|
||||
restart_at.await
|
||||
|
|
Loading…
Reference in a new issue