Class EntityDataStore

packages/db/src/entity-data-store.ts:16–185

import { deepObjectAssign } from './utils/deep-merge.js';

export class EntityDataStore implements EntityStore {
  constructor(public storagePrefix: string[] = []) {}

  getEntity(
    storage: KVStoreOrTransaction,
    collection: string,
    id: string
  ): Promise<DBEntity | undefined> {
    const prefixedStorage = storage.scope(this.storagePrefix);
    return prefixedStorage.get([collection, id]);
  }

  async getCollectionStats(
    storage: KVStoreOrTransaction,
    knownCollections: CollectionName[] | undefined
  ): Promise<Map<string, number>> {
    const prefixedStorage = storage.scope(this.storagePrefix);
    const stats = new Map<string, number>();
    if (knownCollections) {
      for (const collection of knownCollections) {
        const count = await prefixedStorage.count({ prefix: [collection] });
        stats.set(collection, count);
      }
    } else {
      for await (const [[collection]] of prefixedStorage.scan({
        prefix: [],
      })) {
        stats.set(
          collection as string,
          (stats.get(collection as string) || 0) + 1
        );
      }
    }
    return stats;
  }

  async applyChanges(
    tx: KVStoreTransaction,
    changes: DBChanges,
    options: ApplyChangesOptions
  ): Promise<DBChanges> {
    const prefixedTx = tx.scope(this.storagePrefix);
    const appliedChanges: DBChanges = {};
    const deltas: Delta[] = [];
    const getInsertChangeset = async (
      collection: string,
      id: string,
      change: Change
    ): Promise<Delta> => {
      const current = await this.getEntity(tx, collection, id);
      const isUpsert = !!current;
      if (options.entityChangeValidator) {
        options.entityChangeValidator(collection, change, {
          ignoreRequiredProperties: isUpsert,
        });
      }
      return {
        collection,