nodejs+socket.io实现websocket通信初探

因为项目中有需要实时获取后台数据的场景，之前一直是使用http心跳请求的方法。因为websocket与此模式相比有很大的性能提升，而且可以提高实时性，所以对websocket作了一些研究。这里是使用nodejs+socket.io来实现的。

达成目标

将原来心跳请求后台数据的方式，修改为通过socket连接后台统一推送的方式。后台的数据由别的进程写入文件或写入redis，这里实现的是读取文件的方式。

前期准备

安装nodejs(略)

服务器端

新建一个项目目录，这里是sockettest
进入sockettest目录，安装express模块和socketio模块

npm install --save express@4.10.2
npm install --save socket.io

新建package.json文件，在其中写入如下内容：

{
  "name": "socket-test",
  "version": "0.0.1",
  "description": "my first socket.io app",
  "dependencies": {
    "express": "^4.10.2",
    "socket.io": "^1.7.2"
  }
}

新建index.html，用于作为默认的访问显示页面，因为这里不会用到它，内容随意；
新建trends.js文件，在其中写入内容：

var app = require('express')();
var http = require('http').Server(app);
var io = require('socket.io')(http);
var fs = require('fs');

#默认打开文件
app.get('/', function(req, res){
    res.sendfile('index.html');
});

#用于存储所有socket以广播数据
var iolist = [];

#定义socket on connection（连入）事件行为
io.on('connection', function(socket){
    #将连入socket加入列表
    iolist.push(socket); 
    #记录index，在disconnect（断开连接）发生时将对应的socket删除
    var sockex = iolist.indexOf(socket); 
    #定义on disconnect事件行为
    socket.on('disconnect', function(){
        #将断开连接的socket从广播列表里删除
        iolist.splice(sockex, 1);
    });
});

# 数据广播进程:每1秒钟广播一次
setInterval(function() {
    # 如果没有正在连接的socket，直接返回；
    if (iolist.length <= 0) return;
    var trends = fs.readFileSync('./data/trends.json','utf-8');#trends数据
    var coins = fs.readFileSync('./data/coins.json','utf-8');#coins数据
    #向所有socket连接发送数据
    for (i in iolist) {
        # 向客户端发送trends数据
        iolist[i].emit('trends', trends);
        # 向客户端发送coins数据
        iolist[i].emit('coins', coins);
    }
}, 1000);

# 服务器侦听在sockettest.com的3000端口上
http.listen(3000, function(){
    # 输出到标准输出
    console.log('listening on sockettest.com:3000');
});

新建data目录，并在下面新建两个文件trends与coins，用于存放socket服务器将要读取的数据。
新建public目录，在其中新建一个文件index.html，文件内容如下：

<!--引入必要的js文件-->
<script type="text/javascript" src="http://sockettest:3000/socket.io/socket.io.js"></script>
<script type="text/javascript">
        //新建socket
        var socket = io('http://sockettest.com:3000');
        socketdata(socket);

        function socketdata() {
            #定义接收到coins类型数据时的行为
            socket.on('coins', function(msg){
                console.log(msg);
            }
            #定义接收到trends类型数据时的行为
            socket.on('trends', function(msg){
                console.log(msg);
            }
        }
</script>

代码部署

刚才之所以要建两个index.html文件，是为了能够方便地在既有的web项目中使用nodejs提供的socket服务。这样我们把public/index.html可以部署在别的服务器中，比如nginx或tomcat之类，然后在根目下启动socket的服务器，为其提供socket服务。
首先在刚才的项目根目录下执行

node ./trends.js

并保持终端运行，然后再把项目部署在nginx里，通过chrome下访问nginx提供的web服务：

http://hostname/public/index.html

打开开发者模式，就能在console里看到每隔一秒便会收到来自node服务器的socket推送消息了。通过修改data目录下的两个文件，可以看到写入到文件的数据也会实时地推送到客户端这里来。

参考文章

nodejs官方聊天室demo