refactor: Delete sqlite by clearing tables instead of

erasing files

Author: ashchan

packages/neuron-wallet/src/database/chain/cleaner.ts

@@ -1,21 +1,14 @@
-import fs from 'fs'
-import path from 'path'
-import env from 'env'
+import { getConnection } from 'typeorm'
+import InputEntity from './entities/input'
+import OutputEntity from './entities/output'
+import TransactionEntity from './entities/transaction'
+import SyncInfoEntity from './entities/sync-info'
 
 // Clean local sqlite storage
 export default class ChainCleaner {
-  // Delete all sqlite files under cells folder.
-  // It doesn't handle error or throw exception when deleting fails.
-  public static clean() {
-    const folder = path.join(env.fileBasePath, 'cells')
-    fs.readdir(folder, (err, files) => {
-      if (err) {
-        return
-      }
-
-      for (const file of files.filter(f => { return path.extname(f) === '.sqlite' })) {
-        fs.unlink(path.join(folder, file), () => {})
-      }
-    })
+  public static async clean() {
+    for (const entity of [InputEntity, OutputEntity, TransactionEntity, SyncInfoEntity]) {
+      await getConnection().getRepository(entity).clear()
+    }
   }
 }