Netty实现客户端登录案例

Netty实现客户端登录案例

这个小案例主要是实现netty登录,分为客户端,及服务端,以及包含了序列化,编解码等相关内容,主要的代码会放在最后 gitHub 地址上,有需要可以看一下
首先来简单说明一下登录流程

1.客户端构建登录请求对象,对对象进行编码写入到ByteBuf对象中
2.服务端接收到对象后,进行解码,并对解码后的对象校验
3.服务端校验通过后,构建登录响应对象,重新对这个对象进行编码,写回到客户端中
4.客户端接收到响应数据,解码,获取响应对象,处理对应逻辑

一 : 构建客户端,服务端
服务端 :

public static void main(String[] args) {
        //接收新连接的线程组
        NioEventLoopGroup bossGroup = new NioEventLoopGroup();
        //处理每一个连接的数据读写线程组
        NioEventLoopGroup workerGroup = new NioEventLoopGroup();

        //引导服务端的启动工作
        ServerBootstrap bootstrap = new ServerBootstrap();

        bootstrap
                //引导类配置线程组
                .group(bossGroup,workerGroup)
                //服务端IO模型为Nio
                .channel(NioServerSocketChannel.class)
                .childHandler(new ChannelInitializer<NioSocketChannel>() {
                    @Override
                    protected void initChannel(NioSocketChannel ch) throws Exception {
                        ch.pipeline().addLast(new ServerHandler());
                    }
                })
                //绑定监听端口
                .bind(8000);
    }

客户端 :

public static void main(String[] args) {
        //
        NioEventLoopGroup workerGroup = new NioEventLoopGroup();

        Bootstrap bootstrap = new Bootstrap();
        bootstrap
                .group(workerGroup)
                .channel(NioSocketChannel.class)
                .handler(new ChannelInitializer<SocketChannel>() {
                    @Override
                    protected void initChannel(SocketChannel sc) throws Exception {
                        sc.pipeline().addLast(new ClientHandler());
                    }
                });

        bootstrap.connect("127.0.0.1",8000);
    }

二 . 客户端服务端逻辑处理
服务端逻辑处理对象 :

public class ServerHandler extends ChannelInboundHandlerAdapter {
    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        ByteBuf byteBuf = (ByteBuf) msg;

        //解码
        Packet packet = PacketCodeC.INSTANCE.decode(byteBuf);

        //判断是否是登录请求数据包
        if (packet instanceof LoginRequestPacket) {
            LoginRequestPacket loginRequestPacket = (LoginRequestPacket) packet;
            loginRequestPacket.setVersion(packet.getVersion());
            //登录校验
            if (valid(loginRequestPacket)) {
                //校验成功
                loginRequestPacket.setSuccess(Boolean.TRUE);
                System.out.println("登录成功");
            } else {
                //校验失败
                System.out.println("登录失败");
                loginRequestPacket.setReason("账号密码校验失败");
                loginRequestPacket.setSuccess(Boolean.FALSE);
            }

            //编码
            ByteBuf responseByteBuf = PacketCodeC.INSTANCE.encode(ctx, loginRequestPacket);

            ctx.channel().writeAndFlush(responseByteBuf);

        }
    }

    private boolean valid(LoginRequestPacket loginRequestPacket) {
        return true;
    }
}

客户端逻辑处理对象 :

public class ClientHandler extends ChannelInboundHandlerAdapter {
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(new Date() + "客户端开始登录");

        //创建登录对象
        LoginRequestPacket requestPacket = new LoginRequestPacket();
        requestPacket.setUserId(UUID.randomUUID().toString());
        requestPacket.setUserName("张三老师");
        requestPacket.setPassword("123");

        //编码
        ByteBuf byteBuf = PacketCodeC.INSTANCE.encode(ctx,requestPacket);

        ctx.channel().writeAndFlush(byteBuf);
    }

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        System.out.println("客户端接收服务端返回消息");
        ByteBuf byteBuf = (ByteBuf) msg;

        //解码
        Packet packet = PacketCodeC.INSTANCE.decode(byteBuf);

        if (packet instanceof LoginRequestPacket) {
            LoginRequestPacket responsePacket = (LoginRequestPacket) packet;

            if (responsePacket.getSuccess().equals(Boolean.TRUE)) {
                System.out.println(new Date() + ": 客户端登录成功");
            }else {
                System.out.println(new Date() + ": 客户端登录失败,失败原因 : " +
                        responsePacket.getReason());
            }
        }
    }
}

源码地址 : https://github.com/xiatiandao/Netty-IMServer/tree/master