原生Node 实现聊天室

痛定思痛。 2022-01-20 00:41 216阅读 0赞

1、最基础的聊天室代码（有注释）

/**
     * 1.创建一个服务器
     * 2. 客户端可以连接服务器
     * 3.客户端可以发言，然后广播给大家
     * 4.客户端连接和退出后都要通知大家。
     * 5.显示当前的在线人数
     */
    let net = require('net'); // 利用TCP协议实现，所以引入net模块，用telnet充当客户端
    let clients = {}; // 所有客户端的socket存在这个对象里
    // 创建TCP服务器
    let server = net.createServer(function (socket) {
        // 获取当前人数，也就是TCP链接数
        server.getConnections(function (err, count) {
            socket.write(`weclome,there is ${count} users now,please input your username\r\n`);
        })
        let nickname; // 每个人的名字
        setEncoding('utf8');// 编码设置为utf8
        socket.on('data', function (data) { // 接收数据，socket是既是可写流也是可读流，所以有data事件
            data = data.replace(/\r\n/, ''); // 将回车(windows回车是\r\n，linux是\n)替换
            if (data == 'byebye') { // 如果输入的字符是byebye则关闭这个socket链接
                socket.end();
            } else {
                if (nickname) { // 输入名字后，所有信息广播给其他人
                    broadcast(nickname, `${nickname}:${data}`);
                } else { // 第一次需要输入名字，并将这个人的sokcet存入clients对象，后面要用来广播
                    nickname = data;
                    clients[nickname] = socket;
                    broadcast(nickname, `welcome ${nickname} joined us!`);
                }
            }
    
        });
        socket.on('end', function () { // 用户推出销毁socket，因为socket数量是有限的
            clients[nickname].destroy()
            delete clients[nickname]
            broadcast(nickname, `welcome ${nickname} joined us!`);
        });
    }).listen(8088);
    // 广播方法，将除了自己的socket外，全部用socket发送数据
    function broadcast(nickname, msg) {
        for (let key in clients) {
            if (key != nickname) {
                clients[key].write(msg + '\r\n');
            }
        }
    }
    复制代码

2、增加私聊功能

// 添加私聊方法
    function sendTo(toUser, text, selfSocket){
        let userSocekt = clients[toUser]
        if(userSocekt){
            userSocekt.write(`${nickname}: ${text}`)
        }else{
            selfSocket.write(`用户名不正确或者对方已经下线`)
        }
    }
    // 同时创建net服务器改造一下，根据不同的需求，分类广播和私聊功能
    let server = net.createServer(function (socket) {
        // 获取当前人数，也就是TCP链接数
        server.getConnections(function (err, count) {
            socket.write(`weclome,there is ${count} users now,please input your username\r\n`);
        })
        let nickname; // 每个人的名字
        setEncoding('utf8');// 编码设置为utf8
        socket.on('data', function (data) { // 接收数据，socket是既是可写流也是可读流，所以有data事件
            // 发送数据格式为b: 发送内容，b代表要广播， s: 用户名：内容，s代表要私聊
            let type = data.slice(0) // 取出第一个字符知道要用私聊还是广播功能
            switch(type){
                case 'b':
                    let text = data.slice(2)
                    boardcast(text)
                    break;
                case 's':
                    let values = data.split(':')
                    let toUser = values[1]
                    let text = values[2]
                    sendTo(toUser, text, socket)
                    break;
                default:
                    socket.write('此命令不能识别, 请重新输入')
                    break;
            }
        });
        socket.on('end', function () { // 用户推出销毁socket，因为socket数量是有限的
            clients[nickname].destroy()
            delete clients[nickname]
            broadcast(nickname, `welcome ${nickname} joined us!`);
        });
    }).listen(8088);
    复制代码