Update AdvService.java

hujunbiao · Sep 9, 2020 · b2d44da · b2d44da
1 parent 6163b91
commit b2d44da
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/framework/src/main/java/org/tron/core/net/service/AdvService.java b/framework/src/main/java/org/tron/core/net/service/AdvService.java
@@ -39,6 +39,11 @@
 @Slf4j(topic = "net")
 @Component
 public class AdvService {
+
+  private final int MAX_INV_TO_FETCH_CACHE_SIZE = 100_000;
+  private final int MAX_TRX_CACHE_SIZE = 50_000;
+  private final int MAX_BLOCK_CACHE_SIZE = 10;
+  private final int MAX_SPREAD_SIZE = 1_000;
 
   @Autowired
   private TronNetDelegate tronNetDelegate;
@@ -48,13 +53,13 @@ public class AdvService {
   private ConcurrentHashMap<Item, Long> invToSpread = new ConcurrentHashMap<>();
 
   private Cache<Item, Long> invToFetchCache = CacheBuilder.newBuilder()
-      .maximumSize(100_000).expireAfterWrite(1, TimeUnit.HOURS).recordStats().build();
+      .maximumSize(MAX_INV_TO_FETCH_CACHE_SIZE).expireAfterWrite(1, TimeUnit.HOURS).recordStats().build();
 
   private Cache<Item, Message> trxCache = CacheBuilder.newBuilder()
-      .maximumSize(50_000).expireAfterWrite(1, TimeUnit.HOURS).recordStats().build();
+      .maximumSize(MAX_TRX_CACHE_SIZE).expireAfterWrite(1, TimeUnit.HOURS).recordStats().build();
 
   private Cache<Item, Message> blockCache = CacheBuilder.newBuilder()
-      .maximumSize(10).expireAfterWrite(1, TimeUnit.MINUTES).recordStats().build();
+      .maximumSize(MAX_BLOCK_CACHE_SIZE).expireAfterWrite(1, TimeUnit.MINUTES).recordStats().build();
 
   private ScheduledExecutorService spreadExecutor = Executors.newSingleThreadScheduledExecutor();
 
@@ -63,8 +68,6 @@ public class AdvService {
   @Getter
   private MessageCount trxCount = new MessageCount();
 
-  private int maxSpreadSize = 1_000;
-
   private boolean fastForward = Args.getInstance().isFastForward();
 
   public void init() {
@@ -144,7 +147,7 @@ public void broadcast(Message msg) {
       return;
     }
 
-    if (invToSpread.size() > maxSpreadSize) {
+    if (invToSpread.size() > MAX_SPREAD_SIZE) {
       logger.warn("Drop message, type: {}, ID: {}.", msg.getType(), msg.getMessageId());
       return;
     }