feat: Support TLS config

Cargo.toml

@@ -19,7 +19,7 @@ default-members = [
 exclude = ["raft-rs", "raftify-cli"]
 
 [workspace.package]
-version = "0.1.78"
+version = "0.1.80"
 authors = ["Lablup Inc."]
 edition = "2021"
 description = "Experimental High level Raft framework"

examples/Cargo.toml

@@ -10,3 +10,7 @@ keywords.workspace = true
 
 [dependencies]
 raftify.workspace = true
+
+[features]
+default = []
+tls = []

examples/memstore/Cargo.toml

@@ -26,11 +26,12 @@ serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 tokio = { version = "1.40", features = ["full"] }
 color-backtrace = "0.6.1"
-raftify = { version = "0.1.78" }
-raftify_cli = { version = "0.1.1" }
+raftify = { version = "0.1.80" }
+raftify_cli = { version = "0.1.80" }
 
 [features]
 default = ["heed_storage"]
 inmemory_storage = []
 heed_storage = []
 rocksdb_storage = []
+tls = []

examples/memstore/dynamic-members/src/main.rs

@@ -70,19 +70,17 @@ async fn main() -> std::result::Result<(), Box<dyn std::error::Error>> {
         Some(peer_addr) => {
             log::info!("Running in Follower mode");
 
-            let ticket = Raft::request_id(options.raft_addr.clone(), peer_addr.clone())
+            let ticket = Raft::request_id(options.raft_addr.clone(), peer_addr.clone(), None)
                 .await
                 .unwrap();
             let node_id = ticket.reserved_id;
-            let mut cfg = build_config(node_id);
-
-            cfg.initial_peers = Some(ticket.peers.clone().into());
+            let cfg = build_config(node_id, Some(ticket.peers.clone().into()));
 
             #[cfg(feature = "inmemory_storage")]
             let log_storage = MemStorage::create();
 
             #[cfg(feature = "heed_storage")]
-            let log_storage = HeedStorage::create(&cfg.log_dir, &cfg.clone(), logger.clone())
+            let log_storage = HeedStorage::create(cfg.get_log_dir(), &cfg.clone(), logger.clone())
                 .expect("Failed to create storage");
 
             #[cfg(feature = "rocksdb_storage")]
@@ -121,13 +119,13 @@ async fn main() -> std::result::Result<(), Box<dyn std::error::Error>> {
             }
 
             let leader_node_id = 1;
-            let cfg = build_config(leader_node_id);
+            let cfg = build_config(leader_node_id, None);
 
             #[cfg(feature = "inmemory_storage")]
             let log_storage = MemStorage::create();
 
             #[cfg(feature = "heed_storage")]
-            let log_storage = HeedStorage::create(&cfg.log_dir, &cfg.clone(), logger.clone())
+            let log_storage = HeedStorage::create(cfg.get_log_dir(), &cfg.clone(), logger.clone())
                 .expect("Failed to create storage");
 
             #[cfg(feature = "rocksdb_storage")]

examples/memstore/src/client/main.rs

@@ -7,7 +7,7 @@ use memstore_example_harness::state_machine::LogEntry;
 #[actix_rt::main]
 async fn main() {
     println!("---Message propose---");
-    let mut leader_client = create_client(&"127.0.0.1:60061").await.unwrap();
+    let mut leader_client = create_client(&"127.0.0.1:60061", None).await.unwrap();
 
     leader_client
         .propose(raft_service::ProposeArgs {

examples/memstore/static-members/Cargo.toml

@@ -33,4 +33,3 @@ default = ["heed_storage"]
 inmemory_storage = []
 heed_storage = []
 rocksdb_storage = []
-

examples/memstore/static-members/src/main.rs

@@ -71,14 +71,13 @@ async fn main() -> std::result::Result<(), Box<dyn std::error::Error>> {
         .get_node_id_by_addr(options.raft_addr.clone())
         .unwrap();
 
-    let mut cfg = build_config(node_id);
-    cfg.initial_peers = Some(initial_peers.clone());
+    let cfg = build_config(node_id, Some(initial_peers.clone()));
 
     #[cfg(feature = "inmemory_storage")]
     let log_storage = MemStorage::create();
 
     #[cfg(feature = "heed_storage")]
-    let log_storage = HeedStorage::create(&cfg.log_dir, &cfg.clone(), logger.clone())
+    let log_storage = HeedStorage::create(cfg.get_log_dir(), &cfg.clone(), logger.clone())
         .expect("Failed to create storage");
 
     #[cfg(feature = "rocksdb_storage")]

examples/src/config.rs

@@ -1,8 +1,11 @@
-use raftify::{Config, RaftConfig};
+use raftify::{Config, ConfigBuilder, Peers, RaftConfig};
+
+#[cfg(feature = "tls")]
+use raftify::TlsConfig;
 
 use crate::utils::{ensure_directory_exist, get_storage_path};
 
-pub fn build_config(node_id: u64) -> Config {
+pub fn build_config(node_id: u64, initial_peers: Option<Peers>) -> Config {
     let raft_config = RaftConfig {
         id: node_id,
         election_tick: 10,
@@ -14,14 +17,32 @@ pub fn build_config(node_id: u64) -> Config {
     let storage_pth = get_storage_path("./logs", node_id);
     ensure_directory_exist(&storage_pth).expect("Failed to create storage directory");
 
-    Config {
-        bootstrap_from_snapshot: false,
-        tick_interval: 0.2,
-        log_dir: storage_pth.clone(),
-        save_compacted_logs: true,
-        compacted_log_dir: storage_pth,
-        compacted_log_size_threshold: 1024 * 1024 * 1024,
-        raft_config,
-        ..Default::default()
+    #[allow(unused_mut)]
+    let mut config_builder = ConfigBuilder::new()
+        .log_dir("./logs".to_owned())
+        .save_compacted_logs(true)
+        .compacted_log_dir("./logs".to_owned())
+        .compacted_log_size_threshold(1024 * 1024 * 1024)
+        .raft_config(raft_config)
+        .tick_interval(0.2);
+
+    #[cfg(feature = "tls")]
+    {
+        let client_tls_config = TlsConfig {
+            cert_path: None,
+            key_path: None,
+            ca_cert_path: Some("./ca_cert.pem".to_string()),
+            domain_name: Some("localhost".to_string()),
+        };
+
+        config_builder = config_builder.global_client_tls_config(client_tls_config);
     }
+
+    let config_builder = if let Some(peers) = initial_peers {
+        config_builder.initial_peers(peers)
+    } else {
+        config_builder
+    };
+
+    config_builder.build()
 }