WebSocket(伍) 断开连接

原文链接 https://www.web-tinker.com/article/20310.html

WebSocket是很民主的，啥都要协商！建立连接时需要握手协议，连断开连接都需要双方共同完成！其实断开连接直接断开TCP连接就可以了，但是这有点暴力。文明点的方法是发个请求，让对方自己断开。客户端要主动断开就必须向服务器发送8这个操作码。
　　首先是服务器主导断开的情况，最简单的方法是直接把TCP连接断开，这里就不演示了。由于这对客户端来说是个意外断开，WebSocket对象采取应急措施也触发close事件。咱是文明人，当然要做点有绅士风度的事情。于是咱不从服务器断开连接，而是向客户端发送个请求断开的操作码来请求客户端自己断开。
　　其实就是个操作码为8的帧。但要注意的是数据部分比较特殊。当然如果嫌麻烦可以不传，不过要是不传就和前面的霸王硬上弓一样无节操了。数据部分的前两个字节是状态码，之后的部分是关闭连接原因的文本描述，这些东西可以传到客户端。
　　（encodeDataFrame与decodeDataFrame函数见生成数据帧和解析数据帧）

//客户端程序
var ws=new WebSocket("ws://127.0.0.1:8000");
ws.onclose=function(e){
  console.log(e);
  ws.close(); //关闭TCP连接
};

//服务器程序
var crypto=require('crypto');
var WS='258EAFA5-E914-47DA-95CA-C5AB0DC85B11';
require('net').createServer(function(o){
  var key;
  o.on('data',function(e){
    if(!key){
      //握手
      key=e.toString().match(/Sec-WebSocket-Key: (.+)/)[1];
      key=crypto.createHash('sha1').update(key+WS).digest('base64');
      o.write('HTTP/1.1 101 Switching Protocols\r\n');
      o.write('Upgrade: websocket\r\n');
      o.write('Connection: Upgrade\r\n');
      o.write('Sec-WebSocket-Accept: '+key+'\r\n');
      o.write('\r\n');
      //构造断连请求的数据部分，前面留两字节存放状态码
      var buf=new Buffer('\0\0孩子，地球太危险了，快回火星去吧！');
      buf.writeUInt16BE(1000,0); //在头两个字节写入一个状态码
      //发送断连请求
      o.write(encodeDataFrame({FIN:1,Opcode:8,PayloadData:buf}));
    };
  });
}).listen(8000);

客户端会在onclose的参数中接收到一个这样的东西，状态码和结束原因描述分别在code和reason两个参数中。规范文档中规定了很多状态码的含义，不过这个目前不是强制性的，我就不列举了。见RFC6455#section-7.4。客户端在收到服务器的这个断连请求后应该调用close方法来关闭，否则连接会先入停滞状态等待客户端响应。
　　服务器主导断开的情况就是这样。下面是客户端主导断开的情况。客户端先要调用close方法，这个操作会发送一个断连请求到服务器上，服务器收到这个请求后把TCP连接断开即可。但是服务器程序是自己写的，这个请求也需要自己解析。

//客户端程序
var ws=new WebSocket("ws://127.0.0.1:8000");
ws.onopen=function(){
  ws.close(); //发起断连请求
};
ws.onclose=function(e){
  console.log(e);
};

//服务端程序序
var crypto=require('crypto');
var WS='258EAFA5-E914-47DA-95CA-C5AB0DC85B11';
require('net').createServer(function(o){
  var key;
  o.on('data',function(e){
    if(!key){
      //握手
      key=e.toString().match(/Sec-WebSocket-Key: (.+)/)[1];
      key=crypto.createHash('sha1').update(key+WS).digest('base64');
      o.write('HTTP/1.1 101 Switching Protocols\r\n');
      o.write('Upgrade: websocket\r\n');
      o.write('Connection: Upgrade\r\n');
      o.write('Sec-WebSocket-Accept: '+key+'\r\n');
      o.write('\r\n');
    }else{
      var frame=decodeDataFrame(e);
      console.log(frame);
      if(frame.Opcode==8){
        //这里也可以发送个结束包来给客户端的onclose中带参数
        //var buf=new Buffer('\0\0孩子，地球太危险了，快回火星去吧！');
        //buf.writeUInt16BE(1000,0);
        //o.write(encodeDataFrame({FIN:1,Opcode:8,PayloadData:buf}));
        o.end(); //断开连接
      };
    };
  });
}).listen(8000);

总之，客户端直接调用close方法并不会关闭连接，而是发送请求到服务器请求对方。服务器接收请求后可以断开连接。这会触发客户端的close事件。当然，在断开之前也可以发送个同样的断连请求，并包含状态码和原因描述。