socket.io实现私聊

最新推荐文章于 2024-06-23 13:13:02 发布

koastal

最新推荐文章于 2024-06-23 13:13:02 发布

阅读量1.4w

点赞数 4

分类专栏： nodejs

本文链接：https://blog.csdn.net/koastal/article/details/53677766

版权

nodejs 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

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

koastal

关注

4
点赞
踩
26

收藏

觉得还不错? 一键收藏
2
评论
socket.io实现私聊

socket.io官网api地址 http://socket.io/docs/server-api/socket.io官方api以及很多教程都知识介绍了广播消息和在当前会话发送消息，在很多时候我们需要实现向某个指定的在线用户发送消息，也就是所谓的“私聊”。socket.io服务端基本代码结构var app = require('http').createServer(handler)var io
复制链接

扫一扫

专栏目录