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

Put raft configs into a namespace.

Jing Yang 4 лет назад
Родитель
Сommit
48e7a866e5

+ 1 - 1
test_configs/src/lib.rs

@@ -1,4 +1,4 @@
-pub mod config;
+pub mod raft;
 mod rpcs;
 
 pub use rpcs::{make_rpc_handler, register_server, RpcClient};

+ 4 - 6
test_configs/src/config/mod.rs → test_configs/src/raft/config.rs

@@ -12,8 +12,6 @@ use rand::{thread_rng, Rng};
 use crate::register_server;
 use ruaft::{ApplyCommandMessage, Persister, Raft, Term};
 
-pub mod persister;
-
 struct ConfigState {
     rafts: Vec<Option<Raft<i32>>>,
     connected: Vec<bool>,
@@ -23,7 +21,7 @@ struct LogState {
     committed_logs: Vec<Vec<i32>>,
     results: Vec<Result<()>>,
     max_index: usize,
-    saved: Vec<Arc<persister::Persister>>,
+    saved: Vec<Arc<super::Persister>>,
 }
 
 pub struct Config {
@@ -290,7 +288,7 @@ impl Config {
             raft.kill();
         }
         let mut log = self.log.lock();
-        log.saved[index] = Arc::new(persister::Persister::new());
+        log.saved[index] = Arc::new(super::Persister::new());
         log.saved[index].save_state(data);
     }
 
@@ -445,7 +443,7 @@ impl Config {
 #[macro_export]
 macro_rules! make_config {
     ($server_count:expr, $unreliable:expr) => {
-        $crate::config::make_config(
+        $crate::raft::config::make_config(
             $server_count,
             $unreliable,
             stdext::function_name!(),
@@ -475,7 +473,7 @@ pub fn make_config(
     });
 
     let mut saved = vec![];
-    saved.resize_with(server_count, || Arc::new(persister::Persister::new()));
+    saved.resize_with(server_count, || Arc::new(super::Persister::new()));
     let log = Arc::new(Mutex::new(LogState {
         committed_logs: vec![vec![]; server_count],
         results: vec![],

+ 4 - 0
test_configs/src/raft/mod.rs

@@ -0,0 +1,4 @@
+pub mod config;
+mod persister;
+
+pub use persister::Persister;

+ 0 - 0
test_configs/src/config/persister/mod.rs → test_configs/src/raft/persister.rs


+ 1 - 1
tests/agreement_tests.rs

@@ -1,7 +1,7 @@
 #![allow(clippy::identity_op)]
 use rand::{thread_rng, Rng};
 use scopeguard::defer;
-use test_configs::{config, make_config};
+use test_configs::{make_config, raft::config};
 
 #[test]
 fn basic_agreement() -> config::Result<()> {

+ 1 - 1
tests/election_tests.rs

@@ -1,5 +1,5 @@
 use scopeguard::defer;
-use test_configs::{config, make_config};
+use test_configs::{make_config, raft::config};
 
 #[test]
 fn initial_election() -> config::Result<()> {

+ 1 - 1
tests/persist_tests.rs

@@ -6,7 +6,7 @@ use std::sync::Arc;
 
 use rand::{thread_rng, Rng};
 use scopeguard::defer;
-use test_configs::{config, make_config};
+use test_configs::{make_config, raft::config};
 
 #[test]
 fn persist1() -> config::Result<()> {