feat(db): improve performance for db query

1. add cache for db
tronprotocol · Nov 8, 2022 · 5a96e5d · 5a96e5d
1 parent 2d88f6e
commit 5a96e5d
Show file tree

Hide file tree

Showing 17 changed files with 382 additions and 74 deletions.
diff --git a/chainbase/src/main/java/org/tron/core/db2/common/DB.java b/chainbase/src/main/java/org/tron/core/db2/common/DB.java
@@ -15,7 +15,7 @@ public interface DB<K, V> extends Iterable<Map.Entry<K, V>>, Instance<DB<K, V>>
 
   void remove(K k);
 
-  Iterator iterator();
+  Iterator<Map.Entry<K, V>> iterator();
 
   void close();
 

diff --git a/chainbase/src/main/java/org/tron/core/db2/core/AbstractSnapshot.java b/chainbase/src/main/java/org/tron/core/db2/core/AbstractSnapshot.java
@@ -15,8 +15,6 @@ public abstract class AbstractSnapshot<K, V> implements Snapshot {
 
   protected WeakReference<Snapshot> next;
 
-  protected boolean isOptimized;
-
   @Override
   public Snapshot advance() {
     return new SnapshotImpl(this);
@@ -36,9 +34,4 @@ public void setNext(Snapshot next) {
   public String getDbName() {
     return db.getDbName();
   }
-
-  @Override
-  public boolean isOptimized(){
-    return isOptimized;
-  }
 }
diff --git a/chainbase/src/main/java/org/tron/core/db2/core/Snapshot.java b/chainbase/src/main/java/org/tron/core/db2/core/Snapshot.java
@@ -46,6 +46,4 @@ static boolean isImpl(Snapshot snapshot) {
   void updateSolidity();
 
   String getDbName();
-
-  boolean isOptimized();
 }
diff --git a/chainbase/src/main/java/org/tron/core/db2/core/SnapshotImpl.java b/chainbase/src/main/java/org/tron/core/db2/core/SnapshotImpl.java
@@ -30,16 +30,6 @@ public class SnapshotImpl extends AbstractSnapshot<Key, Value> {
     }
     previous = snapshot;
     snapshot.setNext(this);
-    // inherit
-    isOptimized = snapshot.isOptimized();
-    // merge for DynamicPropertiesStore，about 100 keys
-    if (isOptimized) {
-      if (root == previous ){
-        Streams.stream(root.iterator()).forEach( e -> put(e.getKey(),e.getValue()));
-      }else {
-        merge(previous);
-      }
-    }
   }
 
   @Override
@@ -50,10 +40,6 @@ public byte[] get(byte[] key) {
   private byte[] get(Snapshot head, byte[] key) {
     Snapshot snapshot = head;
     Value value;
-    if (isOptimized) {
-      value = db.get(Key.of(key));
-      return value == null ? null: value.getBytes();
-    }
     while (Snapshot.isImpl(snapshot)) {
       if ((value = ((SnapshotImpl) snapshot).db.get(Key.of(key))) != null) {
         return value.getBytes();

diff --git a/chainbase/src/main/java/org/tron/core/db2/core/SnapshotRoot.java b/chainbase/src/main/java/org/tron/core/db2/core/SnapshotRoot.java
@@ -1,14 +1,17 @@
 package org.tron.core.db2.core;
 
-import ch.qos.logback.core.encoder.ByteArrayUtil;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Streams;
 import java.util.HashMap;
 import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.stream.Collectors;
 import lombok.Getter;
+import org.tron.common.cache.CacheManager;
+import org.tron.common.cache.TronCache;
+import org.tron.common.parameter.CommonParameter;
 import org.tron.common.utils.ByteArray;
 import org.tron.core.ChainBaseManager;
 import org.tron.core.capsule.AccountCapsule;
@@ -23,11 +26,17 @@ public class SnapshotRoot extends AbstractSnapshot<byte[], byte[]> {
   private Snapshot solidity;
   private boolean isAccountDB;
 
+  private TronCache<WrappedByteArray, WrappedByteArray> cache;
+  private static final List<String> CACHE_DBS = CommonParameter.getInstance()
+      .getStorage().getCacheDbs();
+
   public SnapshotRoot(DB<byte[], byte[]> db) {
     this.db = db;
     solidity = this;
-    isOptimized = "properties".equalsIgnoreCase(db.getDbName());
     isAccountDB = "account".equalsIgnoreCase(db.getDbName());
+    if (CACHE_DBS.contains(this.db.getDbName())) {
+      this.cache = CacheManager.allocate(this.db.getDbName());
+    }
   }
 
   private boolean needOptAsset() {
@@ -37,11 +46,18 @@ private boolean needOptAsset() {
 
   @Override
   public byte[] get(byte[] key) {
-    return db.get(key);
+    WrappedByteArray cache = getCache(key);
+    if (cache != null) {
+      return cache.getBytes();
+    }
+    byte[] value = db.get(key);
+    putCache(key, value);
+    return value;
   }
 
   @Override
   public void put(byte[] key, byte[] value) {
+    byte[] v = value;
     if (needOptAsset()) {
       if (ByteArray.isEmpty(value)) {
         remove(key);
@@ -56,10 +72,10 @@ public void put(byte[] key, byte[] value) {
       }
       assetStore.putAccount(item.getInstance());
       item.clearAsset();
-      db.put(key, item.getData());
-    } else {
-      db.put(key, value);
+      v = item.getData();
     }
+    db.put(key, v);
+    putCache(key, v);
   }
 
   @Override
@@ -68,6 +84,7 @@ public void remove(byte[] key) {
       ChainBaseManager.getInstance().getAccountAssetStore().deleteAccount(key);
     }
     db.remove(key);
+    putCache(key, null);
   }
 
   @Override
@@ -81,6 +98,7 @@ public void merge(Snapshot from) {
       processAccount(batch);
     } else {
       ((Flusher) db).flush(batch);
+      putCache(batch);
     }
   }
 
@@ -97,6 +115,7 @@ public void merge(List<Snapshot> snapshots) {
       processAccount(batch);
     } else {
       ((Flusher) db).flush(batch);
+      putCache(batch);
     }
   }
 
@@ -120,11 +139,37 @@ private void processAccount(Map<WrappedByteArray, WrappedByteArray> batch) {
       }
     });
     ((Flusher) db).flush(accounts);
+    putCache(accounts);
     if (assets.size() > 0) {
       assetStore.updateByBatch(AccountAssetStore.convert(assets));
     }
   }
 
+  private boolean cached() {
+    return Objects.nonNull(this.cache);
+  }
+
+  private void putCache(byte[] key, byte[] value) {
+    if (cached()) {
+      cache.put(WrappedByteArray.of(key), WrappedByteArray.of(value));
+    }
+  }
+
+  private void putCache(Map<WrappedByteArray, WrappedByteArray> values) {
+    if (cached()) {
+      values.forEach(cache::put);
+    }
+  }
+
+  private WrappedByteArray getCache(byte[] key) {
+    if (cached()) {
+      return cache.getIfPresent(WrappedByteArray.of(key));
+    }
+    return null;
+  }
+
+  // second cache
+
   @Override
   public Snapshot retreat() {
     return this;
@@ -142,11 +187,13 @@ public Iterator<Map.Entry<byte[], byte[]>> iterator() {
 
   @Override
   public void close() {
+    CacheManager.release(this.getDbName());
     ((Flusher) db).close();
   }
 
   @Override
   public void reset() {
+    CacheManager.release(this.getDbName());
     ((Flusher) db).reset();
   }
 

diff --git a/chainbase/src/main/java/org/tron/core/service/MortgageService.java b/chainbase/src/main/java/org/tron/core/service/MortgageService.java
@@ -1,14 +1,9 @@
 package org.tron.core.service;
 
 import com.google.protobuf.ByteString;
-
 import java.math.BigInteger;
-import java.util.ArrayList;
 import java.util.Comparator;
-import java.util.HashMap;
 import java.util.List;
-import java.util.Map;
-
 import lombok.Getter;
 import lombok.Setter;
 import lombok.extern.slf4j.Slf4j;
@@ -18,7 +13,6 @@
 import org.tron.common.utils.StringUtil;
 import org.tron.core.capsule.AccountCapsule;
 import org.tron.core.capsule.WitnessCapsule;
-import org.tron.core.config.Parameter.ChainConstant;
 import org.tron.core.exception.BalanceInsufficientException;
 import org.tron.core.store.AccountStore;
 import org.tron.core.store.DelegationStore;
@@ -52,31 +46,17 @@ public void initStore(WitnessStore witnessStore, DelegationStore delegationStore
   }
 
   public void payStandbyWitness() {
-    List<WitnessCapsule> witnessCapsules = witnessStore.getAllWitnesses();
-    Map<ByteString, WitnessCapsule> witnessCapsuleMap = new HashMap<>();
-    List<ByteString> witnessAddressList = new ArrayList<>();
-    for (WitnessCapsule witnessCapsule : witnessCapsules) {
-      witnessAddressList.add(witnessCapsule.getAddress());
-      witnessCapsuleMap.put(witnessCapsule.getAddress(), witnessCapsule);
-    }
-    witnessAddressList.sort(Comparator.comparingLong((ByteString b) -> witnessCapsuleMap.get(b).getVoteCount())
-            .reversed().thenComparing(Comparator.comparingInt(ByteString::hashCode).reversed()));
-    if (witnessAddressList.size() > ChainConstant.WITNESS_STANDBY_LENGTH) {
-      witnessAddressList = witnessAddressList.subList(0, ChainConstant.WITNESS_STANDBY_LENGTH);
+    List<WitnessCapsule> witnessStandbys = witnessStore.getWitnessStandby();
+    long voteSum = witnessStandbys.stream().mapToLong(WitnessCapsule::getVoteCount).sum();
+    if (voteSum < 1) {
+      return;
     }
-    long voteSum = 0;
     long totalPay = dynamicPropertiesStore.getWitness127PayPerBlock();
-    for (ByteString b : witnessAddressList) {
-      voteSum += witnessCapsuleMap.get(b).getVoteCount();
-    }
-
-    if (voteSum > 0) {
-      for (ByteString b : witnessAddressList) {
-        double eachVotePay = (double) totalPay / voteSum;
-        long pay = (long) (witnessCapsuleMap.get(b).getVoteCount() * eachVotePay);
-        logger.debug("pay {} stand reward {}", Hex.toHexString(b.toByteArray()), pay);
-        payReward(b.toByteArray(), pay);
-      }
+    double eachVotePay = (double) totalPay / voteSum;
+    for (WitnessCapsule w : witnessStandbys) {
+      long pay = (long) (w.getVoteCount() * eachVotePay);
+      payReward(w.getAddress().toByteArray(), pay);
+      logger.debug("pay {} stand reward {}", Hex.toHexString(w.getAddress().toByteArray()), pay);
     }
   }
 

diff --git a/chainbase/src/main/java/org/tron/core/store/WitnessStore.java b/chainbase/src/main/java/org/tron/core/store/WitnessStore.java
@@ -1,6 +1,8 @@
 package org.tron.core.store;
 
 import com.google.common.collect.Streams;
+import java.util.ArrayList;
+import java.util.Comparator;
 import java.util.List;
 import java.util.Map.Entry;
 import java.util.stream.Collectors;
@@ -9,16 +11,23 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Component;
+import org.tron.common.cache.CacheManager;
+import org.tron.common.cache.TronCache;
 import org.tron.core.capsule.WitnessCapsule;
+import org.tron.core.config.Parameter;
 import org.tron.core.db.TronStoreWithRevoking;
 
 @Slf4j(topic = "DB")
 @Component
 public class WitnessStore extends TronStoreWithRevoking<WitnessCapsule> {
 
+  private final TronCache<Integer, List<WitnessCapsule>> witnessStandby;
+
   @Autowired
   protected WitnessStore(@Value("witness") String dbName) {
     super(dbName);
+    String strategy = "initialCapacity=1,maximumSize=1,expireAfterAccess=6h";
+    witnessStandby = CacheManager.allocate("witnessStandby", strategy);
   }
 
   /**
@@ -35,4 +44,33 @@ public WitnessCapsule get(byte[] key) {
     byte[] value = revokingDB.getUnchecked(key);
     return ArrayUtils.isEmpty(value) ? null : new WitnessCapsule(value);
   }
+
+  public List<WitnessCapsule> getWitnessStandby() {
+    List<WitnessCapsule> list =
+        witnessStandby.getIfPresent(Parameter.ChainConstant.WITNESS_STANDBY_LENGTH);
+    if (list != null) {
+      return list;
+    }
+    return updateWitnessStandby(null);
+  }
+
+  public List<WitnessCapsule> updateWitnessStandby(List<WitnessCapsule> all) {
+    List<WitnessCapsule> ret;
+    if (all == null) {
+      all = getAllWitnesses();
+    }
+    all.sort(Comparator.comparingLong(WitnessCapsule::getVoteCount)
+        .reversed().thenComparing(Comparator.comparingInt(
+            (WitnessCapsule w) -> w.getAddress().hashCode()).reversed()));
+    if (all.size() > Parameter.ChainConstant.WITNESS_STANDBY_LENGTH) {
+      ret = new ArrayList<>(all.subList(0, Parameter.ChainConstant.WITNESS_STANDBY_LENGTH));
+    } else {
+      ret = new ArrayList<>(all);
+    }
+    // trim voteCount = 0
+    ret.removeIf(w -> w.getVoteCount() < 1);
+    witnessStandby.put(Parameter.ChainConstant.WITNESS_STANDBY_LENGTH, ret);
+    return ret;
+  }
+
 }
diff --git a/common/src/main/java/org/tron/common/cache/CacheManager.java b/common/src/main/java/org/tron/common/cache/CacheManager.java
@@ -0,0 +1,45 @@
+package org.tron.common.cache;
+
+import com.google.common.cache.CacheLoader;
+import com.google.common.cache.CacheStats;
+import com.google.common.collect.Maps;
+import java.util.Map;
+import java.util.stream.Collectors;
+import org.tron.common.parameter.CommonParameter;
+
+public class CacheManager {
+
+  private static final Map<String, TronCache<?, ?>> CACHES  = Maps.newConcurrentMap();
+
+  public static <K, V> TronCache<K, V> allocate(String name) {
+    TronCache<K, V> cache = new TronCache<>(name, CommonParameter.getInstance()
+        .getStorage().getCacheStrategy(name));
+    CACHES.put(name, cache);
+    return cache;
+  }
+
+  public  static <K, V> TronCache<K, V> allocate(String name, String strategy) {
+    TronCache<K, V> cache = new TronCache<>(name, strategy);
+    CACHES.put(name, cache);
+    return cache;
+  }
+
+  public  static <K, V> TronCache<K, V> allocate(String name, String strategy,
+                                                 CacheLoader<K, V> loader) {
+    TronCache<K, V> cache = new TronCache<>(name, strategy, loader);
+    CACHES.put(name, cache);
+    return cache;
+  }
+
+  public static void release(String name) {
+    TronCache cache = CACHES.remove(name);
+    if (cache != null) {
+      cache.invalidateAll();
+    }
+  }
+
+  public static Map<String, CacheStats> stats() {
+    return CACHES.values().stream().collect(Collectors.toMap(TronCache::getName, TronCache::stats));
+  }
+
+}