【原创】nodejs 实现简单功能基本完整的 WebSocket

实现了加入房间，群发消息功能，没做私聊。

const WS = require("ws").Server;
var chat = new ChatRoom;
new WS({ port: 8080 }).on("connection", ws => {
        var user = new ChatUser(ws);
        ws.on("message", res => user.onMsg(res));
        ws.on("close", res => user.exit(res));
        ws.on("error", err => console.log(err));
});
console.log("WebSocket 服务侦听于端口 8080");

function ChatRoom() {
        this.join = (user, roomName) => {
                let room = rooms[roomName] || { name: roomName, id: 0, users: new Object, count: 0 };
                if(!room.id) rooms[roomName] = room;
                user.id = ++room.id; room.count++;
                user.room = room;
                room.users[user.id] = user;
                return user.id;
        };

this.kick = user => {
                if(!user.room) return;
                let room = user.room;
                room.count--;
                delete user.room;
                delete room.users[user.id];
                if(room.count) return;
                delete rooms[ room.name ];
                console.log(`房间 ${room.name} 被销毁`);
        };
        var rooms = new Object;
}

this.send = msg => {
                if(ws.readyState != 1) return console.log("暂时无法发送", ws.readyState);
                ws.send(JSON.stringify(msg));
        };

this.sendAll = msg => {
                var room = this.room; if(!room) return;
                for(var x in room.users) room.users[x].send(msg);
        };

this.exit = () => ws.readyState != 3 ? ws.close() : chat.kick(this);

this.doJoin = data => {
                data.id = chat.join(this, data.room);
                this.send({ type: "join", data: data.id });
                this.sendAll({ type: "welcome", data });
        };
}

[您可以先修改代码再运行]