fix(net):fix P2P message rate limit issue

Author: jwrct

framework/src/main/java/org/tron/core/net/P2pRateLimiter.java

@@ -6,21 +6,23 @@
 
 public class P2pRateLimiter {
   private final Cache<Byte, RateLimiter> rateLimiters = CacheBuilder.newBuilder()
-      .maximumSize(256).build();
+      .maximumSize(32).build();
 
-  public void register (Byte type, double rate) {
-    rateLimiters.put(type, RateLimiter.create(rate));
+  public void register(Byte type, double rate) {
+    RateLimiter rateLimiter = RateLimiter.create(Double.POSITIVE_INFINITY);
+    rateLimiter.setRate(rate);
+    rateLimiters.put(type, rateLimiter);
   }
 
-  public void acquire (Byte type) {
+  public void acquire(Byte type) {
     RateLimiter rateLimiter = rateLimiters.getIfPresent(type);
     if (rateLimiter == null) {
       return;
     }
     rateLimiter.acquire();
   }
 
-  public boolean tryAcquire (Byte type) {
+  public boolean tryAcquire(Byte type) {
     RateLimiter rateLimiter = rateLimiters.getIfPresent(type);
     if (rateLimiter == null) {
       return true;

framework/src/main/java/org/tron/core/net/peer/PeerConnection.java

@@ -173,7 +173,7 @@ public void setChannel(Channel channel) {
     this.nodeStatistics = TronStatsManager.getNodeStatistics(channel.getInetAddress());
     lastInteractiveTime = System.currentTimeMillis();
     p2pRateLimiter.register(SYNC_BLOCK_CHAIN.asByte(), 2);
-    p2pRateLimiter.register(FETCH_INV_DATA.asByte(), 1);
+    p2pRateLimiter.register(FETCH_INV_DATA.asByte(), 2);
     p2pRateLimiter.register(P2P_DISCONNECT.asByte(), 1);
   }
 

framework/src/test/java/org/tron/core/net/P2pRateLimiterTest.java

@@ -12,8 +12,10 @@ public void test() {
     P2pRateLimiter limiter = new P2pRateLimiter();
     limiter.register(SYNC_BLOCK_CHAIN.asByte(), 2);
     limiter.acquire(SYNC_BLOCK_CHAIN.asByte());
-    limiter.tryAcquire(SYNC_BLOCK_CHAIN.asByte());
     boolean ret = limiter.tryAcquire(SYNC_BLOCK_CHAIN.asByte());
+    Assert.assertTrue(ret);
+    limiter.tryAcquire(SYNC_BLOCK_CHAIN.asByte());
+    ret = limiter.tryAcquire(SYNC_BLOCK_CHAIN.asByte());
     Assert.assertFalse(ret);
     ret = limiter.tryAcquire(FETCH_INV_DATA.asByte());
     Assert.assertTrue(ret);

framework/src/test/java/org/tron/core/net/messagehandler/FetchInvDataMsgHandlerTest.java

@@ -1,5 +1,7 @@
 package org.tron.core.net.messagehandler;
 
+import static org.tron.core.net.message.MessageTypes.FETCH_INV_DATA;
+
 import com.google.common.cache.Cache;
 import com.google.common.cache.CacheBuilder;
 import java.lang.reflect.Field;
@@ -22,8 +24,6 @@
 import org.tron.core.net.service.adv.AdvService;
 import org.tron.protos.Protocol;
 
-import static org.tron.core.net.message.MessageTypes.FETCH_INV_DATA;
-
 public class FetchInvDataMsgHandlerTest {
 
   @Test