nodejs-websocket 实现单聊和群聊

下载nodejs-websocket
https://github.com/sitegui/nodejs-websocket
最新公司有需求交给我一个任务，研究下关于websocket的即时通讯。写了一个在上面的基础上有优化了界面，也增加了功能
Demo地址
https://github.com/zhouyujuan/nodejs-websocket

使用nodejs-websocket的步骤
1、npm init 创建package.json
2、sudo npm install nodejs-websocket 就可以了
3、创建一个wsServer.js 是服务器文件
4、服务器端的代码可以先看https://github.com/sitegui/nodejs-websocket
上面的使用方法。
5、启动服务、node wsServer.js

wsServer.js 也就是服务器的代码

我们要先加载nodejs-websocket

var ws = require("nodejs-websocket")

// Scream server example: "hi" -> "HI!!!"
创建一个server
var server = ws.createServer(function (conn) {
    console.log("New connection")
接受到发送的消息
    conn.on("text", function (str) {
        console.log("Received "+str)
        conn.sendText(str.toUpperCase()+"!!!")
    })
关闭时触发的消息
    conn.on("close", function (code, reason) {
        console.log("Connection closed")
    })
}).listen(8001)

在服务端，接受到发起会话的消息时需要做很多的处理。判断会话所在的群组，判断会话对象的状态。等。
我在这里做的处理的思路
1、在chat界面发出了会话请求，这里会把发起会话人的基本信息整合成一个对象发送到服务器
// 连接成功建立的回调方法

    websocket.onopen = function(e) {
        // 建立连接后，要根据页面的url得知发起会话的人是想给谁发，或者在那个群里发
        var url = window.location.href;
        var splitobj = spliturl(url);
        user.name = splitobj.username;
        user.group = splitobj.groupnumber;
        if (user.group == undefined) {
            user.lineType = 1; //单聊
        } else {
            user.lineType = 0; //群聊
        }
        user.type = "enter";
        user.toname = splitobj.toname;
        websocket.send(JSON.stringify(user));
        // 这里是把发起会话的人的信息告诉服务器
        // 发送消息
        document.getElementById("sendBtn").onclick = function() {
            var txt = document.getElementById("sendText").value;
            document.getElementById('sendText').value = '';
            if (txt) {
                user.type = "message";
                user.data = txt;
                websocket.send(JSON.stringify(user));
            }
        }
    }

这里的信息包括了发起会话的人的
姓名（name）
发送给谁的（toname）这个在单聊的时候用到
群聊的id（group）群聊的id
发起会话的类型：（type）enter(进入)、 message(发消息)、leave(离开)

服务器接受到消息后

 var user;
    conn.on("text", function(str) {
        // 接收到客户端的消息时的处理
        // 这里的打印会在终端显示出来
        user = JSON.parse(str);
         // 这里相当于在有数据库情况下的一个查表的过程
         // 当有人在群里说话的时候，表示这个人是在线状态、
         // 这样他的好友在给他单独发消息的时候，发送的信息不回在这个群聊的会话列表展示
        if (user.group != undefined) {
            if (user.group == 1) {
                if (groups[0].members.contains(user.name)) {
                    console.log('可以进入群聊');
                    onlines.push(user.name);  //群聊的人在线状态的标识
                } else {
                    return;
                }
            } else if (user.group == 2) {
                if (groups[1].members.contains(user.name)) {
                    console.log('可以进入群聊');
                    onlines.push(user.name);
                } else {
                    return;
                }
            }
        } else {
            connections.push(user); // 如果不是群聊是单聊
        }

        // 当发起单聊的人超过1个人的时候，就要确定单聊的路线
        if (connections.length > 1) {
            for (var i = 0; i < connections.length; i++) {
                for (var j = i + 1; j < connections.length; j++) {
                    // 这里是判断单聊的路线，toname是要发送的人和要接受的人是否一样
                    if (connections[i].toname == connections[j].name &&
                        connections[j].toname == connections[i].name) {
                        signallines.push(connections[j].toname);
                        signallines.push(connections[j].name);

                    }
                }
            }
        }
       // 下面这些就是通过上面的信息，组织下要发送的信息

        var mes = {};
        mes.type = user.type;
        if (mes.type == 'enter') {
            mes.data = user.name;
        } else {
            mes.data = user.data;
        }
        mes.num = user.count;
        mes.name = user.name; //发起会话的人的姓名  这里最好使用id
        mes.toname = user.toname; //只会在单聊的时候存在，要发送给谁，也最好用id
        mes.groupnumber = user.group; //群聊的时候，群聊的id

        if (user.group != undefined) {
            // 你要单独给好友聊天，这里检测好友是否在群聊
            if (onlines.contains(mes.toname)) {
                console.log('我要发送的对象在线')
                 broadcast(JSON.stringify("好友在忙碌"));
            } else {
                broadcast(JSON.stringify(mes));
            }
        } else {
            // 你要单独给好友聊天，这里检测好友是否在和别人聊天
            if (signallines.contains(mes.toname) && signallines.contains(mes.name)) {
                broadcast(JSON.stringify(mes));
            } else {
                console.log('我要发送的---对象在线')
                broadcast(JSON.stringify("好友在忙碌"));       
            }
        }
        console.log('broadcast', JSON.stringify(mes));
    })

服务器接受到消息后。会判断这个会话是要在哪里发起（在群里，单聊的）
（1）如果带有群号，
验证下之后就可以吧这个user放到群聊在线的数组中(online)标识了一个状态；
（2）如果是单聊
放入单聊的数组中 connections.
对单聊的数组进行判断，如果大于两个人，我们需要在单聊的数组中找到配对的
也就是数组中一个对象A的toname 和另一个对象B 的 name 相等并且，B的toname和Aname相等。
这样就配对成功了

其实，这部分如果建了数据库处理起来很好，这里只是Dmeo所以没有做详尽的处理。

（3）通过（1）和（2）的操作我们基本知道了发起会话的人的应该把他的会话转向哪里了
1、是转到那个群
2、是转到那个人哪里
3、还是告诉他好友在忙，无法发送消息

当会话关闭是

   conn.on("close", function(code, reason) {
        clientCount = 0;
        var mes = {}
        mes.type = "leave"
        mes.data = user.name + 'leave';
        // 离开群组或者离开单聊的时候，要把状态清空
        onlines.remove(user.name);
        connections.remove(user.name);
        signallines.remove(user.name);
        broadcast(JSON.stringify(mes))
    })

当离开的时候也清除记录，改变状态
发送消息的方法。

function broadcast(str) {
    console.log('str', str);
    // 取到server下面的所有连接
    server.connections.forEach(function(connection) {
        connection.sendText(str);
    })
}

nodejs-websocket 实现单聊和群聊

nodejs-websocket 实现单聊和群聊

猜你喜欢