Class ModelService

packages/axe-api/src/Services/ModelService.ts:12–97

import { ModelHooks, PhaseFunction, SerializationFunction } from "../Types";

class ModelService implements IModelService {
  name: string;
  instance: Model;
  relations: IRelation[];
  columns: IColumn[];
  columnNames: string[];
  hooks: ModelHooks = {} as ModelHooks;
  events: ModelHooks = {} as ModelHooks;
  children: IModelService[];
  isRecursive: boolean;
  queryLimits: IQueryLimitConfig[];
  serialize: SerializationFunction | null;
  cacheConfiguration: Record<string, ICacheConfiguration>;

  constructor(name: string, instance: Model) {
    this.name = name;
    this.instance = instance;
    this.relations = [];
    this.columns = [];
    this.columnNames = [];
    this.children = [];
    this.isRecursive = false;
    this.queryLimits = [];
    this.serialize = null;
    this.cacheConfiguration = {};
  }

  setColumns(columns: IColumn[]) {
    this.columns = columns;
    this.columnNames = this.columns.map((i) => i.name);
  }

  setCacheConfiguration(handler: string, cache: ICacheConfiguration) {
    this.cacheConfiguration[handler] = cache;
  }

  getCacheConfiguration(handler: HandlerTypes) {
    return this.cacheConfiguration[handler];
  }

  setExtensions(
    type: Extensions,
    hookFunctionType: HookFunctionTypes,
    data: PhaseFunction,
  ) {
    if (type == Extensions.Hooks) {
      this.setHooks(hookFunctionType, data);
    } else if (type == Extensions.Events) {
      this.setEvents(hookFunctionType, data);
    } else {
      throw new Error("Undefined hook type.");
    }
  }

  setQueryLimits(limits: IQueryLimitConfig[]) {
    this.queryLimits = limits;
  }