48 lines
969 B
TypeScript
48 lines
969 B
TypeScript
|
import { EventEmitter } from 'events';
|
||
|
|
import type WebSocket from 'ws';
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Wrapper around ws WebSocket that only receives events via emit().
|
||
|
|
* This prevents double-handling when used with RedisSyncExtension.
|
||
|
|
*/
|
||
|
|
export class WsSocketWrapper extends EventEmitter {
|
||
|
|
private ws: WebSocket;
|
||
|
|
readyState = 1;
|
||
|
|
|
||
|
|
constructor(ws: WebSocket) {
|
||
|
|
super();
|
||
|
|
this.ws = ws;
|
||
|
|
this.once('close', () => {
|
||
|
|
this.readyState = 3;
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
close(code?: number, reason?: string) {
|
||
|
|
if (this.readyState !== 1) return;
|
||
|
|
this.readyState = 3;
|
||
|
|
try {
|
||
|
|
this.ws.close(code, reason);
|
||
|
|
} catch (e) {
|
||
|
|
// Socket already closed
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
ping() {
|
||
|
|
if (this.readyState !== 1) return;
|
||
|
|
try {
|
||
|
|
this.ws.ping();
|
||
|
|
} catch (e) {
|
||
|
|
// Socket already closed
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
send(message: Uint8Array) {
|
||
|
|
if (this.readyState !== 1) return;
|
||
|
|
try {
|
||
|
|
this.ws.send(message);
|
||
|
|
} catch (e) {
|
||
|
|
// Socket already closed
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|