feat(net):optimize rate limit logic

Author: jwrct

framework/src/main/java/org/tron/core/net/messagehandler/SyncBlockChainMsgHandler.java

@@ -31,13 +31,6 @@ public void processMessage(PeerConnection peer, TronMessage msg) throws P2pExcep
 
     SyncBlockChainMessage syncBlockChainMessage = (SyncBlockChainMessage) msg;
 
-    if (!peer.getP2pRateLimiter().tryAcquire(msg.getType().asByte())) {
-      // Discard messages that exceed the rate limit
-      logger.warn("{} message from peer {} exceeds the rate limit",
-          msg.getType(), peer.getInetSocketAddress());
-      return;
-    }
-
     if (!check(peer, syncBlockChainMessage)) {
       peer.disconnect(Protocol.ReasonCode.BAD_PROTOCOL);
       return;
@@ -65,6 +58,14 @@ public void processMessage(PeerConnection peer, TronMessage msg) throws P2pExcep
   }
 
   private boolean check(PeerConnection peer, SyncBlockChainMessage msg) throws P2pException {
+    if (peer.getRemainNum() > 0
+        && !peer.getP2pRateLimiter().tryAcquire(msg.getType().asByte())) {
+      // Discard messages that exceed the rate limit
+      logger.warn("{} message from peer {} exceeds the rate limit",
+          msg.getType(), peer.getInetSocketAddress());
+      return false;
+    }
+
     List<BlockId> blockIds = msg.getBlockIds();
     if (CollectionUtils.isEmpty(blockIds)) {
       throw new P2pException(TypeEnum.BAD_MESSAGE, "SyncBlockChain blockIds is empty");