原生Node 實現聊天室

lio-mengxiang發表於2019-03-12

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);
複製程式碼

相關文章