Class PivotCollectionPersister

packages/sql/src/PivotCollectionPersister.ts:59–258

}

export class PivotCollectionPersister<Entity extends object> {
  readonly #inserts = new Map<string, InsertStatement<Entity>>();
  readonly #upserts = new Map<string, InsertStatement<Entity>>();
  readonly #deletes = new Map<string, DeleteStatement<Entity>>();
  readonly #batchSize: number;
  #order = 0;
  readonly #meta: EntityMetadata<Entity>;
  readonly #driver: AbstractSqlDriver;
  readonly #ctx?: Transaction;
  readonly #schema?: string;
  readonly #loggerContext?: Dictionary;
  readonly #abort?: AbortQueryOptions;

  constructor(
    meta: EntityMetadata<Entity>,
    driver: AbstractSqlDriver,
    ctx?: Transaction,
    schema?: string,
    loggerContext?: Dictionary,
    abort?: AbortQueryOptions,
  ) {
    this.#meta = meta;
    this.#driver = driver;
    this.#ctx = ctx;
    this.#schema = schema;
    this.#loggerContext = loggerContext;
    this.#abort = abort;
    this.#batchSize = this.#driver.config.get('batchSize');
  }

  enqueueUpdate(
    prop: EntityProperty<Entity>,
    insertDiff: Primary<Entity>[][],
    deleteDiff: Primary<Entity>[][] | boolean,
    pks: Primary<Entity>[],
    isInitialized = true,
  ) {
    if (insertDiff.length) {
      if (isInitialized) {
        this.enqueueInsert(prop, insertDiff, pks);
      } else {
        this.enqueueUpsert(prop, insertDiff, pks);
      }
    }

    if (deleteDiff === true || (Array.isArray(deleteDiff) && deleteDiff.length)) {
      this.enqueueDelete(prop, deleteDiff, pks);
    }
  }

  private enqueueInsert(prop: EntityProperty<Entity>, insertDiff: Primary<Entity>[][], pks: Primary<Entity>[]) {
    for (const fks of insertDiff) {
      const statement = this.createInsertStatement(prop, fks, pks);
      const hash = statement.getHash();

      if (prop.owner || !this.#inserts.has(hash)) {
        this.#inserts.set(hash, statement);
      }