Просмотр исходного кода

tau/ raft-diag: using new raft api

ghassmo 4 лет назад
Родитель
Сommit
1292f4c4dd
2 измененных файлов с 16 добавлено и 9 удалено
  1. 8 6
      bin/tau/taud/src/main.rs
  2. 8 3
      script/research/raft-diag/src/main.rs

+ 8 - 6
bin/tau/taud/src/main.rs

@@ -11,7 +11,7 @@ use structopt_toml::StructOptToml;
 
 use darkfi::{
     async_daemonize, net,
-    raft::{NetMsg, ProtocolRaft, Raft},
+    raft::{NetMsg, ProtocolRaft, Raft, RaftSettings},
     rpc::server::listen_and_serve,
     util::{
         cli::{get_log_config, get_log_level, spawn_config},
@@ -172,11 +172,13 @@ async fn realmain(settings: Args, executor: Arc<Executor<'_>>) -> Result<()> {
     let seen_net_msgs = Arc::new(Mutex::new(FxHashMap::default()));
 
     let datastore_raft = datastore_path.join("tau.db");
-    let mut raft = Raft::<EncryptedTask>::new(
-        net_settings.inbound.clone(),
-        datastore_raft,
-        seen_net_msgs.clone(),
-    )?;
+    let raft_settings = RaftSettings {
+        datastore_path: datastore_raft,
+        external_addr: net_settings.external_addr.clone(),
+        ..RaftSettings::default()
+    };
+
+    let mut raft = Raft::<EncryptedTask>::new(raft_settings, seen_net_msgs.clone())?;
 
     let commits_received: Arc<Mutex<Vec<String>>> = Arc::new(Mutex::new(vec![]));
 

+ 8 - 3
script/research/raft-diag/src/main.rs

@@ -10,7 +10,7 @@ use url::Url;
 
 use darkfi::{
     net,
-    raft::{DataStore, NetMsg, ProtocolRaft, Raft},
+    raft::{DataStore, NetMsg, ProtocolRaft, Raft, RaftSettings},
     util::{
         cli::{get_log_config, get_log_level},
         expand_path,
@@ -126,8 +126,13 @@ async fn start(args: Args, executor: Arc<Executor<'_>>) -> Result<()> {
 
     let seen_net_msgs = Arc::new(Mutex::new(FxHashMap::default()));
 
-    let mut raft =
-        Raft::<Message>::new(net_settings.inbound.clone(), datastore_raft, seen_net_msgs.clone())?;
+    let raft_settings = RaftSettings {
+        datastore_path: datastore_raft,
+        external_addr: net_settings.external_addr.clone(),
+        ..RaftSettings::default()
+    };
+
+    let mut raft = Raft::<Message>::new(raft_settings, seen_net_msgs.clone())?;
 
     //
     // P2p setup