@proteinjs/db 1.2.0 → 1.3.0

package/src/TableWatcherRunner.ts

@@ -0,0 +1,167 @@
+import { Logger } from '@proteinjs/util';
+import { TableWatcherMap, getTableWatcherMap } from './TableWatcher';
+import { Record } from './Record';
+import { Table } from './Table';
+import { QueryBuilder } from '@proteinjs/db-query';
+
+export class TableWatcherRunner<R extends Record = Record> {
+  private static tableWatcherMap: TableWatcherMap;
+  private logger = new Logger(this.constructor.name);
+
+  constructor() {
+    if (!TableWatcherRunner.tableWatcherMap) {
+      TableWatcherRunner.tableWatcherMap = getTableWatcherMap();
+    }
+  }
+
+  async runBeforeInsertTableWatchers<T extends R>(
+    table: Table<T>,
+    record: Omit<T, keyof R>
+  ): Promise<Omit<T, keyof R>> {
+    const tableWatcherMap = TableWatcherRunner.tableWatcherMap[table.name];
+    if (!tableWatcherMap) {
+      return record;
+    }
+
+    const tableWatchers = tableWatcherMap.beforeInsert;
+    if (tableWatchers.length === 0) {
+      return record;
+    }
+
+    this.logger.info(`(${table.name}) Running before-insert table watchers`);
+    let updatedRecord: any = record;
+    for (const tableWatcher of tableWatchers) {
+      this.logger.info(`(${table.name}) Running ${tableWatcher.name()}.beforeInsert`);
+      updatedRecord = await tableWatcher.beforeInsert(updatedRecord);
+      this.logger.info(`(${table.name}) Finished running ${tableWatcher.name()}.beforeInsert`);
+    }
+    this.logger.info(`(${table.name}) Finished running before-insert table watchers`);
+
+    return updatedRecord;
+  }
+
+  async runAfterInsertTableWatchers<T extends R>(table: Table<T>, record: T): Promise<void> {
+    const tableWatcherMap = TableWatcherRunner.tableWatcherMap[table.name];
+    if (!tableWatcherMap) {
+      return;
+    }
+
+    const tableWatchers = tableWatcherMap.afterInsert;
+    if (tableWatchers.length === 0) {
+      return;
+    }
+
+    this.logger.info(`(${table.name}) Running after-insert table watchers`);
+    for (const tableWatcher of tableWatchers) {
+      this.logger.info(`(${table.name}) Running ${tableWatcher.name()}.afterInsert`);
+      await tableWatcher.afterInsert(record);
+      this.logger.info(`(${table.name}) Finished running ${tableWatcher.name()}.afterInsert`);
+    }
+    this.logger.info(`(${table.name}) Finished running after-insert table watchers`);
+  }
+
+  async runBeforeUpdateTableWatchers<T extends R>(
+    table: Table<T>,
+    record: Partial<T>,
+    qb: QueryBuilder<T>
+  ): Promise<Partial<T>> {
+    const tableWatcherMap = TableWatcherRunner.tableWatcherMap[table.name];
+    if (!tableWatcherMap) {
+      return record;
+    }
+
+    const tableWatchers = tableWatcherMap.beforeUpdate;
+    if (tableWatchers.length === 0) {
+      return record;
+    }
+
+    this.logger.info(`(${table.name}) Running before-update table watchers`);
+    const qbCopy = QueryBuilder.fromQueryBuilder(qb, table.name);
+    let updatedRecord: any = record;
+    for (const tableWatcher of tableWatchers) {
+      this.logger.info(`(${table.name}) Running ${tableWatcher.name()}.beforeUpdate`);
+      updatedRecord = await tableWatcher.beforeUpdate(updatedRecord, qbCopy);
+      this.logger.info(`(${table.name}) Finished running ${tableWatcher.name()}.beforeUpdate`);
+    }
+    this.logger.info(`(${table.name}) Finished running before-update table watchers`);
+
+    return updatedRecord;
+  }
+
+  async runAfterUpdateTableWatchers<T extends R>(
+    table: Table<T>,
+    recordUpdateCount: number,
+    record: Partial<T>,
+    qb: QueryBuilder<T>
+  ): Promise<void> {
+    const tableWatcherMap = TableWatcherRunner.tableWatcherMap[table.name];
+    if (!tableWatcherMap) {
+      return;
+    }
+
+    const tableWatchers = tableWatcherMap.afterUpdate;
+    if (tableWatchers.length === 0) {
+      return;
+    }
+
+    this.logger.info(`(${table.name}) Running after-update table watchers`);
+    const qbCopy = QueryBuilder.fromQueryBuilder(qb, table.name);
+    for (const tableWatcher of tableWatchers) {
+      this.logger.info(`(${table.name}) Running ${tableWatcher.name()}.afterUpdate`);
+      await tableWatcher.afterUpdate(recordUpdateCount, record, qbCopy);
+      this.logger.info(`(${table.name}) Finished running ${tableWatcher.name()}.afterUpdate`);
+    }
+    this.logger.info(`(${table.name}) Finished running after-update table watchers`);
+  }
+
+  async runBeforeDeleteTableWatchers<T extends R>(
+    table: Table<T>,
+    recordsToDelete: T[],
+    qb: QueryBuilder<T>
+  ): Promise<void> {
+    const tableWatcherMap = TableWatcherRunner.tableWatcherMap[table.name];
+    if (!tableWatcherMap) {
+      return;
+    }
+
+    const tableWatchers = tableWatcherMap.beforeDelete;
+    if (tableWatchers.length === 0) {
+      return;
+    }
+
+    this.logger.info(`(${table.name}) Running before-delete table watchers`);
+    const qbCopy = QueryBuilder.fromQueryBuilder(qb, table.name);
+    for (const tableWatcher of tableWatchers) {
+      this.logger.info(`(${table.name}) Running ${tableWatcher.name()}.beforeDelete`);
+      await tableWatcher.beforeDelete(recordsToDelete, qbCopy);
+      this.logger.info(`(${table.name}) Finished running ${tableWatcher.name()}.beforeDelete`);
+    }
+    this.logger.info(`(${table.name}) Finished running before-delete table watchers`);
+  }
+
+  async runAfterDeleteTableWatchers<T extends R>(
+    table: Table<T>,
+    recordDeleteCount: number,
+    deletedRecords: T[],
+    qb: QueryBuilder<T>
+  ): Promise<void> {
+    const tableWatcherMap = TableWatcherRunner.tableWatcherMap[table.name];
+    if (!tableWatcherMap) {
+      return;
+    }
+
+    const tableWatchers = tableWatcherMap.afterDelete;
+    if (tableWatchers.length === 0) {
+      return;
+    }
+
+    this.logger.info(`(${table.name}) Running after-delete table watchers`);
+    const qbCopy = QueryBuilder.fromQueryBuilder(qb, table.name);
+    for (const tableWatcher of tableWatchers) {
+      this.logger.info(`(${table.name}) Running ${tableWatcher.name()}.afterDelete`);
+      await tableWatcher.afterDelete(recordDeleteCount, deletedRecords, qbCopy);
+      this.logger.info(`(${table.name}) Finished running ${tableWatcher.name()}.afterDelete`);
+    }
+    this.logger.info(`(${table.name}) Finished running after-delete table watchers`);
+  }
+}