hub.ts

import * as fmt from './sse_formatter';
import { ISseFunctions } from './sse_middleware';

export interface IHub extends ISseFunctions {
    /**
     * Remember a new client.
     *
     * @param funcs SSE functions bound to the client.
     */
    register(funcs: ISseFunctions): void;

    /**
     * Unregister a known client.
     *
     * @param funcs SSE function to unregister.
     */
    unregister(funcs: ISseFunctions): void;
}

export class Hub implements IHub {
    protected readonly clients = new Set<ISseFunctions>();

    public register(funcs: ISseFunctions): void {
        this.clients.add(funcs);
    }

    public unregister(funcs: ISseFunctions): void {
        this.clients.delete(funcs);
    }

    public data(data: fmt.SseValue, id?: string): void {
        this.clients.forEach(client => client.data(data, id));
    }

    public event(event: string, data: fmt.SseValue, id?: string): void {
        this.clients.forEach(client => client.event(event, data, id));
    }

    public comment(comment: string): void {
        this.clients.forEach(client => client.comment(comment));
    }
}