|
|
/** * 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(); };
|