add_val_maxSize_for_revokingDB

Yrp · Yrp · commit df946d7f7dbc · 2018-04-12T17:06:29.000+08:00
diff --git a/src/main/java/org/tron/core/db/AbstractRevokingStore.java b/src/main/java/org/tron/core/db/AbstractRevokingStore.java
@@ -8,6 +8,7 @@
 import java.util.Map;
 import java.util.Objects;
 import java.util.Set;
+import java.util.concurrent.atomic.AtomicInteger;
 import lombok.AllArgsConstructor;
 import lombok.EqualsAndHashCode;
 import lombok.Getter;
@@ -26,6 +27,7 @@ public abstract class AbstractRevokingStore implements RevokingDatabase {
   private Deque<RevokingState> stack = new LinkedList<>();
   private boolean disabled = true;
   private int activeDialog = 0;
+  private AtomicInteger maxSize = new AtomicInteger(DEFAULT_STACK_MAX_SIZE);
 
   @Override
   public Dialog buildDialog() {
@@ -43,7 +45,7 @@ public synchronized Dialog buildDialog(boolean forceEnable) {
       disabled = false;
     }
 
-    while (stack.size() > DEFAULT_STACK_MAX_SIZE) {
+    while (stack.size() > maxSize.get()) {
       stack.poll();
     }
 
@@ -241,6 +243,14 @@ private void addIfEmtpy() {
     }
   }
 
+  public void setMaxSize(int maxSize) {
+    this.maxSize.set(maxSize);
+  }
+
+  public int getMaxSize() {
+    return maxSize.get();
+  }
+
   @Slf4j
   @Getter // only for unit test
   public static class Dialog implements AutoCloseable {
