NIO模擬粘問題的代碼具體實現以及詳細註釋

(1)項目結構

(2)服務端

package com.nio.caib.server;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;

public class NettyNioServer {

    public void bind(int port)throws Exception{
        // 配置服務端的NIO線程組,NioEventLoopGroup是個線程組,包含了一組NIO線程
        // 專門用於網路時間的處理,實際上它們就是Reactor線程組
        NioEventLoopGroup groupParent = new NioEventLoopGroup();
        NioEventLoopGroup groupChild = new NioEventLoopGroup();

        try {
            // 創建ServerBootstrap對象,它是Netty用於啓動Nio服務的輔助類啓動器
            // 目的是降低服務端的開發複雜度
            ServerBootstrap bootstrap = new ServerBootstrap();
            // 將兩個線程組當參數傳遞到ServerBootstrap中
            // 設置創建的channel爲NioServerSocketChannel
            // 配置NioServerSocketChannel的TCP參數,此處的backlog設置爲1024
            // 綁定IO事件的處理類ChildChannelHandler,用於處理網絡IO事件,例如記錄日誌,對消息進行編碼等
            bootstrap.group(groupParent,groupChild)
                    .channel(NioServerSocketChannel.class)
                    .option(ChannelOption.SO_BACKLOG, 1024)
                    .childHandler(new ChildChannelHandler());
            // 調用bind方法綁定端口,調用同步阻塞方法sync等待綁定操作成功
            // 返回值主要用於異步操作的通知回調
            ChannelFuture future = bootstrap.bind(port).sync();
            // 等待服務端監聽端口關閉
            future.channel().closeFuture().sync();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }finally {
            // 退出,釋放系統資源
            groupParent.shutdownGracefully();
            groupChild.shutdownGracefully();
        }
    }

    private class ChildChannelHandler extends ChannelInitializer<SocketChannel> {

        @Override
        protected void initChannel(SocketChannel socketChannel) throws Exception {
            socketChannel.pipeline().addLast(new NettyNioServerHandler());
        }
    }


    public static void main(String[] args)throws Exception {
        int port = 8084;
        if (args != null && args.length > 0){
            try {
                port = Integer.valueOf(args[0]);
            } catch (NumberFormatException e) {
                e.printStackTrace();
            }

        }
        new NettyNioServer().bind(port);

    }


}
package com.nio.caib.server;


import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;

import java.util.Date;

public class NettyNioServerHandler extends ChannelInboundHandlerAdapter{

    // 計數
    private int counter;

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        // 將msg轉換成Netty的ByteBuf對象
        ByteBuf buf = (ByteBuf) msg;
        // buf.readableBytes()獲取緩衝區可讀的字節數
        // 根據可讀的字節數創建新的數組
        byte[] req = new byte[buf.readableBytes()];
        // 將緩衝區的字節數組複製到新建的字節byte數組
        buf.readBytes(req);
        // 對這個字節數組進編碼
        String body = new String(req, "utf-8").substring(0, req.length - System.getProperty("line.separator").length());
        /**
         * 沒讀到一條客戶端發來的消息就記一次數,然後發送應答消息給客戶端
         * 服務端接受到的消息總個數應該和客戶端發送的消息總數相同
         */
        System.out.println("服務端收到客戶端發來的的消息是: " + body + " ;目前的計數結果是: " +  ++counter);

        // 判斷客戶端發來的消息和服務端預設值的消息是否相同
        // 如果相同就返回給客戶端當前的時間
        String str = "NOW TIME IS NOW";
        String nowTime = "NOTE RIGHT";
        if (str.trim().equals(body.trim())){
            nowTime = new Date().toString();
        }
        ByteBuf resp = Unpooled.copiedBuffer((nowTime + System.getProperty("line.separator")).getBytes());
        // 異步發送應答消息給客戶端
        ctx.writeAndFlush(resp);
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
       // 當發生異常的時候,關閉ChannelHandlerContext,釋放和ChannelHandlerContext相關聯的句柄等資源
        ctx.close();
    }
}

(3)客戶端

package com.nio.caib.client;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;

public class NettyNioClient {

    public void connect(int port, String host)throws Exception{
        // 配置服務端的NIO線程組,NioEventLoopGroup是個線程組,包含了一組NIO線程
        // 專門用於網路時間的處理,實際上它們就是Reactor線程組
        NioEventLoopGroup group = new NioEventLoopGroup();

        try {
            // 創建ServerBootstrap對象,它是Netty用於啓動Nio服務的輔助類啓動器
            // 目的是降低服務端的開發複雜度
            Bootstrap bootstrap = new Bootstrap();
            // 於服務端不同channel(NioSocketChannel.class)
            /**
             * handler,創建匿名內部類,實現initChannel方法,
             * 作用是當創建NioSocketChannel成功之後
             * 在進行初始化時,將它的channelHandler設置到ChannelPipeline中,
             * 用於處理網絡IO事件
             */
            bootstrap.group(group)
                    .channel(NioSocketChannel.class)
                    .option(ChannelOption.TCP_NODELAY, true)
                    .handler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel socketChannel) throws Exception {
                            socketChannel.pipeline().addLast(new NettyClientHandler());
                        }
                    });
            // 發起異步連接操作,調用同步阻塞方法等待連接成功
            ChannelFuture sync = bootstrap.connect(host, port).sync();
            // 等待客戶端鏈路關閉
            sync.channel().closeFuture().sync();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }finally {
            // 退出,釋放資源
            group.shutdownGracefully();
        }

    }

    public static void main(String[] args) throws Exception{
        int port = 8084;
        if (args != null && args.length > 0){
            try {
                port = Integer.valueOf(args[0]);
            } catch (NumberFormatException e) {
                e.printStackTrace();
            }

        }
        new NettyNioClient().connect(port,"localhost");
    }
}
package com.nio.caib.client;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;

import java.util.logging.Logger;

public class NettyClientHandler extends ChannelInboundHandlerAdapter {
    /**
     * 日誌
     */
    private static  final Logger logger = Logger.getLogger(NettyClientHandler.class.getName());

    // 計數
    private int counter;
    private byte[] req;

    public NettyClientHandler() {
        req = ("NOW TIME IS NOW" + System.getProperty("line.separator")).getBytes();
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        /**
         * 循環發送100條消息,每發送一條就刷新一次,保證每條消息都會被寫入channel中
         */
        ByteBuf message = null;
        for (int i = 0; i < 100; i++) {
            message = Unpooled.buffer(req.length);
            message.writeBytes(req);
            ctx.writeAndFlush(message);
        }
    }

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        ByteBuf buf = (ByteBuf) msg;
        // buf.readableBytes()獲取緩衝區可讀的字節數
        // 根據可讀的字節數創建新的數組
        byte[] req = new byte[buf.readableBytes()];
        // 將緩衝區的字節數組複製到新建的字節byte數組
        buf.readBytes(req);
        // 編碼
        String body = new String(req, "utf-8");
        // 打印服務端返回的消息
        /**
         * 客戶端每接收到服務端一條應答消息之後,就打印一次計數器
         */
        System.out.println("客戶端收到服務端返回的消息是: " + body + " ;計數器的計數結果是: " + ++counter);
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        // 釋放資源
        logger.warning("不期而遇的異常:" + cause.getMessage());
        ctx.close();
    }
}

(4)運行結果:

客戶端

服務端

 

NIO解決粘包問題的代碼具體實現以及詳細註釋URL:

https://blog.csdn.net/wildwolf_001/article/details/81136981

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章