Node.js 教程第十三篇——WebSocket

lemontreeshy

2019-06-27

关注关注

重温 HTTP 协议

HTTP 协议可以总结几个特点：

一次性的、无状态的短连接：客户端发起请求、服务端响应、结束。
被动性响应：只有当客户端请求时才被执行，给予响应，不能主动向客户端发起响应。
信息安全性：得在服务器添加 SSL 证书，访问时用 HTTPS。
跨域：服务器默认不支持跨域，可在服务端设置支持跨域的代码或对应的配置。

认识 TCP

TCP 协议可以总结几个特点：

有状态的长连接：客户端发起连接请求，服务端响应并建立连接，连接会一直保持直到一方主动断开。
主动性：建立起与客户端的连接后，服务端可主动向客户端发起调用。
信息安全性：同样可以使用 SSL 证书进行信息加密，访问时用 WSS 。
跨域：默认支持跨域。

认识 WebSocket

WebSocket 目前由 W3C 进行标准化。WebSocket 已经受到 Firefox 4、Chrome 4、Opera 10.70 以及Safari 5 等浏览器的支持。
如果在前端我们可以把 AJAX 请求当作一个 HTTP 协议的实现，那么，WebSocket 就是 TCP 协议的一种实现。

服务端 API

安装第三方模块 ws：npm install ws
开启一个 WebSocket 的服务器，端口为 8080

var socketServer = require('ws').Server;
var wss = new socketServer({
    port: 8080
});

也可以利用 Express 来开启 WebSocket 的服务器

var app = require('express')();
var server = require('http').Server(app);

var socketServer = require('ws').Server;
var wss = new socketServer({server: server, port: 8080});

用 on 来进行事件监听
connection：连接监听，当客户端连接到服务端时触发该事件
close：连接断开监听，当客户端断开与服务器的连接时触发
message：消息接受监听，当客户端向服务端发送信息时触发该事件
send: 向客户端推送信息

wss.on('connection', function (client) {
    client.on('message', function (_message) {
        var _messageObj = JSON.parse(_message);
        //status = 1 表示正常聊天
        _messageObj.status = 1;
        this.message = _messageObj;
        //把客户端的消息广播给所有在线的用户
        wss.broadcast(_messageObj);
    });

    // 退出聊天  
    client.on('close', function() {  
        try{
            this.message = this.message || {};
            // status = 0 表示退出聊天
            this.message.status = 0;
            //把客户端的消息广播给所有在线的用户
            wss.broadcast(this.message);  
        }catch(e){  
            console.log('刷新页面了');  
        }  
    });  
});

//定义广播方法
wss.broadcast = function broadcast(_messageObj) {  
    wss.clients.forEach(function(client) { 
        client.send(JSON.stringify(_messageObj))
    });  
};

客户端 API

在支持 WebSocket 的浏览器下实例化 WebSocket ，参数为 WebSocket 服务器地址，建立与服务器的连接

if(!WebSocket){
    $('.connState').text("您的浏览器不支持WebSocket");
    return false;
} 
//连接 socket 服务器
var socket = new WebSocket('ws://localhost:8080');

onopen：当网络连接建立时触发该事件

//监听 socket 的连接
socket.onopen = function(){
    $('.connState').text("服务已连接 ws://localhost:8080");
}

onclose：当服务端关闭时触发该事件

//监听服务端断开
socket.onclose = function(){
    $('.connState').text("服务已断开");
    socket = null;
}

close: 在客户端断开与服务端的连接 socket.close();
onerror：当网络发生错误时触发该事件

//监听服务端异常
socket.onerror = function(){
    $('.connState').text("服务错误");
    socket = null;
}

onmessage：当接收到服务器发来的消息的时触发的事件，也是通信中最重要的一个监听事件

//监听服务端广播过来的消息
socket.onmessage = function(msg){
    var msgObj = JSON.parse(msg.data);
    if(msgObj.status == 0){
        $('<p>' + msgObj.nickname + '[' + msgObj.time + ']退出聊天</p>').appendTo('.msgList');
    } else{
        $('<p>' + msgObj.nickname + '[' + msgObj.time + ']：' + msgObj.message + '</p>').appendTo('.msgList');
    }
}

send：向服务端推送消息

var sendMessage = function(_mess){
    if(socket){
        var myDate = new Date();
        var now = myDate.getMonth() + '-' + myDate.getDate() + ' ' + myDate.getHours() + ':' + myDate.getMinutes() + ':' + myDate.getSeconds();                
        
        var mesObj = {
            nickname: $('#nickName').val(),
            message: _mess || $('#mesBox').val(),
            time: now
        }
        //向服务端发送消息
        socket.send(JSON.stringify(mesObj));
    }            
}