使用socket.io实现多房间通信聊天室

websocket的实现有很多种，像ws和socket.io，这里使用的是socket.io来实现多房间的效果。

这里的使用没有使用socket.io官方提供的namespace和room，而是完全通过一个namespace实现的。数据传输使用JSON格式，封装了消息规范

消息体规范

const actionType = {

    join:'JOIN',//加入

    leave:'LEAVE',//离开

    talk:'TALK',//消息

    action:'ACTION',//用户操作

    push:'PUSH'//系统推送

}//消息体

class MSG {

    constructor(type,body){

        this.type = type;

        this.body= body;

    }
}复制代码

安装使用

npm install socket.io-rooms --save

demo演示

把项目从github上clone下来后，执行npm start,然后打开example/index.html即可品尝到演示效果

使用方式

服务端Server

const {User,Rooms}  = require('./index')

const server = require('http').createServer();

const io = require('socket.io')(server);

//大厅

io.on('connection', client => {

    let user = new User();

    client.emit('user',user);

    client.on('join', data => {

        /* 加入某个房间 */

        Rooms.join(data,user,io,client)

    });

    client.on('message',msg=>{

        if(user.roomId){

            // io.to(user.roomId).emit('message',msg)

            if(msg.type == 'update'){

                user.update(msg.body);

            }

            msg.user = user.uid;

            Rooms.send(user.roomId,msg)

        }else{

            io.emit('message',msg)

        }

        console.log(msg)

    })

    client.on('disconnect', () => {

        /* … */

        console.log("连接断开")

        Rooms.leave(user)

    });

});

server.listen();

这里传输统一使用`JSON`格式，消息`title`也以`message`为主,这里端口写的80，你可以使用其他端口，如果你是Express，也可以共用80端口。

客户端调用Client

       const socket = io('http://localhost');

        log =(...args)=>{

            document.getElementById('log').innerHTML +='<br/>'+args.map(item=>JSON.stringify(item)).join(' ')+'=>'+(+new Date());

        } 

        log(socket.id)

        let user ={},room,client;

        socket.on('connect', (c) => {

            log('connect ...', socket.id);

            socket.on('user',u=>{

                user = u;

                log('用户ID',u.uid)

            });

        });

        socket.on('message',msg=>{

            log('message:',msg)

        });

        function joinroom(num){

            //加入房间号为1的房间

            socket.emit('join',num);

        }

        function send(){

            let msg = document.getElementById('msg').value;

            socket.emit('message',{type:'TALK',body:msg})

            // setInterval(function(){

            //     socket.emit('message',{type:'TALK',body:+new Date()})

            // },2000)

        }

在用户信息上，为了增加扩展性，添加了update的操作类型进行同步用户信息，这在实际中很有用。

代码很简单，就是两个类的实现， Rooms和User类，这里没有限定房间的数量，可以在初始化的时候先固定房间名和数量。源码托管于github，地址为：github.com/tianxiangbi… ,如果觉得有用，加颗小星星吧