react-use-websocket/src/lib/manage-subscribers.ts

React Hook for WebSocket communication

import { Subscriber } from './types';

export type Subscribers = {
    [url: string]: Set<Subscriber>,
}
  
const subscribers: Subscribers = {};
const EMPTY_LIST: Subscriber[] = [];

export const getSubscribers = (url: string): Subscriber[] => {
    if (hasSubscribers(url)) {
        return Array.from(subscribers[url]);
    }
    return EMPTY_LIST;
};

export const hasSubscribers = (url: string): boolean => {
    return subscribers[url]?.size > 0;
};

export const addSubscriber = (url: string, subscriber: Subscriber): void => {
    subscribers[url] = subscribers[url] || new Set<Subscriber>();
    subscribers[url].add(subscriber);
};

export const removeSubscriber = (url: string, subscriber: Subscriber): void => {
    subscribers[url].delete(subscriber);
};