remove reasoncode class

Author: lvs007

src/main/java/org/tron/common/overlay/client/PeerClient.java

@@ -18,10 +18,10 @@
 import org.springframework.stereotype.Component;
 import org.tron.common.overlay.discover.Node;
 import org.tron.common.overlay.discover.NodeHandler;
-import org.tron.common.overlay.message.ReasonCode;
 import org.tron.common.overlay.server.TronChannelInitializer;
 import org.tron.core.config.args.Args;
 import org.tron.core.net.node.NodeImpl;
+import org.tron.protos.Protocol.ReasonCode;
 
 @Component
 public class PeerClient {

src/main/java/org/tron/common/overlay/discover/NodeStatistics.java

@@ -21,7 +21,7 @@
 import static java.lang.Math.min;
 
 import java.util.concurrent.atomic.AtomicLong;
-import org.tron.common.overlay.message.ReasonCode;
+import org.tron.protos.Protocol.ReasonCode;
 
 public class NodeStatistics {
 

src/main/java/org/tron/common/overlay/message/DisconnectMessage.java

@@ -1,8 +1,8 @@
 package org.tron.common.overlay.message;
 
-import com.google.protobuf.InvalidProtocolBufferException;
 import org.tron.core.net.message.MessageTypes;
 import org.tron.protos.Protocol;
+import org.tron.protos.Protocol.ReasonCode;
 
 public class DisconnectMessage extends P2pMessage {
 
@@ -16,7 +16,7 @@ public DisconnectMessage(byte type, byte[] rawData) throws Exception{
   public DisconnectMessage(ReasonCode reasonCode) {
     this.disconnectMessage = Protocol.DisconnectMessage
         .newBuilder()
-        .setReason(Protocol.ReasonCode.forNumber(reasonCode.getReason()))
+        .setReason(reasonCode)
         .build();
     this.type = MessageTypes.P2P_DISCONNECT.asByte();
     this.data = this.disconnectMessage.toByteArray();
@@ -26,6 +26,10 @@ public int getReason() {
     return this.disconnectMessage.getReason().getNumber();
   }
 
+  public ReasonCode getReasonCode() {
+    return disconnectMessage.getReason();
+  }
+
   @Override
   public String toString() {
     return new StringBuilder().append(super.toString()).append("reason: ").append(this.disconnectMessage.getReason()).toString();

src/main/java/org/tron/common/overlay/message/ReasonCode.java

@@ -36,12 +36,12 @@
 import org.tron.common.overlay.message.DisconnectMessage;
 import org.tron.common.overlay.message.HelloMessage;
 import org.tron.common.overlay.message.MessageCodec;
-import org.tron.common.overlay.message.ReasonCode;
 import org.tron.common.overlay.message.StaticMessages;
 import org.tron.core.db.ByteArrayWrapper;
 import org.tron.core.exception.P2pException;
 import org.tron.core.net.peer.PeerConnectionDelegate;
 import org.tron.core.net.peer.TronHandler;
+import org.tron.protos.Protocol.ReasonCode;
 
 @Component
 @Scope("prototype")

src/main/java/org/tron/common/overlay/server/Channel.java

@@ -17,8 +17,8 @@
  */
 package org.tron.common.overlay.server;
 
-import static org.tron.common.overlay.message.ReasonCode.DUPLICATE_PEER;
-import static org.tron.common.overlay.message.ReasonCode.TOO_MANY_PEERS;
+import static org.tron.protos.Protocol.ReasonCode.DUPLICATE_PEER;
+import static org.tron.protos.Protocol.ReasonCode.TOO_MANY_PEERS;
 
 import java.net.InetAddress;
 import java.net.InetSocketAddress;
@@ -36,9 +36,9 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.tron.common.overlay.client.PeerClient;
-import org.tron.common.overlay.message.ReasonCode;
 import org.tron.core.config.args.Args;
 import org.tron.core.db.ByteArrayWrapper;
+import org.tron.protos.Protocol.ReasonCode;
 
 
 @Component

src/main/java/org/tron/common/overlay/server/ChannelManager.java

@@ -34,10 +34,10 @@
 import org.tron.common.overlay.message.HelloMessage;
 import org.tron.common.overlay.message.P2pMessage;
 import org.tron.common.overlay.message.P2pMessageFactory;
-import org.tron.common.overlay.message.ReasonCode;
 import org.tron.core.config.args.Args;
 import org.tron.core.db.Manager;
 import org.tron.core.net.peer.PeerConnection;
+import org.tron.protos.Protocol.ReasonCode;
 
 @Component
 @Scope("prototype")
@@ -92,7 +92,8 @@ protected void decode(ChannelHandlerContext ctx, ByteBuf buffer, List<Object> ou
         break;
       case P2P_DISCONNECT:
         if (channel.getNodeStatistics() != null){
-          channel.getNodeStatistics().nodeDisconnectedRemote(ReasonCode.fromInt(((DisconnectMessage) msg).getReason()));
+          channel.getNodeStatistics()
+              .nodeDisconnectedRemote(((DisconnectMessage) msg).getReasonCode());
         }
         channel.close();
         break;

src/main/java/org/tron/common/overlay/server/HandshakeHandler.java

@@ -16,7 +16,7 @@
 import org.springframework.stereotype.Component;
 import org.tron.common.overlay.message.Message;
 import org.tron.common.overlay.message.PingMessage;
-import org.tron.common.overlay.message.ReasonCode;
+import org.tron.protos.Protocol.ReasonCode;
 
 @Component
 @Scope("prototype")

src/main/java/org/tron/common/overlay/server/MessageQueue.java

@@ -30,7 +30,6 @@
 import org.springframework.stereotype.Component;
 import org.tron.common.overlay.message.DisconnectMessage;
 import org.tron.common.overlay.message.P2pMessage;
-import org.tron.common.overlay.message.ReasonCode;
 
 @Component
 @Scope("prototype")
@@ -77,7 +76,8 @@ public void channelRead0(final ChannelHandlerContext ctx, P2pMessage msg) throws
         channel.getPeerStats().pong(sendPingTime);
         break;
       case P2P_DISCONNECT:
-        channel.getNodeStatistics().nodeDisconnectedRemote(ReasonCode.fromInt(((DisconnectMessage) msg).getReason()));
+        channel.getNodeStatistics()
+            .nodeDisconnectedRemote(((DisconnectMessage) msg).getReasonCode());
         channel.close();
         break;
       default:

src/main/java/org/tron/common/overlay/server/P2pHandler.java

@@ -39,7 +39,6 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import org.tron.common.overlay.message.Message;
-import org.tron.common.overlay.message.ReasonCode;
 import org.tron.common.overlay.server.Channel.TronState;
 import org.tron.common.overlay.server.SyncPool;
 import org.tron.common.utils.ExecutorLoop;
@@ -71,6 +70,7 @@
 import org.tron.core.net.peer.PeerConnectionDelegate;
 import org.tron.protos.Protocol;
 import org.tron.protos.Protocol.Inventory.InventoryType;
+import org.tron.protos.Protocol.ReasonCode;
 import org.tron.protos.Protocol.Transaction;
 
 @Slf4j