Merge pull request #4665 from halibobo1205/fix/db

fix(db): fix concurrent error

Author: halibobo1205

chainbase/src/main/java/org/tron/common/storage/leveldb/LevelDbDataSourceImpl.java

@@ -174,9 +174,14 @@ public Path getDbPath() {
    * reset database.
    */
   public void resetDb() {
-    closeDB();
-    FileUtil.recursiveDelete(getDbPath().toString());
-    initDB();
+    resetDbLock.writeLock().lock();
+    try {
+      closeDB();
+      FileUtil.recursiveDelete(getDbPath().toString());
+      initDB();
+    } finally {
+      resetDbLock.writeLock().unlock();
+    }
   }
 
   @Override
@@ -506,9 +511,17 @@ private DBIterator getDBIterator() {
    */
   @Override
   public List<String> getStats() throws Exception {
-    String stat = database.getProperty("leveldb.stats");
-    String[] stats = stat.split("\n");
-    return Arrays.stream(stats).skip(3).collect(Collectors.toList());
+    resetDbLock.readLock().lock();
+    try {
+      if (!isAlive()) {
+        return Collections.emptyList();
+      }
+      String stat = database.getProperty("leveldb.stats");
+      String[] stats = stat.split("\n");
+      return Arrays.stream(stats).skip(3).collect(Collectors.toList());
+    } finally {
+      resetDbLock.readLock().unlock();
+    }
   }
 
   @Override

chainbase/src/main/java/org/tron/common/storage/rocksdb/RocksDbDataSourceImpl.java

@@ -109,9 +109,14 @@ public void closeDB() {
 
   @Override
   public void resetDb() {
-    closeDB();
-    FileUtil.recursiveDelete(getDbPath().toString());
-    initDB();
+    resetDbLock.writeLock().lock();
+    try {
+      closeDB();
+      FileUtil.recursiveDelete(getDbPath().toString());
+      initDB();
+    } finally {
+      resetDbLock.writeLock().unlock();
+    }
   }
 
   private boolean quitIfNotAlive() {
@@ -528,9 +533,17 @@ public RocksDbDataSourceImpl newInstance() {
    */
   @Override
   public List<String> getStats() throws Exception {
-    String stat = database.getProperty("rocksdb.levelstats");
-    String[] stats = stat.split("\n");
-    return Arrays.stream(stats).skip(2).collect(Collectors.toList());
+    resetDbLock.readLock().lock();
+    try {
+      if (!isAlive()) {
+        return Collections.emptyList();
+      }
+      String stat = database.getProperty("rocksdb.levelstats");
+      String[] stats = stat.split("\n");
+      return Arrays.stream(stats).skip(2).collect(Collectors.toList());
+    } finally {
+      resetDbLock.readLock().unlock();
+    }
   }
 
   @Override

chainbase/src/main/java/org/tron/core/db/TronDatabase.java

@@ -59,7 +59,7 @@ protected TronDatabase(String dbName) {
   }
 
   @PostConstruct
-  private void init() {
+  protected void init() {
     dbStatService.register(dbSource);
   }
 

chainbase/src/main/java/org/tron/core/store/CheckPointV2Store.java

@@ -45,4 +45,9 @@ public void forEach(Consumer action) {
   public Spliterator spliterator() {
     return null;
   }
-}
+
+  @Override
+  protected void init() {
+  }
+
+}