4. 彤哥说netty系列之Java NIO实现群聊(自己跟自己聊上瘾了)
你好,我是彤哥,本篇是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。
最后,也欢迎来我的公从号彤哥读源码系统地学习源码&架构的知识。
上一篇: 苹果削皮后一天都没变色,是要这样做
下一篇: 腾讯:视频会议开美颜 男人占了近一半