4. 彤哥说netty系列之Java NIO实现群聊(自己跟自己聊上瘾了)
2019-11-20 00:03
1541 查看
你好,我是彤哥,本篇是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。
最后,也欢迎来我的公从号彤哥读源码系统地学习源码&架构的知识。
相关文章推荐
- 看透SpringMVC系列(二)用NIO自己手动实现HTTP协议
- java nio&netty系列之二reactor模型基础
- Netty学习之旅------线程模型前置篇Reactor反应堆设计模式实现(基于java.nio)
- 3. 彤哥说netty系列之Java BIO NIO AIO进化史
- 自己动手实现FleaJAVA框架 -- vb2005xu自己动手系列
- java网络socket编程(七)之java中NIO实现聊天系统的群聊功能
- LIS最长递增子系列nlogn算法Java实现
- 自己琢磨的 图像相似度算法 JAVA版实现
- 关于java nio在windows下实现
- 用java代码实现一个自己的栈.
- Java 语义网编程系列五: Jena+sparql实现语义数据查询
- 使用BlazeDS实现java与flex传对象时,RemoteClass中包含自己写的类,后台无法转换的问题
- 自己总结的Java实现文件的读写操作
- API入门系列之六 -自己实现MessageBox
- 黑莓开发学习入门系列,自己动手实现一个日历软件(二)
- 用Annotation,Future,动态代理实现自己的JAVA方法运行超时应用
- Java实现文件的复制和新Nio包通道的运用--Thinking in java
- 自己用java代码实现的:“约瑟夫”问题
- Java程序员从笨鸟到菜鸟之(三十五)细谈struts2(一)自己实现struts2框架
- 深入Java集合学习系列:LinkedHashMap的实现原理