import type { MappedPromiseItem } from './trpc';

export const promiseMap = new Map<number, Map<string, MappedPromiseItem>>();

/**
 * Clear mapped promises by socketId.
 * @param socketId The id of socket
 */
export const clearMappedPromises = (socketId: number) => {
  if (socketId) {
    promiseMap.get(socketId)?.clear();
    promiseMap.delete(socketId);
  }
};

/**
 * Clear all the mapped promises.
 */
export const clearAllMappedPromises = () => {
  for (const map of promiseMap.values()) {
    map.clear();
  }
  promiseMap.clear();
};