sequelize-adapter/test/myUnitOfWork.ts

Use Unit Of Wrok pattern to wrap sequelize up.

import { Op, Sequelize } from 'sequelize';
import { IChangeObject } from '../src/IChangeObject';
import { UnitOfWorkBase } from '../src/unitOfWorkBase';
import { UserRepository } from './userRepository';

function getOperatorsAliases() {
  const op = Op;
  return {
    $eq: op.eq,
    $ne: op.ne,
    $gte: op.gte,
    $gt: op.gt,
    $lte: op.lte,
    $lt: op.lt,
    $not: op.not,
    $in: op.in,
    $notIn: op.notIn,
    $is: op.is,
    $like: op.like,
    $notLike: op.notLike,
    $iLike: op.iLike,
    $notILike: op.notILike,
    $regexp: op.regexp,
    $notRegexp: op.notRegexp,
    $iRegexp: op.iRegexp,
    $notIRegexp: op.notIRegexp,
    $between: op.between,
    $notBetween: op.notBetween,
    $overlap: op.overlap,
    $contains: op.contains,
    $contained: op.contained,
    $adjacent: op.adjacent,
    $strictLeft: op.strictLeft,
    $strictRight: op.strictRight,
    $noExtendRight: op.noExtendRight,
    $noExtendLeft: op.noExtendLeft,
    $and: op.and,
    $or: op.or,
    $any: op.any,
    $all: op.all,
    // $values: op.values,
    $col: op.col
  };
}

export class MyUnitOfWork extends UnitOfWorkBase {

  private static _instance: Sequelize;
  static async getInstance() {
    if (!MyUnitOfWork._instance) {
      const operatorsAliases = getOperatorsAliases();

      // setup db => test in memory
      const setting = {
        host: 'localhost',
        dbName: 'test_db',
        username: 'root',
        password: '1234',
        type: 'sqlite', // 'mysql'
      };
      const s = new Sequelize(setting.dbName, setting.username, setting.password, {
        dialect: 'sqlite',
        pool: {
          max: 5,
          min: 0,
          acquire: 30000,
          idle: 10000,
        },
        logging: false,
        operatorsAliases
      });

      try {
        await s.authenticate();
        console.log('connect db successfully.');
      } catch (err) {
        throw err;
      }

      MyUnitOfWork._instance = s;
    }
    return new this(MyUnitOfWork._instance);
  }

  private constructor(public db: Sequelize) {
    super();
    this.init();
  }

  reps = {
    user: new UserRepository(this),
  };

  beforeSaveChange(addedEntities: IChangeObject[], updatedEntities: IChangeObject[], deletedEntities: IChangeObject[]) {
    // do something...
  }
  afterSaveChange() {
    // do something...
  }

  async close() {
    MyUnitOfWork._instance = null;
    await this.db.close();
  }

  private init() {
    // setup retrying setting
    this.retryingOption = {
      count: 3,
      watingMillisecond: 1000
    };

    // setup repositories
    this.__reps = this.reps;
  }

}