@socket.io/redis-streams-adapter 0.2.0 → 0.2.1

package/dist/cluster-adapter.js

@@ -1,420 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ClusterAdapter = exports.MessageType = void 0;
-const socket_io_adapter_1 = require("socket.io-adapter");
-const debug_1 = require("debug");
-const util_1 = require("./util");
-const debug = (0, debug_1.default)("socket.io-adapter");
-const EMITTER_UID = "emitter";
-const DEFAULT_TIMEOUT = 5000;
-var MessageType;
-(function (MessageType) {
-    MessageType[MessageType["INITIAL_HEARTBEAT"] = 1] = "INITIAL_HEARTBEAT";
-    MessageType[MessageType["HEARTBEAT"] = 2] = "HEARTBEAT";
-    MessageType[MessageType["BROADCAST"] = 3] = "BROADCAST";
-    MessageType[MessageType["SOCKETS_JOIN"] = 4] = "SOCKETS_JOIN";
-    MessageType[MessageType["SOCKETS_LEAVE"] = 5] = "SOCKETS_LEAVE";
-    MessageType[MessageType["DISCONNECT_SOCKETS"] = 6] = "DISCONNECT_SOCKETS";
-    MessageType[MessageType["FETCH_SOCKETS"] = 7] = "FETCH_SOCKETS";
-    MessageType[MessageType["FETCH_SOCKETS_RESPONSE"] = 8] = "FETCH_SOCKETS_RESPONSE";
-    MessageType[MessageType["SERVER_SIDE_EMIT"] = 9] = "SERVER_SIDE_EMIT";
-    MessageType[MessageType["SERVER_SIDE_EMIT_RESPONSE"] = 10] = "SERVER_SIDE_EMIT_RESPONSE";
-    MessageType[MessageType["BROADCAST_CLIENT_COUNT"] = 11] = "BROADCAST_CLIENT_COUNT";
-    MessageType[MessageType["BROADCAST_ACK"] = 12] = "BROADCAST_ACK";
-})(MessageType = exports.MessageType || (exports.MessageType = {}));
-function encodeOptions(opts) {
-    return {
-        rooms: [...opts.rooms],
-        except: [...opts.except],
-        flags: opts.flags,
-    };
-}
-function decodeOptions(opts) {
-    return {
-        rooms: new Set(opts.rooms),
-        except: new Set(opts.except),
-        flags: opts.flags,
-    };
-}
-class ClusterAdapter extends socket_io_adapter_1.Adapter {
-    #opts;
-    #uid;
-    #heartbeatTimer;
-    #nodesMap = new Map(); // uid => timestamp of last message
-    #requests = new Map();
-    #ackRequests = new Map();
-    constructor(nsp, opts) {
-        super(nsp);
-        this.#opts = opts;
-        this.#uid = (0, util_1.randomId)();
-    }
-    initHeartbeat() {
-        this.#publish({
-            type: MessageType.INITIAL_HEARTBEAT,
-        });
-    }
-    #scheduleHeartbeat() {
-        if (this.#heartbeatTimer) {
-            clearTimeout(this.#heartbeatTimer);
-        }
-        this.#heartbeatTimer = setTimeout(() => {
-            this.#publish({
-                type: MessageType.HEARTBEAT,
-            });
-        }, this.#opts.heartbeatInterval);
-    }
-    close() {
-        clearTimeout(this.#heartbeatTimer);
-    }
-    async onMessage(message, offset) {
-        if (message.uid === this.#uid) {
-            return debug("ignore message from self");
-        }
-        if (message.uid && message.uid !== EMITTER_UID) {
-            this.#nodesMap.set(message.uid, Date.now());
-        }
-        debug("new event of type %d from %s", message.type, message.uid);
-        switch (message.type) {
-            case MessageType.INITIAL_HEARTBEAT:
-                this.#publish({
-                    type: MessageType.HEARTBEAT,
-                });
-                break;
-            case MessageType.BROADCAST: {
-                const withAck = message.data.requestId !== undefined;
-                if (withAck) {
-                    super.broadcastWithAck(message.data.packet, decodeOptions(message.data.opts), (clientCount) => {
-                        debug("waiting for %d client acknowledgements", clientCount);
-                        this.#publish({
-                            type: MessageType.BROADCAST_CLIENT_COUNT,
-                            data: {
-                                requestId: message.data.requestId,
-                                clientCount,
-                            },
-                        });
-                    }, (arg) => {
-                        debug("received acknowledgement with value %j", arg);
-                        this.#publish({
-                            type: MessageType.BROADCAST_ACK,
-                            data: {
-                                requestId: message.data.requestId,
-                                packet: arg,
-                            },
-                        });
-                    });
-                }
-                else {
-                    const packet = message.data.packet;
-                    const opts = decodeOptions(message.data.opts);
-                    this.#addOffsetIfNecessary(packet, opts, offset);
-                    super.broadcast(packet, opts);
-                }
-                break;
-            }
-            case MessageType.BROADCAST_CLIENT_COUNT: {
-                const request = this.#ackRequests.get(message.data.requestId);
-                request?.clientCountCallback(message.data.clientCount);
-                break;
-            }
-            case MessageType.BROADCAST_ACK: {
-                const request = this.#ackRequests.get(message.data.requestId);
-                request?.ack(message.data.packet);
-                break;
-            }
-            case MessageType.SOCKETS_JOIN:
-                super.addSockets(decodeOptions(message.data.opts), message.data.rooms);
-                break;
-            case MessageType.SOCKETS_LEAVE:
-                super.delSockets(decodeOptions(message.data.opts), message.data.rooms);
-                break;
-            case MessageType.DISCONNECT_SOCKETS:
-                super.disconnectSockets(decodeOptions(message.data.opts), message.data.close);
-                break;
-            case MessageType.FETCH_SOCKETS: {
-                debug("calling fetchSockets with opts %j", message.data.opts);
-                const localSockets = await super.fetchSockets(decodeOptions(message.data.opts));
-                this.#publish({
-                    type: MessageType.FETCH_SOCKETS_RESPONSE,
-                    data: {
-                        requestId: message.data.requestId,
-                        sockets: localSockets.map((socket) => {
-                            // remove sessionStore from handshake, as it may contain circular references
-                            const { sessionStore, ...handshake } = socket.handshake;
-                            return {
-                                id: socket.id,
-                                handshake,
-                                rooms: [...socket.rooms],
-                                data: socket.data,
-                            };
-                        }),
-                    },
-                });
-                break;
-            }
-            case MessageType.FETCH_SOCKETS_RESPONSE: {
-                const requestId = message.data.requestId;
-                const request = this.#requests.get(requestId);
-                if (!request) {
-                    return;
-                }
-                request.current++;
-                message.data.sockets.forEach((socket) => request.responses.push(socket));
-                if (request.current === request.expected) {
-                    clearTimeout(request.timeout);
-                    request.resolve(request.responses);
-                    this.#requests.delete(requestId);
-                }
-                break;
-            }
-            case MessageType.SERVER_SIDE_EMIT: {
-                const packet = message.data.packet;
-                const withAck = message.data.requestId !== undefined;
-                if (!withAck) {
-                    this.nsp._onServerSideEmit(packet);
-                    return;
-                }
-                let called = false;
-                const callback = (arg) => {
-                    // only one argument is expected
-                    if (called) {
-                        return;
-                    }
-                    called = true;
-                    debug("calling acknowledgement with %j", arg);
-                    this.#publish({
-                        type: MessageType.SERVER_SIDE_EMIT_RESPONSE,
-                        data: {
-                            requestId: message.data.requestId,
-                            packet: arg,
-                        },
-                    });
-                };
-                packet.push(callback);
-                this.nsp._onServerSideEmit(packet);
-                break;
-            }
-            case MessageType.SERVER_SIDE_EMIT_RESPONSE: {
-                const requestId = message.data.requestId;
-                const request = this.#requests.get(requestId);
-                if (!request) {
-                    return;
-                }
-                request.current++;
-                request.responses.push(message.data.packet);
-                if (request.current === request.expected) {
-                    clearTimeout(request.timeout);
-                    request.resolve(null, request.responses);
-                    this.#requests.delete(requestId);
-                }
-            }
-        }
-    }
-    async broadcast(packet, opts) {
-        const onlyLocal = opts.flags?.local;
-        if (!onlyLocal) {
-            try {
-                const offset = await this.#publish({
-                    type: MessageType.BROADCAST,
-                    data: {
-                        packet,
-                        opts: encodeOptions(opts),
-                    },
-                });
-                this.#addOffsetIfNecessary(packet, opts, offset);
-            }
-            catch (e) {
-                return debug("error while broadcasting message: %s", e.message);
-            }
-        }
-        super.broadcast(packet, opts);
-    }
-    /**
-     * Adds an offset at the end of the data array in order to allow the client to receive any missed packets when it
-     * reconnects after a temporary disconnection.
-     *
-     * @param packet
-     * @param opts
-     * @param offset
-     * @private
-     */
-    #addOffsetIfNecessary(packet, opts, offset) {
-        if (!this.nsp.server.opts.connectionStateRecovery) {
-            return;
-        }
-        const isEventPacket = packet.type === 2;
-        // packets with acknowledgement are not stored because the acknowledgement function cannot be serialized and
-        // restored on another server upon reconnection
-        const withoutAcknowledgement = packet.id === undefined;
-        const notVolatile = opts.flags?.volatile === undefined;
-        if (isEventPacket && withoutAcknowledgement && notVolatile) {
-            packet.data.push(offset);
-        }
-    }
-    broadcastWithAck(packet, opts, clientCountCallback, ack) {
-        const onlyLocal = opts?.flags?.local;
-        if (!onlyLocal) {
-            const requestId = (0, util_1.randomId)();
-            this.#publish({
-                type: MessageType.BROADCAST,
-                data: {
-                    packet,
-                    requestId,
-                    opts: encodeOptions(opts),
-                },
-            });
-            this.#ackRequests.set(requestId, {
-                clientCountCallback,
-                ack,
-            });
-            // we have no way to know at this level whether the server has received an acknowledgement from each client, so we
-            // will simply clean up the ackRequests map after the given delay
-            setTimeout(() => {
-                this.#ackRequests.delete(requestId);
-            }, opts.flags.timeout);
-        }
-        super.broadcastWithAck(packet, opts, clientCountCallback, ack);
-    }
-    serverCount() {
-        return Promise.resolve(1 + this.#nodesMap.size);
-    }
-    /**
-     *
-     * @param opts
-     * @param rooms
-     */
-    addSockets(opts, rooms) {
-        super.addSockets(opts, rooms);
-        const onlyLocal = opts.flags?.local;
-        if (onlyLocal) {
-            return;
-        }
-        this.#publish({
-            type: MessageType.SOCKETS_JOIN,
-            data: {
-                opts: encodeOptions(opts),
-                rooms,
-            },
-        });
-    }
-    delSockets(opts, rooms) {
-        super.delSockets(opts, rooms);
-        const onlyLocal = opts.flags?.local;
-        if (onlyLocal) {
-            return;
-        }
-        this.#publish({
-            type: MessageType.SOCKETS_LEAVE,
-            data: {
-                opts: encodeOptions(opts),
-                rooms,
-            },
-        });
-    }
-    disconnectSockets(opts, close) {
-        super.disconnectSockets(opts, close);
-        const onlyLocal = opts.flags?.local;
-        if (onlyLocal) {
-            return;
-        }
-        this.#publish({
-            type: MessageType.DISCONNECT_SOCKETS,
-            data: {
-                opts: encodeOptions(opts),
-                close,
-            },
-        });
-    }
-    #getExpectedResponseCount() {
-        this.#nodesMap.forEach((lastSeen, uid) => {
-            const nodeSeemsDown = Date.now() - lastSeen > this.#opts.heartbeatTimeout;
-            if (nodeSeemsDown) {
-                debug("node %s seems down", uid);
-                this.#nodesMap.delete(uid);
-            }
-        });
-        return this.#nodesMap.size;
-    }
-    async fetchSockets(opts) {
-        const localSockets = await super.fetchSockets(opts);
-        const expectedResponseCount = this.#getExpectedResponseCount();
-        if (opts.flags?.local || expectedResponseCount === 0) {
-            return localSockets;
-        }
-        const requestId = (0, util_1.randomId)();
-        return new Promise((resolve, reject) => {
-            const timeout = setTimeout(() => {
-                const storedRequest = this.#requests.get(requestId);
-                if (storedRequest) {
-                    reject(new Error(`timeout reached: only ${storedRequest.current} responses received out of ${storedRequest.expected}`));
-                    this.#requests.delete(requestId);
-                }
-            }, opts.flags.timeout || DEFAULT_TIMEOUT);
-            const storedRequest = {
-                type: MessageType.FETCH_SOCKETS,
-                resolve,
-                timeout,
-                current: 0,
-                expected: expectedResponseCount,
-                responses: localSockets,
-            };
-            this.#requests.set(requestId, storedRequest);
-            this.#publish({
-                type: MessageType.FETCH_SOCKETS,
-                data: {
-                    opts: encodeOptions(opts),
-                    requestId,
-                },
-            });
-        });
-    }
-    serverSideEmit(packet) {
-        const withAck = typeof packet[packet.length - 1] === "function";
-        if (!withAck) {
-            return this.#publish({
-                type: MessageType.SERVER_SIDE_EMIT,
-                data: {
-                    packet,
-                },
-            });
-        }
-        const ack = packet.pop();
-        const expectedResponseCount = this.#getExpectedResponseCount();
-        debug('waiting for %d responses to "serverSideEmit" request', expectedResponseCount);
-        if (expectedResponseCount <= 0) {
-            return ack(null, []);
-        }
-        const requestId = (0, util_1.randomId)();
-        const timeout = setTimeout(() => {
-            const storedRequest = this.#requests.get(requestId);
-            if (storedRequest) {
-                ack(new Error(`timeout reached: only ${storedRequest.current} responses received out of ${storedRequest.expected}`), storedRequest.responses);
-                this.#requests.delete(requestId);
-            }
-        }, DEFAULT_TIMEOUT);
-        const storedRequest = {
-            type: MessageType.SERVER_SIDE_EMIT,
-            resolve: ack,
-            timeout,
-            current: 0,
-            expected: expectedResponseCount,
-            responses: [],
-        };
-        this.#requests.set(requestId, storedRequest);
-        this.#publish({
-            type: MessageType.SERVER_SIDE_EMIT,
-            data: {
-                requestId,
-                packet,
-            },
-        });
-    }
-    #publish(message) {
-        this.#scheduleHeartbeat();
-        return this.doPublish({
-            uid: this.#uid,
-            nsp: this.nsp.name,
-            ...message,
-        });
-    }
-}
-exports.ClusterAdapter = ClusterAdapter;