nodejs结合socket.io实现websocket通信功能的方法

本文实例讲述了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目录下的两个文件，可以看到写入到文件的数据也会实时地推送到客户端这里来。

参考文章

http://socket.io/get-started/chat/

希望本文所述对大家nodejs程序设计有所帮助。