packages/shared/lib/helpers/cache.ts

   1 import type { Listeners } from './listeners';
   2 import createListeners from './listeners';
   3
   4 export interface Cache<K, V> extends Pick<Listeners<[K, V | undefined], void>, 'subscribe'>, Map<K, V> {
   5     clearListeners: () => void;
   6 }
   7
   8 /**
   9  * Wraps a map with support for subscribe/unsubscribe on changes.
  10  */
  11 const createCache = <K, V>(map = new Map<K, V>()): Cache<K, V> => {
  12     const listeners = createListeners<[K, V | undefined], void>();
  13     return {
  14         get size() {
  15             return map.size;
  16         },
  17         forEach(callbackfn: (value: V, key: K, map: Map<K, V>) => void, thisArg?: any) {
  18             return map.forEach(callbackfn, thisArg);
  19         },
  20         has: (key: K) => map.has(key),
  21         get: (key: K) => map.get(key),
  22         entries: () => map.entries(),
  23         keys: () => map.keys(),
  24         values: () => map.values(),
  25         [Symbol.iterator]: () => map[Symbol.iterator](),
  26         get [Symbol.toStringTag]() {
  27             return map[Symbol.toStringTag];
  28         },
  29         clear: () => map.clear(),
  30         delete: (key: K) => {
  31             const r = map.delete(key);
  32             listeners.notify(key, undefined);
  33             return r;
  34         },
  35         set(key: K, value: V) {
  36             map.set(key, value);
  37             listeners.notify(key, value);
  38             return this;
  39         },
  40         subscribe: listeners.subscribe,
  41         clearListeners: listeners.clear,
  42     };
  43 };
  44
  45 export default createCache;