Parse data-dir

src/cli.yml

@@ -5,6 +5,13 @@ subcommands:
   - run:
       about: run the picodata instance
       args:
+        - data-dir:
+            long: data-dir
+            help: Here instance persists all its data
+            takes_value: true
+            default_value: "."
+            value_name: path
+
         - listen:
             # short: l
             long: listen

src/main.rs

@@ -62,7 +62,13 @@ fn main_run(matches: &clap::ArgMatches) {
         envp.insert("PICODATA_PEER".to_owned(), peer);
     }
 
-    for arg in ["listen", "instance-id", "replicaset-id", "cluster-id"] {
+    for arg in [
+        "cluster-id",
+        "data-dir",
+        "instance-id",
+        "listen",
+        "replicaset-id",
+    ] {
         if let Some(v) = matches.value_of(arg) {
             let k = format!("PICODATA_{}", arg.to_uppercase().replace("-", "_"));
             envp.insert(k, v.to_owned());

tests/cli.rs

@@ -58,6 +58,7 @@ fn pass_environment() {
     cmd.args(["--cluster-id", "sam"]);
     cmd.args(["--replicaset-id", "r1"]);
     cmd.args(["--instance-id", "i1"]);
+    cmd.args(["--data-dir", "/tmp/picodata-test"]);
     cmd.args(["--peer", "i1,i2"]);
     cmd.args(["--peer", "i3"]);
     cmd.arg("-e").arg(
@@ -78,6 +79,7 @@ fn pass_environment() {
         assert_eq(os.environ()['PICODATA_REPLICASET_ID'], "r1")
         assert_eq(os.environ()['PICODATA_INSTANCE_ID'], "i1")
         assert_eq(os.environ()['PICODATA_PEER'], "i1,i2,i3")
+        assert_eq(os.environ()['PICODATA_DATA_DIR'], "/tmp/picodata-test")
     "#,
     );
     cmd.assert().success();