你好，我是彤哥，本篇是netty系列的第四篇。

欢迎来我的公从号彤哥读源码系统地学习源码&架构的知识。

简介

上一章我们一起学习了Java中的BIO/NIO/AIO的故事，本章将带着大家一起使用纯纯的NIO实现一个越聊越上瘾的“群聊系统”。

业务逻辑分析

首先，我们先来分析一下群聊的功能点：

（1）加入群聊，并通知其他人；

（2）发言，并通知其他人；

（3）退出群聊，并通知其他人；

一个简单的群聊系统差不多这三个功能足够了，为了方便记录用户信息，当用户加入群聊的时候自动给他分配一个用户ID。

业务实现

上代码：

// 这是一个内部类
private static class ChatHolder {
 // 我们只用了一个线程，用普通的HashMap也可以
 static final Map<SocketChannel, String> USER_MAP = new ConcurrentHashMap<>();

 /**
 * 加入群聊
 * @param socketChannel
 */
 static void join(SocketChannel socketChannel) {
 // 有人加入就给他分配一个id，本文来源于公从号“彤哥读源码”
 String userId = "用户"+ ThreadLocalRandom.current().nextInt(Integer.MAX_VALUE);
 send(socketChannel, "您的id为：" + userId + "\n\r");

 for (SocketChannel channel : USER_MAP.keySet()) {
 send(channel, userId + " 加入了群聊" + "\n\r");
 }

 // 将当前用户加入到map中
 USER_MAP.put(socketChannel, userId);
 }

 /**
 * 退出群聊
 * @param socketChannel
 */
 static void quit(SocketChannel socketChannel) {
 String userId = USER_MAP.get(socketChannel);
 send(socketChannel, "您退出了群聊" + "\n\r");
 USER_MAP.remove(socketChannel);

 for (SocketChannel channel : USER_MAP.keySet()) {
 if (channel != socketChannel) {
 send(channel, userId + " 退出了群聊" + "\n\r");
 }
 }
 }

 /**
 * 扩散说话的内容
 * @param socketChannel
 * @param content
 */
 public static void propagate(SocketChannel socketChannel, String content) {
 String userId = USER_MAP.get(socketChannel);
 for (SocketChannel channel : USER_MAP.keySet()) {
 if (channel != socketChannel) {
 send(channel, userId + ": " + content + "\n\r");
 }
 }
 }

 /**
 * 发送消息
 * @param socketChannel
 * @param msg
 */
 static void send(SocketChannel socketChannel, String msg) {
 try {
 ByteBuffer writeBuffer = ByteBuffer.allocate(1024);
 writeBuffer.put(msg.getBytes());
 writeBuffer.flip();
 socketChannel.write(writeBuffer);
 } catch (Exception e) {
 e.printStackTrace();
 }
 }
}

服务端代码

服务端代码直接使用上一章NIO的实现，只不过这里要把上面实现的业务逻辑适时地插入到相应的事件中。

（1）accept事件，即连接建立的时候，说明加入了群聊；

（2）read事件，即读取数据的时候，说明有人说话了；

（3）连接断开的时候，说明退出了群聊；

OK，直接上代码，为了与上一章的代码作区分，彤哥特意加入了一些标记：

public class ChatServer {
 public static void main(String[] args) throws IOException {
 Selector selector = Selector.open();
 ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
 serverSocketChannel.bind(new InetSocketAddress(8080));
 serverSocketChannel.configureBlocking(false);
 // 将accept事件绑定到selector上
 serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

 while (true) {
 // 阻塞在select上
 selector.select();
 Set<SelectionKey> selectionKeys = selector.selectedKeys();
 // 遍历selectKeys
 Iterator<SelectionKey> iterator = selectionKeys.iterator();
 while (iterator.hasNext()) {
 SelectionKey selectionKey = iterator.next();
 // 如果是accept事件
 if (selectionKey.isAcceptable()) {
 ServerSocketChannel ssc = (ServerSocketChannel) selectionKey.channel();
 SocketChannel socketChannel = ssc.accept();
 System.out.println("accept new conn: " + socketChannel.getRemoteAddress());
 socketChannel.configureBlocking(false);
 socketChannel.register(selector, SelectionKey.OP_READ);
 // 加入群聊，本文来源于公从号“彤哥读源码”
 ChatHolder.join(socketChannel);
 } else if (selectionKey.isReadable()) {
 // 如果是读取事件
 SocketChannel socketChannel = (SocketChannel) selectionKey.channel();
 ByteBuffer buffer = ByteBuffer.allocate(1024);
 // 将数据读入到buffer中
 int length = socketChannel.read(buffer);
 if (length > 0) {
 buffer.flip();
 byte[] bytes = new byte[buffer.remaining()];
 // 将数据读入到byte数组中
 buffer.get(bytes);

 // 换行符会跟着消息一起传过来
 String content = new String(bytes, "UTF-8").replace("\r\n", "");
 if (content.equalsIgnoreCase("quit")) {
 // 退出群聊，本文来源于公从号“彤哥读源码”
 ChatHolder.quit(socketChannel);
 selectionKey.cancel();
 socketChannel.close();
 } else {
 // 扩散，本文来源于公从号“彤哥读源码”
 ChatHolder.propagate(socketChannel, content);
 }
 }
 }
 iterator.remove();
 }
 }
 }
}

测试

打开四个XSHELL客户端，分别连接telnet 127.0.0.1 8080，然后就可以开始群聊了。

彤哥发现，自己跟自己聊天也是会上瘾的，完全停不下来，不行了，我再去自聊一会儿^^

总结

本文彤哥跟着大家一起实现了“群聊系统”，去掉注释也就100行左右的代码，是不是非常简单？这就是NIO网络编程的魅力，我发现写网络编程也上瘾了^^

问题

这两章我们都没有用NIO实现客户端，你知道怎么实现吗？

提示：服务端需要监听accept事件，所以需要有一个ServerSocketChannel，而客户端是直接去连服务器了，所以直接用SocketChannel就可以了，一个SocketChannel就相当于一个Connection。

最后，也欢迎来我的公从号彤哥读源码系统地学习源码&架构的知识。