socket.io实现私聊

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/koastal/article/details/53677766

socket.io官网api地址
http://socket.io/docs/server-api/

socket.io官方api以及很多教程都知识介绍了广播消息和在当前会话发送消息,在很多时候我们需要实现向某个指定的在线用户发送消息,也就是所谓的“私聊”。

socket.io服务端基本代码结构

var app = require('http').createServer(handler)
var io = require('socket.io')(app);
var fs = require('fs');
app.listen(3000);
function handler (req, res) {
    fs.readFile(__dirname + '/index.html',
        function (err, data) {
            res.writeHead(200);
            res.end(data);
        });
}
io.on('connection', function(socket){
    console.log('connection is established! '+socket.id);
    socket.on('disconnect', function(){
        console.log('connection is disconnect!');
        console.log(socket.id);
    });
});

如上述代码所示,每个在线的用户都对应一个socket对象,可以通过socket.id获取其唯一的id。

网上很多人是将所有的socket存储到一个全局数据中,socket.id为键名,socket对象为键值,然后我们可以再建立一个保存用户名和socket.id映射关系的数组;或者直接将用户名作为键名,socket对象作为键值。
这种方式需要维护一个比较大的全局数组,如果用户量很大时,就会比较慢。

事实上,socket.io在运行时,本身就会维护所有的socket对象,这样才能实现消息的广播。

socket.io版本<1,使用下面的方法,将返回所有在线的socket对象数组

io.sockets.clients();

socket.io版本>1,使用下面的属性,将返回所有在线的socket对象集合

io.sockets.sockets;

我们以最新版1.7.2为例,通过上面的属性,就可以获取所有在线的socket集合,没有必要再去存储一个全局数组了。

nodejs的underscore扩展中的findWhere方法,可以是我们很容易在对象集合中,通过对象的属性值找到该对象并返回。

所以,我们现在只需要维护一个socket.id和用户名映射关系的全局数组即可。

建立websocket连接之后,客户端首先设置一下自己的用户名,服务器端将用户名和socket.id的对应关系保存,当客户端A向B发送私聊信息时,需要带上客户端B的用户名。我们通过B的用户名,解析得到客户端B的socket.id,从socket集合中获取响应的socket,然后再发送消息。

var app = require('http').createServer(handler)
var io = require('socket.io')(app);
var fs = require('fs');
var _ = require('underscore');
app.listen(3000);
var hashName = new Array();

function handler (req, res) {
    fs.readFile(__dirname + '/index.html',
        function (err, data) {
            res.writeHead(200);
            res.end(data);
        });
}

io.on('connection', function(socket){
    console.log('connection is established!');
    socket.on('setName',function (data) {
        var name = data;  
        hashName[name] = socket.id;
        console.log(hashName);
    });
    socket.on('sayTo',function (data) {
        var toName = data.to;
        var toId;
        if(toId = hashName[toName]){
            var toSocket = _.findWhere(io.sockets.sockets,{id:toId});
            toSocket.emit('message',data.msg);
        }
    })
    socket.on('disconnect', function(){
        console.log('connection is disconnect!');
    });
});

还有一种方案更加直接,我们将用户名直接设置为socket的name属性,通过name属性来找到socket,并发送消息。

...

io.on('connection', function(socket){
    console.log('connection is established!');
    socket.on('setName',function (data) {
        socket.name = data;
    });
    socket.on('sayTo',function (data) {
        var toName = data.to;
        var toSocket;
        if(toSocket = _.findWhere(io.sockets.sockets,{name:toName})){
            toSocket.emit('message',data.msg);
        }
    })
    socket.on('disconnect', function(){
        console.log(socket.name+' connection is disconnect!');
    });
});

客户端代码:

index.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>socket.io演示客户端</title>
    <script src="https://cdn.socket.io/socket.io-1.4.5.js"></script>
</head>
<body>
<h1>socket.io演示客户端</h1>
</body>
<script>
    var socket = io.connect('127.0.0.1:3000');
    socket.on('message',function (data) {
        console.log(data);
    });
</script>
</html>

参考资料:

https://github.com/nswbmw/N-chat/wiki/%E7%AC%AC%E4%BA%94%E7%AB%A0-%E7%94%A8%E6%88%B7%E5%8F%91%E8%AF%9D

http://bobcaojin198306.blog.163.com/blog/static/9610418201411315012664/

https://github.com/kiroChen/node_chat/blob/master/chat.js

展开阅读全文

没有更多推荐了,返回首页