Defer sending peer status after transitioning to full sync mode

Author: foriequal0

sync/src/block/extension.rs

@@ -17,6 +17,7 @@
 use std::collections::hash_map::Entry;
 use std::collections::{HashMap, HashSet};
 use std::fs;
+use std::mem::discriminant;
 use std::sync::Arc;
 use std::time::Duration;
 
@@ -174,6 +175,10 @@ impl Extension {
     }
 
     fn send_status(&mut self, id: &NodeId) {
+        if discriminant(&self.state) != discriminant(&State::Full) {
+            return
+        }
+
         let chain_info = self.client.chain_info();
         self.api.send(
             id,
@@ -190,6 +195,10 @@ impl Extension {
     }
 
     fn send_status_broadcast(&mut self) {
+        if discriminant(&self.state) != discriminant(&State::Full) {
+            return
+        }
+
         let chain_info = self.client.chain_info();
         for id in self.connected_nodes.iter() {
             self.api.send(
@@ -1057,6 +1066,7 @@ impl Extension {
             downloader.update_pivot(best_hash);
         }
         self.state = State::Full;
+        self.send_status_broadcast();
     }
 }