java - Netty writeAndFlush() 不完整

Question

我的 tcp 客户端请求 netty 服务器，netty 服务器使用 writeAndFlush() 返回 393718 字节。但客户端只收到 262142 字节。我使用“tcpdump -A”来抓取数据包，也小于 393718。所以我认为正确的是在 netty writeAndFlush() 函数中？

这是代码

TCP 服务器:

public static void main (String args[]) {

        int processorsNumber = Runtime.getRuntime().availableProcessors() * 3;
        ThreadFactory threadFactory = new DefaultThreadFactory("work thread pool");
        EventLoopGroup bossLoopGroup = new NioEventLoopGroup(1);
        EventLoopGroup workLoopGroup = new NioEventLoopGroup(processorsNumber, threadFactory, SelectorProvider.provider());

        try {
            ServerBootstrap sb = new ServerBootstrap();
            sb.group(bossLoopGroup , workLoopGroup)
                    .channel(NioServerSocketChannel.class)
                    .option(ChannelOption.SO_BACKLOG, 128)
                    .childOption(ChannelOption.SO_KEEPALIVE, true)
                    .childHandler(new ChannelInitializer<NioSocketChannel>() {
                        @Override
                        protected void initChannel(NioSocketChannel ch) throws Exception {
                            //ch.pipeline().addLast(new LineBasedFrameDecoder(1000));
                            ch.pipeline().addLast(new TcpServerHandler());
                        }
                    });

            sb.bind(new InetSocketAddress("0.0.0.0", 18881)).sync().channel().closeFuture().sync();
        } catch (InterruptedException e) {
            logger.error("netty error", e);
            e.printStackTrace();
        } finally {
            bossLoopGroup.shutdownGracefully();
            workLoopGroup.shutdownGracefully();
        }
    }

服务器处理程序:

public void channelRead(ChannelHandlerContext ctx, Object msg) {

        ByteBuf in = (ByteBuf) msg;
        String cmd = in.toString(CharsetUtil.UTF_8);
        logger.info(cmd);

        String retCode = "";
        String file = "E:\\sz\\app.log";
        try {
            FileReader fr = new FileReader(file);
            BufferedReader br = new BufferedReader(fr);
            String buffer = null;
            while ((buffer = br.readLine()) != null) {
                retCode += buffer;
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }

        logger.info("return：" + retCode);
        byte[] result = retCode.getBytes();
        logger.info("======================="+result.length);
        ByteBuf resultBuf = Unpooled.buffer(result.length);
        resultBuf.writeBytes(result);
        ctx.writeAndFlush(resultBuf);

        ctx.close();
    }

客户端:

public static void main(String[] args) {
        StringBuilder response = new StringBuilder();
        try {
            Socket socket = new Socket();
            InetSocketAddress address = new InetSocketAddress("127.0.0.1", 18881);
            socket.connect(address, 3000);
            socket.setSoTimeout(10 * 1000);

            OutputStream clientRequest = socket.getOutputStream();
            String cmd = "cmd";
            clientRequest.write(cmd.getBytes());
            clientRequest.flush();

            InputStream clientResponse = socket.getInputStream();
            int maxLen = 1024;
            byte[] contextBytes = new byte[maxLen];
            int readLen;
            while ((readLen = clientResponse.read(contextBytes, 0, maxLen)) != -1) {
                response.append(new String(contextBytes, 0, readLen));
            }

            clientRequest.close();
            clientResponse.close();
        } catch (IOException e) {
            logger.error("tcp error", e);
            e.printStackTrace();
        }
        String result = response.toString();
        System.out.println(result);
        System.out.println(result.getBytes().length);
    }

Answer 1

你写完直接关闭连接，在异步框架中，这是自讨苦吃。

这里发生的事情是，一条新的传入消息通过 Netty 线程到达您的 channelRead。然后，此消息通过您的处理程序进行处理，最后，使用 writeAndFlush 将其发回，但是，发送回数据的调用部分发生在后台，因此代码继续向前，然后它看到关闭连接的调用，它很乐意在这一点上这样做，因为你告诉它。

此时发生的是尝试发送数据的 TCP 层与接收关闭连接的调用的 TCP 层之间的竞争条件，因为在各层的不同部分使用了缓冲区，所以发生了这种情况限制为 262142 字节，但是如果连接的未知方面发生变化，此限制可能会波动。

这个问题的一个解决方案是等到 Netty 说写成功了，然后才关闭它。由于我们不想在异步应用程序中等待，这意味着我们必须注册一个监听器，在这种情况下，幸运就在我们这边，netty 为我们提供了一个监听器，它只是为我们关闭了 channel 。你可以像这样使用它:

ctx.writeAndFlush(resultBuf).addListener(ChannelFutureListener.CLOSE);