用java + netty 实现一个聊天室功能

首先，需要了解什么是Netty。Netty是一个异步事件驱动的网络应用框架，用于快速开发可维护的高性能协议服务器和客户端。

接下来，我们可以开始实现聊天室功能。一个简单的聊天室应该具备以下功能：

1. 用户连接和断开连接的处理；
2. 用户发送消息和接收消息的处理；
3. 消息广播给所有在线用户。

下面是一个简单的实现：

1. 用户连接和断开连接的处理

Netty提供了ChannelHandlerAdapter和ChannelInboundHandlerAdapter两个抽象类，我们可以继承其中一个来实现自己的Handler。这里我们使用ChannelInboundHandlerAdapter。

public class ChatServerHandler extends ChannelInboundHandlerAdapter {
    // 用户列表，用于保存所有连接的用户
    private static List<Channel> channels = new ArrayList<>();

    // 新用户连接时，将连接加入用户列表
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        channels.add(ctx.channel());
        System.out.println(ctx.channel().remoteAddress() + " 上线了");
    }

    // 用户断开连接时，将连接从用户列表中移除
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        channels.remove(ctx.channel());
        System.out.println(ctx.channel().remoteAddress() + " 下线了");
    }
}

2. 用户发送消息和接收消息的处理

Netty的数据传输是通过ByteBuf来实现的，因此我们需要将ByteBuf转换为字符串进行处理。

public class ChatServerHandler extends ChannelInboundHandlerAdapter {
    // 用户列表，用于保存所有连接的用户
    private static List<Channel> channels = new ArrayList<>();

    // 新用户连接时，将连接加入用户列表
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        channels.add(ctx.channel());
        System.out.println(ctx.channel().remoteAddress() + " 上线了");
    }

    // 用户断开连接时，将连接从用户列表中移除
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        channels.remove(ctx.channel());
        System.out.println(ctx.channel().remoteAddress() + " 下线了");
    }

    // 接收用户发送的消息并处理
    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        ByteBuf buf = (ByteBuf) msg;
        String received = buf.toString(CharsetUtil.UTF_8);
        System.out.println(ctx.channel().remoteAddress() + ": " + received);
        broadcast(ctx, received);
    }

    // 将消息广播给所有在线用户
    private void broadcast(ChannelHandlerContext ctx, String msg) {
        for (Channel channel : channels) {
            if (channel != ctx.channel()) {
                channel.writeAndFlush(Unpooled.copiedBuffer(msg, CharsetUtil.UTF_8));
            }
        }
    }
}

3. 消息广播给所有在线用户

我们可以使用broadcast方法将接收到的消息广播给所有在线用户。

public class ChatServerHandler extends ChannelInboundHandlerAdapter {
    // 用户列表，用于保存所有连接的用户
    private static List<Channel> channels = new ArrayList<>();

    // 新用户连接时，将连接加入用户列表
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        channels.add(ctx.channel());
        System.out.println(ctx.channel().remoteAddress() + " 上线了");
    }

    // 用户断开连接时，将连接从用户列表中移除
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        channels.remove(ctx.channel());
        System.out.println(ctx.channel().remoteAddress() + " 下线了");
    }

    // 接收用户发送的消息并处理
    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        ByteBuf buf = (ByteBuf) msg;
        String received = buf.toString(CharsetUtil.UTF_8);
        System.out.println(ctx.channel().remoteAddress() + ": " + received);
        broadcast(ctx, received);
    }

    // 将消息广播给所有在线用户
    private void broadcast(ChannelHandlerContext ctx, String msg) {
        for (Channel channel : channels) {
            if (channel != ctx.channel()) {
                channel.writeAndFlush(Unpooled.copiedBuffer(msg, CharsetUtil.UTF_8));
            }
        }
    }
}

接下来我们需要编写一个启动类，用于启动聊天室服务器。

public class ChatServer {
    public static void main(String[] args) throws Exception {
        // 创建EventLoopGroup
        EventLoopGroup bossGroup = new NioEventLoopGroup();
        EventLoopGroup workerGroup = new NioEventLoopGroup();

        try {
            // 创建ServerBootstrap
            ServerBootstrap serverBootstrap = new ServerBootstrap();
            serverBootstrap.group(bossGroup, workerGroup)
                    .channel(NioServerSocketChannel.class)
                    .childHandler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel ch) throws Exception {
                            ch.pipeline().addLast(new ChatServerHandler());
                        }
                    });

            // 启动服务器
            ChannelFuture channelFuture = serverBootstrap.bind(8888).sync();
            System.out.println("服务器启动成功");

            // 关闭服务器
            channelFuture.channel().closeFuture().sync();
        } finally {
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }
    }
}

现在，我们就完成了一个简单的聊天室服务器。可以通过运行ChatServer类启动服务器，然后使用telnet命令连接服务器进行聊天。

telnet localhost 8888

输入发送的消息，即可将消息广播给所有在线用户。