scoresaber-reloaded/src/db/cache.js

import eventBus from "../utils/broadcast-channel-pubsub";

export default (name, getObjKey) => {
  let cache = {};

  // update data cached on another node
  eventBus.on("cache-key-set-" + name, ({ key, value }, isLocal) =>
    !isLocal ? set(key, value, false) : null
  );
  eventBus.on("cache-all-set" + name, ({ data }, isLocal) =>
    !isLocal ? setAll(data, false) : null
  );
  eventBus.on("cache-merge-" + name, ({ data }, isLocal) =>
    !isLocal ? merge(data, false) : null
  );
  eventBus.on("cache-key-forget-" + name, ({ key }, isLocal) =>
    !isLocal ? forget(key, false) : null
  );
  eventBus.on("cache-flush-" + name, (_, isLocal) =>
    !isLocal ? flush(false) : null
  );

  const set = (key, value, emitEvent = true) => {
    cache[key] = value;

    if (emitEvent) eventBus.publish("cache-key-set-" + name, { key, value });

    return value;
  };

  const setAll = (data, emitEvent = true) => {
    cache = data;

    if (emitEvent) eventBus.publish("cache-all-set-" + name, { data });

    return cache;
  };
  const merge = (data, emitEvent = true) => {
    cache = { ...cache, ...data };

    if (emitEvent) eventBus.publish("cache-merge-" + name, { data });

    return cache;
  };

  const get = async (key, fetchFunc) => {
    if (cache.hasOwnProperty(key)) return cache[key];

    const value = await fetchFunc();

    return set(key, value);
  };

  const getByFilter = async (fetchFunc, filterFunc) => {
    if (filterFunc) {
      const obj = Object.values(cache).find(filterFunc);
      if (obj) return obj;
    }

    const value = await fetchFunc();
    if (value === undefined) return value;

    const key = getObjKey(value);

    return set(key, value);
  };

  const getAll = () => cache;

  const has = (key) => cache[key] !== undefined;

  const getKeys = () => Object.keys(cache);

  const forget = (key, emitEvent = true) => {
    delete cache[key];

    if (emitEvent) eventBus.publish("cache-key-forget-" + name, { key });

    return cache;
  };

  const forgetByFilter = (filterFunc, emitEvent = true) => {
    if (!filterFunc) return false;

    Object.keys(cache)
      .filter((key) => filterFunc(cache[key]))
      .forEach((key) => {
        delete cache[key];

        if (emitEvent) eventBus.publish("cache-key-forget-" + name, { key });
      });

    return true;
  };

  const flush = (emitEvent = true) => {
    cache = {};

    if (emitEvent) eventBus.publish("cache-flush-" + name, {});

    return cache;
  };

  return {
    has,
    get,
    getByFilter,
    getAll,
    set,
    setAll,
    merge,
    getKeys,
    forget,
    forgetByFilter,
    flush,
  };
};