avoid using scala.reflect and use a safer way to delete rocksdb data

input-output-hk · Jun 7, 2021 · 7401e36 · 7401e36
1 parent caad67d
commit 7401e36
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/main/scala/io/iohk/ethereum/Mantis.scala b/src/main/scala/io/iohk/ethereum/Mantis.scala
@@ -3,8 +3,8 @@ package io.iohk.ethereum
 import io.iohk.ethereum.nodebuilder.{StdNode, TestNode}
 import io.iohk.ethereum.utils.{Config, Logger}
 
+import java.nio.file.{Files, Paths}
 import java.util.logging.LogManager
-import scala.reflect.io.Directory
 
 object Mantis extends Logger {
   def main(args: Array[String]): Unit = {
@@ -13,8 +13,7 @@ object Mantis extends Logger {
     val node =
       if (Config.testmode) {
         log.info("Starting Mantis in test mode")
-        log.info("Deleting previous database {}", Config.Db.RocksDb.path)
-        Directory(Config.Db.RocksDb.path).deleteRecursively()
+        deleteRocksDBFiles()
         new TestNode
       } else new StdNode
 
@@ -23,4 +22,12 @@ object Mantis extends Logger {
 
     node.start()
   }
+
+  private def deleteRocksDBFiles(): Unit = {
+    log.info("Deleting previous database {}", Config.Db.RocksDb.path)
+    Files.list(Paths.get(Config.Db.RocksDb.path))
+      .map(_.toFile)
+      .filter(_.isFile)
+      .forEach(f => f.delete())
+  }
 }