Add register_is_done to ConsensusEngine

This is a work-around to the currently order-dependant and
non-deterministic initialization of Tendermint ConsensusEngine/Worker.

Author: foriequal0

codechain/run_node.rs

@@ -376,6 +376,7 @@ pub fn run_node(matches: &ArgMatches) -> Result<(), String> {
 
     // drop the scheme to free up genesis state.
     drop(scheme);
+    client.client().engine().register_is_done();
 
     cinfo!(TEST_SCRIPT, "Initialization complete");
 

core/src/consensus/mod.rs

@@ -268,6 +268,8 @@ pub trait ConsensusEngine: Sync + Send {
 
     fn register_snapshot_notify_sender(&self, _sender: SnapshotNotifySender) {}
 
+    fn register_is_done(&self) {}
+
     fn send_snapshot_notify(&self, _block_hash: BlockHash) {}
 
     fn get_best_block_from_best_proposal_header(&self, header: &HeaderView) -> BlockHash {

core/src/consensus/tendermint/engine.rs

@@ -341,10 +341,6 @@ impl ConsensusEngine for Tendermint {
         let extension = service.register_extension(move |api| TendermintExtension::new(inner, timeouts, api));
         let client = Arc::downgrade(&self.client().unwrap());
         self.extension_initializer.send((extension, client)).unwrap();
-
-        let (result, receiver) = crossbeam::bounded(1);
-        self.inner.send(worker::Event::Restore(result)).unwrap();
-        receiver.recv().unwrap();
     }
 
     fn register_time_gap_config_to_worker(&self, time_gap_params: TimeGapParams) {
@@ -363,6 +359,12 @@ impl ConsensusEngine for Tendermint {
         client.add_notify(Arc::downgrade(&self.chain_notify) as Weak<dyn ChainNotify>);
     }
 
+    fn register_is_done(&self) {
+        let (result, receiver) = crossbeam::bounded(1);
+        self.inner.send(worker::Event::Restore(result)).unwrap();
+        receiver.recv().unwrap();
+    }
+
     fn get_best_block_from_best_proposal_header(&self, header: &HeaderView) -> BlockHash {
         header.parent_hash()
     }