github/darkfi
1
1
Fork 0
mirror of https://github.com/darkrenaissance/darkfi.git synced 2026-04-28 03:00:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

consensus/sync: Add protocol mode.

Browse Source
This commit is contained in:
parazyd
2022-04-21 19:10:25 +02:00
parent 8c98f8eb0d
commit 74e0412238
2 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bin/darkfid2/src/main.rs
View File

@@ -472,7 +472,7 @@ async fn realmain(args: Args, ex: Arc<Executor<'_>>) -> Result<()> {
//.register(net::SESSION_ALL, move |channel, p2p| {
.register(!net::SESSION_SEED, move |channel, p2p| {
let state = _state.clone();
async move { ProtocolSync::init(channel, state, p2p).await.unwrap() }
async move { ProtocolSync::init(channel, state, p2p, args.consensus).await.unwrap() }
})
.await;

20
src/consensus2/proto/protocol_sync.rs
View File

@@ -25,6 +25,7 @@ pub struct ProtocolSync {
jobsman: ProtocolJobsManagerPtr,
state: ValidatorStatePtr,
p2p: P2pPtr,
consensus_mode: bool,
}
impl ProtocolSync {
@@ -32,6 +33,7 @@ impl ProtocolSync {
channel: ChannelPtr,
state: ValidatorStatePtr,
p2p: P2pPtr,
consensus_mode: bool,
) -> Result<ProtocolBasePtr> {
let msg_subsystem = channel.get_message_subsystem();
msg_subsystem.add_dispatch::<BlockOrder>().await;
@@ -47,6 +49,7 @@ impl ProtocolSync {
jobsman: ProtocolJobsManager::new("SyncProtocol", channel),
state,
p2p,
consensus_mode,
}))
}
@@ -75,17 +78,18 @@ impl ProtocolSync {
debug!("ProtocolSync::handle_receive_block() received block");
// TODO: The following code should be executed only by replicators, not
// consensus nodes.
// Node stores finalized flock, if it doesn't exist (checking by slot),
// and removes its transactions from the unconfirmed_txs vector.
// Consensus-mode enabled nodes have already performed these steps,
// during proposal finalization.
// Extra validations can be added here.
let info_copy = (*info).clone();
if !self.state.read().await.blockchain.has_block(&info_copy)? {
self.state.write().await.blockchain.add(&[info_copy.clone()])?;
self.state.write().await.remove_txs(info_copy.txs.clone())?;
self.p2p.broadcast(info_copy).await?;
if !self.consensus_mode {
let info_copy = (*info).clone();
if !self.state.read().await.blockchain.has_block(&info_copy)? {
self.state.write().await.blockchain.add(&[info_copy.clone()])?;
self.state.write().await.remove_txs(info_copy.txs.clone())?;
self.p2p.broadcast(info_copy).await?;
}
}
}
}