Spaces:
Running
Running
| ; | |
| Object.defineProperty(exports, "__esModule", { value: true }); | |
| exports.WebSocket = void 0; | |
| const transport_1 = require("../transport"); | |
| const debug_1 = require("debug"); | |
| const debug = (0, debug_1.default)("engine:ws"); | |
| class WebSocket extends transport_1.Transport { | |
| /** | |
| * WebSocket transport | |
| * | |
| * @param req | |
| * @api public | |
| */ | |
| constructor(req) { | |
| super(req); | |
| this.writable = false; | |
| this.perMessageDeflate = null; | |
| } | |
| /** | |
| * Transport name | |
| * | |
| * @api public | |
| */ | |
| get name() { | |
| return "websocket"; | |
| } | |
| /** | |
| * Advertise upgrade support. | |
| * | |
| * @api public | |
| */ | |
| get handlesUpgrades() { | |
| return true; | |
| } | |
| /** | |
| * Advertise framing support. | |
| * | |
| * @api public | |
| */ | |
| get supportsFraming() { | |
| return true; | |
| } | |
| /** | |
| * Writes a packet payload. | |
| * | |
| * @param {Array} packets | |
| * @api private | |
| */ | |
| send(packets) { | |
| this.writable = false; | |
| for (let i = 0; i < packets.length; i++) { | |
| const packet = packets[i]; | |
| const isLast = i + 1 === packets.length; | |
| const send = (data) => { | |
| const isBinary = typeof data !== "string"; | |
| const compress = this.perMessageDeflate && | |
| Buffer.byteLength(data) > this.perMessageDeflate.threshold; | |
| debug('writing "%s"', data); | |
| this.socket.send(data, isBinary, compress); | |
| if (isLast) { | |
| this.writable = true; | |
| this.emit("drain"); | |
| } | |
| }; | |
| if (packet.options && typeof packet.options.wsPreEncoded === "string") { | |
| send(packet.options.wsPreEncoded); | |
| } | |
| else { | |
| this.parser.encodePacket(packet, this.supportsBinary, send); | |
| } | |
| } | |
| } | |
| /** | |
| * Closes the transport. | |
| * | |
| * @api private | |
| */ | |
| doClose(fn) { | |
| debug("closing"); | |
| fn && fn(); | |
| // call fn first since socket.end() immediately emits a "close" event | |
| this.socket.end(); | |
| } | |
| } | |
| exports.WebSocket = WebSocket; | |