|
|
/**
|
|
* Module dependencies.
|
|
*/
|
|
|
|
var Transport = require('../transport')
|
|
, parser = require('engine.io-parser')
|
|
, debug = require('debug')('engine:ws')
|
|
|
|
/**
|
|
* Export the constructor.
|
|
*/
|
|
|
|
module.exports = WebSocket;
|
|
|
|
/**
|
|
* WebSocket transport
|
|
*
|
|
* @param {http.ServerRequest}
|
|
* @api public
|
|
*/
|
|
|
|
function WebSocket (req) {
|
|
Transport.call(this, req);
|
|
var self = this;
|
|
this.socket = req.websocket;
|
|
this.socket.on('message', this.onData.bind(this));
|
|
this.socket.once('close', this.onClose.bind(this));
|
|
this.socket.on('error', this.onError.bind(this));
|
|
this.socket.on('headers', function (headers) {
|
|
self.emit('headers', headers);
|
|
});
|
|
this.writable = true;
|
|
};
|
|
|
|
/**
|
|
* Inherits from Transport.
|
|
*/
|
|
|
|
WebSocket.prototype.__proto__ = Transport.prototype;
|
|
|
|
/**
|
|
* Transport name
|
|
*
|
|
* @api public
|
|
*/
|
|
|
|
WebSocket.prototype.name = 'websocket';
|
|
|
|
/**
|
|
* Advertise upgrade support.
|
|
*
|
|
* @api public
|
|
*/
|
|
|
|
WebSocket.prototype.handlesUpgrades = true;
|
|
|
|
/**
|
|
* Advertise framing support.
|
|
*
|
|
* @api public
|
|
*/
|
|
|
|
WebSocket.prototype.supportsFraming = true;
|
|
|
|
/**
|
|
* Processes the incoming data.
|
|
*
|
|
* @param {String} encoded packet
|
|
* @api private
|
|
*/
|
|
|
|
WebSocket.prototype.onData = function (data) {
|
|
debug('received "%s"', data);
|
|
Transport.prototype.onData.call(this, data);
|
|
};
|
|
|
|
/**
|
|
* Writes a packet payload.
|
|
*
|
|
* @param {Array} packets
|
|
* @api private
|
|
*/
|
|
|
|
WebSocket.prototype.send = function (packets) {
|
|
var self = this;
|
|
for (var i = 0, l = packets.length; i < l; i++) {
|
|
parser.encodePacket(packets[i], this.supportsBinary, function(data) {
|
|
debug('writing "%s"', data);
|
|
self.writable = false;
|
|
self.socket.send(data, function (err){
|
|
if (err) return self.onError('write error', err.stack);
|
|
self.writable = true;
|
|
self.emit('drain');
|
|
});
|
|
});
|
|
}
|
|
};
|
|
|
|
/**
|
|
* Closes the transport.
|
|
*
|
|
* @api private
|
|
*/
|
|
|
|
WebSocket.prototype.doClose = function (fn) {
|
|
debug('closing');
|
|
this.socket.close();
|
|
fn && fn();
|
|
};
|