@the_pixelport/aws-lambda-graphql/dist/MemorySubscriptionManager.js

Apollo server for AWS Lambda with WebSocket subscriptions support over API Gateway v1 + v2

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.MemorySubscriptionManager = void 0;
const iterall_1 = require("iterall");
// polyfill Symbol.asyncIterator
if (Symbol.asyncIterator === undefined) {
    Symbol.asyncIterator = Symbol.for('asyncIterator');
}
class MemorySubscriptionManager {
    constructor({ getSubscriptionNameFromEvent = (event) => event.event, getSubscriptionNameFromConnection = (name) => name, } = {}) {
        this.subscribersByEvent = (event) => {
            return {
                [Symbol.asyncIterator]: () => {
                    const name = this.getSubscriptionNameFromEvent(event);
                    const subscriptions = this.subscriptions.get(name) || [];
                    const subscribers = subscriptions.filter((subscriber) => subscriber.event === name);
                    return (0, iterall_1.createAsyncIterator)([subscribers]);
                },
            };
        };
        this.subscribe = async (names, connection, operation) => {
            names.forEach((n) => {
                const name = this.getSubscriptionNameFromConnection(n, connection);
                const subscriptions = this.subscriptions.get(name);
                const subscription = {
                    connection,
                    operation,
                    event: name,
                    operationId: operation.operationId,
                };
                if (subscriptions == null) {
                    this.subscriptions.set(name, [subscription]);
                }
                else if (!subscriptions.find((s) => s.connection.id === connection.id)) {
                    subscriptions.push({
                        connection,
                        operation,
                        event: name,
                        operationId: operation.operationId,
                    });
                }
            });
        };
        this.unsubscribe = async (subscriber) => {
            const subscriptions = this.subscriptions.get(subscriber.event);
            if (subscriptions) {
                this.subscriptions.set(subscriber.event, subscriptions.filter((s) => s.connection.id !== subscriber.connection.id));
            }
        };
        this.unsubscribeOperation = async (connectionId, operationId) => {
            this.subscriptions.forEach((subscribers, event) => {
                this.subscriptions.set(event, subscribers.filter((subscriber) => subscriber.connection.id !== connectionId &&
                    subscriber.operationId !== operationId));
            });
        };
        this.unsubscribeAllByConnectionId = (connectionId) => {
            for (const key of this.subscriptions.keys()) {
                this.subscriptions.set(key, this.subscriptions
                    .get(key)
                    .filter((s) => s.connection.id === connectionId));
            }
            return Promise.resolve();
        };
        this.subscriptions = new Map();
        this.getSubscriptionNameFromEvent = getSubscriptionNameFromEvent;
        this.getSubscriptionNameFromConnection = getSubscriptionNameFromConnection;
    }
}
exports.MemorySubscriptionManager = MemorySubscriptionManager;
//# sourceMappingURL=MemorySubscriptionManager.js.map