1、优化TCP发送

sec-beidou-fwd/src/main/java/com/imdroid/beidou_fwd/service/TCPClient.java

@@ -22,10 +22,10 @@ public class TCPClient {
     //@Value("${xfz.server.port}")
     private int port;
 
-
     private Bootstrap bootstrap;
     private EventLoopGroup group;
     private Channel channel;
+    ByteBuf sendBuffer;
 
     public void start() {
         new Thread(this::connect, "xfz-tcp-client").start();
@@ -34,6 +34,7 @@ public class TCPClient {
     public void init(String dest_addr, int dest_port) {
         host = dest_addr;
         port = dest_port;
+        sendBuffer = Unpooled.buffer();
         //客户端需要一个事件循环组
         group = new NioEventLoopGroup();
         //创建客户端启动对象
@@ -45,7 +46,7 @@ public class TCPClient {
                     @Override
                     protected void initChannel(SocketChannel ch) throws Exception {
                         //加入处理器
-                        ch.pipeline().addLast(new TcpMessageHandler());
+                        ch.pipeline().addLast(new TcpMessageHandler(TCPClient.this));
                     }
                 });
     }
@@ -81,21 +82,26 @@ public class TCPClient {
         }
     }
 
-
     public void writeAndFlush(String json) {
-        String str = "#" + json + "!";
+        sendBuffer.writeBytes(json.getBytes(StandardCharsets.UTF_8));
-        ByteBuf buf = Unpooled.buffer();
+        if(channel.isActive()) flush();
-        buf.writeBytes(str.getBytes(StandardCharsets.UTF_8));
+        else connect();
-        channel.writeAndFlush(buf).addListener(future -> {
-            if (future.isSuccess()) {
-                logger.info("send {} to xfz server succeed.", str);
-            } else {
-                logger.info("send {} to xfz server failed.", str);
-            }
-        });
     }
 
-    public boolean isActive(){
+    public void onConnectionActive(){
-        return channel.isActive();
+        flush();
+    }
+
+    private void flush(){
+        if(sendBuffer.readableBytes()>0){
+            channel.writeAndFlush(sendBuffer).addListener(future -> {
+                if (future.isSuccess()) {
+                    logger.info("send to xfz server succeed.");
+                    sendBuffer.clear();
+                } else {
+                    logger.info("send to xfz server failed.");
+                }
+            });
+        }
     }
 }

sec-beidou-fwd/src/main/java/com/imdroid/beidou_fwd/service/TcpMessageHandler.java

@@ -15,6 +15,12 @@ public class TcpMessageHandler extends SimpleChannelInboundHandler<ByteBuf> {
 
     private final Logger logger = LoggerFactory.getLogger(this.getClass());
 
+    private final TCPClient tcpClient;
+
+    public TcpMessageHandler(TCPClient tcpClient) {
+        this.tcpClient = tcpClient;
+    }
+
     @Override
     protected void channelRead0(ChannelHandlerContext channelHandlerContext, ByteBuf buf) throws Exception {
         if (logger.isDebugEnabled()) {
@@ -27,6 +33,7 @@ public class TcpMessageHandler extends SimpleChannelInboundHandler<ByteBuf> {
     @Override
     public void channelActive(ChannelHandlerContext ctx) throws Exception {
         logger.info("xfz tcp channel active");
+        tcpClient.onConnectionActive();
     }
 
     @Override

sec-beidou-fwd/src/main/java/com/imdroid/beidou_fwd/task/GXXfzForwarder.java

@@ -82,26 +82,11 @@ public class GXXfzForwarder extends Forwarder{
             data.setZ(NumberUtils.scale(locationRecord.getRb562d() * 0.001, 5));
             sendNum++;
         }
-        String json = GsonUtil.toJson(xfzTcpMessage);
+        String json = "#" + GsonUtil.toJson(xfzTcpMessage) + "!";
         logger.info("project " + projectId + ": push calculation result to XFZ");
         logger.info(json);
-        if(!xfzTcpClient.isActive()) {
+        xfzTcpClient.writeAndFlush(json);
-            xfzTcpClient.connect();
+        return sendNum;
-            try {
-                Thread.sleep(5000);
-            }
-            catch (Exception e){
-
-            }
-        }
-        if(xfzTcpClient.isActive()) {
-            xfzTcpClient.writeAndFlush(json);
-            return sendNum;
-        }
-        else{
-            logger.info("TCP disconnected!");
-            return 0;
-        }
     }
 
 }