feat(conf): support configuring DB storage directory

src/conf/kiwi.conf

@@ -190,6 +190,10 @@ memory 10M
 # Log directory for Kiwi logs
 # log-dir /data/kiwi_rs/logs
 
+# Directory where RocksDB data files are stored.
+# Each database instance creates a subdirectory under this path (e.g., ./db/0, ./db/1).
+# db-dir ./db
+
 # Enable Redis compatibility mode
 # redis-compatible-mode yes
 

src/conf/src/config.rs

@@ -226,6 +226,7 @@ pub struct Config {
     pub binding: String,
     pub timeout: u32,
     pub log_dir: String,
+    pub db_dir: String,
     pub redis_compatible_mode: bool,
     pub db_instance_num: usize,
 
@@ -242,6 +243,7 @@ impl Default for Config {
             timeout: 50,
             memory: 1024 * 1024 * 1024, // 1GB
             log_dir: "/data/kiwi_rs/logs".to_string(),
+            db_dir: "./db".to_string(),
             redis_compatible_mode: false,
 
             rocksdb_max_subcompactions: 0,
@@ -347,6 +349,9 @@ impl Config {
                 "log-dir" => {
                     config.log_dir = value;
                 }
+                "db-dir" => {
+                    config.db_dir = value;
+                }
                 "redis-compatible-mode" => {
                     config.redis_compatible_mode =
                         parse_bool_from_string(&value).map_err(|e| Error::InvalidConfig {

src/conf/src/lib.rs

@@ -60,6 +60,7 @@ mod tests {
 
         assert_eq!(50, config.timeout);
         assert_eq!("/data/kiwi_rs/logs", config.log_dir);
+        assert_eq!("./db", config.db_dir);
         assert!(!config.redis_compatible_mode);
         assert_eq!(3, config.db_instance_num);
 
@@ -78,6 +79,7 @@ mod tests {
             timeout: 100,
             redis_compatible_mode: false,
             log_dir: "".to_string(),
+            db_dir: "./db".to_string(),
             memory: 1024,
             rocksdb_max_subcompactions: 0,
             rocksdb_max_background_jobs: 4,
@@ -104,4 +106,26 @@ mod tests {
         invalid_config.port = 8080;
         assert!(invalid_config.validate().is_ok());
     }
+
+    #[test]
+    fn test_db_dir_default() {
+        let config = Config::default();
+        assert_eq!("./db", config.db_dir);
+    }
+
+    #[test]
+    fn test_db_dir_from_config_file() {
+        use std::io::Write;
+
+        let config_path = "/tmp/kiwi_test_db_dir.conf";
+        let mut f = std::fs::File::create(config_path).unwrap();
+        writeln!(f, "port 7379").unwrap();
+        writeln!(f, "db-dir /data/kiwi/db").unwrap();
+        drop(f);
+
+        let config = Config::load(config_path).unwrap();
+        assert_eq!("/data/kiwi/db", config.db_dir);
+
+        let _ = std::fs::remove_file(config_path);
+    }
 }

src/net/src/lib.rs

@@ -99,7 +99,7 @@ impl ServerFactory {
                 }
             }
             #[cfg(unix)]
-            "unix" => Some(Box::new(unix::UnixServer::new(addr))),
+            "unix" => Some(Box::new(unix::UnixServer::new(addr, None))),
             #[cfg(not(unix))]
             "unix" => None,
             _ => None,
@@ -110,13 +110,14 @@ impl ServerFactory {
     pub fn create_legacy_server(
         protocol: &str,
         addr: Option<String>,
+        db_dir: Option<&str>,
     ) -> Option<Box<dyn ServerTrait>> {
         match protocol.to_lowercase().as_str() {
-            "tcp" => TcpServer::new(addr)
+            "tcp" => TcpServer::new(addr, db_dir)
                 .ok()
                 .map(|s| Box::new(s) as Box<dyn ServerTrait>),
             #[cfg(unix)]
-            "unix" => Some(Box::new(unix::UnixServer::new(addr))),
+            "unix" => Some(Box::new(unix::UnixServer::new(addr, db_dir))),
             #[cfg(not(unix))]
             "unix" => None,
             _ => None,
@@ -172,9 +173,10 @@ impl ServerFactory {
         protocol: &str,
         addr: Option<String>,
         raft_node: Arc<dyn Send + Sync>,
+        db_dir: Option<&str>,
     ) -> Option<Box<dyn ServerTrait>> {
         match protocol.to_lowercase().as_str() {
-            "tcp" => ClusterTcpServer::new(addr, raft_node)
+            "tcp" => ClusterTcpServer::new(addr, raft_node, db_dir)
                 .ok()
                 .map(|s| Box::new(s) as Box<dyn ServerTrait>),
             _ => None,

src/net/src/tcp.rs

@@ -115,10 +115,9 @@ pub struct TcpServer {
 }
 
 impl TcpServer {
-    pub fn new(addr: Option<String>) -> Result<Self, Box<dyn Error>> {
-        // TODO: Get storage options from config
+    pub fn new(addr: Option<String>, db_dir: Option<&str>) -> Result<Self, Box<dyn Error>> {
         let storage_options = Arc::new(StorageOptions::default());
-        let db_path = PathBuf::from("./db");
+        let db_path = PathBuf::from(db_dir.unwrap_or("./db"));
         let mut storage = Storage::new(1, 0);
         let executor = Arc::new(CmdExecutorBuilder::new().build());
 
@@ -165,10 +164,10 @@ impl ClusterTcpServer {
     pub fn new(
         addr: Option<String>,
         raft_node: Arc<dyn Send + Sync>,
+        db_dir: Option<&str>,
     ) -> Result<Self, Box<dyn Error>> {
-        // TODO: Get storage options from config
         let storage_options = Arc::new(StorageOptions::default());
-        let db_path = PathBuf::from("./db");
+        let db_path = PathBuf::from(db_dir.unwrap_or("./db"));
         let mut storage = Storage::new(1, 0);
         let executor = Arc::new(CmdExecutorBuilder::new().build());
 

src/net/src/unix.rs

@@ -33,10 +33,10 @@ pub struct UnixServer {
 }
 
 impl UnixServer {
-    pub fn new(path: Option<String>) -> Self {
+    pub fn new(path: Option<String>, db_dir: Option<&str>) -> Self {
         let path = path.unwrap_or_else(|| "/tmp/kiwidb.sock".to_string());
         let storage_options = Arc::new(StorageOptions::default());
-        let db_path = PathBuf::from("./db");
+        let db_path = PathBuf::from(db_dir.unwrap_or("./db"));
         let mut storage = Storage::new(1, 0);
         storage.open(storage_options, db_path).unwrap();
         let executor = Arc::new(CmdExecutorBuilder::new().build());

src/server/src/main.rs

@@ -130,10 +130,10 @@ fn main() -> std::io::Result<()> {
     info!("Storage components initialized, starting storage server...");
 
     // Initialize storage server in storage runtime (runs in background)
-
+    let db_dir = config.db_dir.clone();
     storage_handle.spawn(async move {
         info!("Initializing storage server...");
-        match initialize_storage_server(storage_receiver).await {
+        match initialize_storage_server(storage_receiver, &db_dir).await {
             Ok(_) => {
                 error!("Storage server exited unexpectedly - this should never happen!");
             }
@@ -260,12 +260,13 @@ fn main() -> std::io::Result<()> {
 /// Initialize the storage server in the storage runtime
 async fn initialize_storage_server(
     request_receiver: tokio::sync::mpsc::Receiver<runtime::StorageRequest>,
+    db_dir: &str,
 ) -> Result<(), DualRuntimeError> {
     info!("Initializing storage server...");
 
     // Create storage options and path
     let storage_options = Arc::new(StorageOptions::default());
-    let db_path = PathBuf::from("./db");
+    let db_path = PathBuf::from(db_dir);
 
     // Create storage instance (not yet opened)
     let mut storage = Storage::new(1, 0); // Single instance, db_id 0