原生Node 实现聊天室

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);
复制代码