Function makeMemoryStore

packages/infra/src/Store/Memory.ts:514–569

()

}

export const makeMemoryStore = () => ({
  make: Effect.fnUntraced(function*<IdKey extends keyof Encoded, Encoded extends FieldValues, R, E>(
    modelName: string,
    idKey: IdKey,
    seed?: Effect.Effect<Iterable<Encoded>, E, R>,
    config?: StoreConfig<Encoded>
  ) {
    const primary = yield* makeMemoryStoreInt<IdKey, Encoded, R, E>(
      modelName,
      idKey,
      "primary",
      seed,
      config?.defaultValues
    )
    const ctx = yield* Effect.context<R>()
    const stores = new Map([["primary", primary]])
    const semaphores = new Map<string, Semaphore.Semaphore>()
    const getSem = (ns: string) => {
      let sem = semaphores.get(ns)
      if (!sem) {
        sem = Semaphore.makeUnsafe(1)
        semaphores.set(ns, sem)
      }
      return sem
    }
    const ensureStore = (namespace: string) =>
      getSem(namespace).withPermits(1)(Effect.suspend(() => {
        const store = stores.get(namespace)
        if (store) return Effect.succeed(store)
        if (config?.allowNamespace && !config.allowNamespace(namespace)) {
          throw new Error(`Namespace ${namespace} not allowed!`)
        }
        return makeMemoryStoreInt(modelName, idKey, namespace, seed, config?.defaultValues)
          .pipe(
            Effect.orDie,
            Effect.provide(ctx),
            Effect.tap((store) => Effect.sync(() => stores.set(namespace, store)))
          )
      }))
    const getStore = !config?.allowNamespace
      ? Effect.succeed(primary)
      : storeId.pipe(Effect.flatMap((namespace) => ensureStore(namespace)))
    const s: Store<IdKey, Encoded> = {
      seedNamespace: (namespace) => ensureStore(namespace).pipe(Effect.asVoid),
      all: Effect.flatMap(getStore, (_) => _.all),
      queryRaw: (...args) => Effect.flatMap(getStore, (_) => _.queryRaw(...args)),
      find: (...args) => Effect.flatMap(getStore, (_) => _.find(...args)),
      filter: (...args) => Effect.flatMap(getStore, (_) => _.filter(...args)),
      set: (...args) => Effect.flatMap(getStore, (_) => _.set(...args)),
      batchSet: (...args) => Effect.flatMap(getStore, (_) => _.batchSet(...args)),
      bulkSet: (...args) => Effect.flatMap(getStore, (_) => _.bulkSet(...args)),
      batchRemove: (...args) => Effect.flatMap(getStore, (_) => _.batchRemove(...args))
    }
    return s
  })
})

export const MemoryStoreLive = StoreMaker.toLayer(Effect.sync(() => makeMemoryStore()))