在nodejs中如何实现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(){
 # 

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

上面是我整理给大家的，希望今后会对大家有帮助。

相关文章：

在vue中使用cli如何实现重构多页面脚手架

在JS中实现点击下拉菜单内容同步输入框

实现输入框与下拉框联动

使用parcel.js打包出错的问题

详细解读vue重构技术