欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

4. 彤哥说netty系列之Java NIO实现群聊(自己跟自己聊上瘾了)

程序员文章站 2022-04-22 20:06:58
你好,我是彤哥,本篇是netty系列的第四篇。 欢迎来我的公从号 彤哥读源码 系统地学习 源码&架构 的知识。 简介 上一章我们一起学习了Java中的BIO/NIO/AIO的故事,本章将带着大家一起使用纯纯的NIO实现一个越聊越上瘾的“群聊系统”。 业务逻辑分析 首先,我们先来分析一下群聊的功能点: ......

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,然后就可以开始群聊了。

4. 彤哥说netty系列之Java NIO实现群聊(自己跟自己聊上瘾了)

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

总结

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

问题

这两章我们都没有用nio实现客户端,你知道怎么实现吗?

提示:服务端需要监听accept事件,所以需要有一个serversocketchannel,而客户端是直接去连服务器了,所以直接用socketchannel就可以了,一个socketchannel就相当于一个connection。

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

4. 彤哥说netty系列之Java NIO实现群聊(自己跟自己聊上瘾了)