dan/discord.js
1
0
Fork 0
mirror of https://github.com/danbulant/discord.js synced 2026-05-24 20:42:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Optimise websocket events

Browse source
This commit is contained in:
Schuyler Cebulskie 2016-11-20 23:39:40 -05:00
parent 0d754d1fbb
commit a4193553e2
1 changed files with 34 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
src/client/websocket/WebSocketManager.js
View file

@ -129,9 +129,7 @@ class WebSocketManager extends EventEmitter {
const item = this._queue[0];
if (this.ws.readyState === WebSocket.OPEN && item) {
if (this._remaining === 0) {
this.client.setTimeout(() => {
this.doQueue();
}, 1000);
this.client.setTimeout(this.doQueue.bind(this), 1000);
return;
}
this._remaining--;
@ -214,21 +212,42 @@ class WebSocketManager extends EventEmitter {
* @returns {boolean}
*/
eventMessage(event) {
let packet = event.data;
try {
if (typeof packet !== 'string') {
if (packet instanceof ArrayBuffer) packet = convertArrayBuffer(packet);
packet = inflate(packet).toString();
}
packet = JSON.parse(packet);
} catch (e) {
return this.eventError(new Error(Constants.Errors.BAD_WS_MESSAGE));
const data = this.tryParseEventData(event.data);
if (data === null) {
this.eventError(new Error(Constants.Errors.BAD_WS_MESSAGE));
return false;
}
this.client.emit('raw', packet);
this.client.emit('raw', data);
if (packet.op === Constants.OPCodes.HELLO) this.client.manager.setupKeepAlive(packet.d.heartbeat_interval);
return this.packetManager.handle(packet);
if (data.op === Constants.OPCodes.HELLO) this.client.manager.setupKeepAlive(data.d.heartbeat_interval);
return this.packetManager.handle(data);
}
/**
* Parses the raw data from a websocket event, inflating it if necessary
* @param {*} data Event data
* @returns {Object}
*/
parseEventData(data) {
if (typeof data !== 'string') {
if (data instanceof ArrayBuffer) data = convertArrayBuffer(data);
data = inflate(data).toString();
}
return JSON.parse(data);
}
/**
* Tries to call `parseEventData()` and return its result, or returns `null` upon thrown errors.
* @param {*} data Event data
* @returns {?Object}
*/
tryParseEventData(data) {
try {
return this.parseEventData(data);
} catch (err) {
return null;
}
}
/**