always create directory

tikv · Jun 16, 2016 · 81e5f3c · 81e5f3c
1 parent b979f1c
commit 81e5f3c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/raftstore/store/snap.rs b/src/raftstore/store/snap.rs
@@ -214,7 +214,15 @@ impl SnapManagerCore {
     }
 
     pub fn try_recover(&self) -> io::Result<()> {
-        for path in try!(fs::read_dir(&self.base)) {
+        let path = Path::new(&self.base);
+        if !path.exists() {
+            try!(fs::create_dir_all(path));
+        }
+        if !path.is_dir() {
+            return Err(io::Error::new(ErrorKind::Other,
+                                      format!("{} should be a directory", path.display())));
+        }
+        for path in try!(fs::read_dir(path)) {
             let p = try!(path);
             if !try!(p.file_type()).is_file() {
                 continue;