Change sync extension to update pivot after transitioning to Full

CodeChain-io · Dec 10, 2019 · afcf587 · afcf587
1 parent eb6b204
commit afcf587
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/sync/src/block/downloader/header.rs b/sync/src/block/downloader/header.rs
@@ -62,6 +62,10 @@ impl HeaderDownloader {
         }
     }
 
+    pub fn update_pivot(&mut self, hash: BlockHash) {
+        self.pivot = hash;
+    }
+
     pub fn best_hash(&self) -> BlockHash {
         self.best_hash
     }

diff --git a/sync/src/block/extension.rs b/sync/src/block/extension.rs
@@ -1048,6 +1048,10 @@ impl Extension {
 
     fn transition_to_full(&mut self) {
         cdebug!(SYNC, "Transitioning state to {:?}", State::Full);
+        let best_hash = self.client.best_block_header().hash();
+        for downloader in self.header_downloaders.values_mut() {
+            downloader.update_pivot(best_hash);
+        }
         self.state = State::Full;
     }
 }