Create workdir if not exist. (wechaty/wechaty-puppet-padplus#134)

huan · Feb 8, 2020 · 98d3963 · 98d3963
1 parent 5bfdb5b
commit 98d3963
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/src/flash-store-sync.ts b/src/flash-store-sync.ts
@@ -34,9 +34,6 @@ export class FlashStoreSync<K = any, V = any> implements Map<K, V> {
 
     if (!this.workdir) {
       this.workdir = path.join(appRoot, '.flash-store-sync')
-      if (!fs.existsSync(this.workdir)) {
-        fs.mkdirSync(this.workdir)
-      }
     }
 
     this.asyncBusyDict = {}
@@ -49,7 +46,6 @@ export class FlashStoreSync<K = any, V = any> implements Map<K, V> {
     this.flashStore = new FlashStore<K, V>(this.workdir)
 
     this.asyncBusyAdd(this.loadStoreToCache())
-
   }
 
   private async loadStoreToCache (): Promise<void> {

diff --git a/src/flash-store.ts b/src/flash-store.ts
@@ -50,13 +50,24 @@ export class FlashStore<K = string, V = any> implements AsyncMap<K, V> {
 
     if (!this.workdir) {
       this.workdir = path.join(appRoot, '.flash-store')
-      if (!fs.existsSync(this.workdir)) {
+    }
+
+    if (fs.existsSync(this.workdir)) {
+      log.verbose('FlashStore', 'constructor(%s)', workdir)
+    } else {
+      /**
+        * Mkdir for the database directory. (only for the last path, no recursive)
+        */
+      log.silly('FlashStore', 'constructor(%s) not exist, creating...', this.workdir)
+      try {
         fs.mkdirSync(this.workdir)
+      } catch (e) {
+        log.error('FlashStore', 'constructor(%s) exception: %s', this.workdir, e && e.message)
+        throw e
       }
+      log.silly('FlashStore', 'constructor(%s) workdir created.', this.workdir)
     }
 
-    log.verbose('FlashStore', 'constructor(%s)', workdir)
-
     // we use seperate workdir for snapdb, leveldb, and rocksdb etc.
     const medeaWorkdir = path.join(this.workdir, 'medea')