1 import * as retry from 'async/retry'
2 import * as Bluebird from 'bluebird'
3 import { Model } from 'sequelize-typescript'
4 import { logger } from './logger'
5 import { Transaction } from 'sequelize'
7 function retryTransactionWrapper <T, A, B, C, D> (
8 functionToRetry: (arg1: A, arg2: B, arg3: C, arg4: D) => Promise<T> | Bluebird<T>,
15 function retryTransactionWrapper <T, A, B, C> (
16 functionToRetry: (arg1: A, arg2: B, arg3: C) => Promise<T> | Bluebird<T>,
22 function retryTransactionWrapper <T, A, B> (
23 functionToRetry: (arg1: A, arg2: B) => Promise<T> | Bluebird<T>,
28 function retryTransactionWrapper <T, A> (
29 functionToRetry: (arg1: A) => Promise<T> | Bluebird<T>,
33 function retryTransactionWrapper <T> (
34 functionToRetry: () => Promise<T> | Bluebird<T>
37 function retryTransactionWrapper <T> (
38 functionToRetry: (...args: any[]) => Promise<T> | Bluebird<T>,
41 return transactionRetryer<T>(callback => {
42 functionToRetry.apply(null, args)
43 .then((result: T) => callback(null, result))
44 .catch(err => callback(err))
47 logger.error(`Cannot execute ${functionToRetry.name} with many retries.`, { err })
52 function transactionRetryer <T> (func: (err: any, data: T) => any) {
53 return new Promise<T>((res, rej) => {
59 const willRetry = (err.name === 'SequelizeDatabaseError')
60 logger.debug('Maybe retrying the transaction function.', { willRetry, err })
65 (err, data) => err ? rej(err) : res(data)
70 function updateInstanceWithAnother <T extends Model<T>> (instanceToUpdate: Model<T>, baseInstance: Model<T>) {
71 const obj = baseInstance.toJSON()
73 for (const key of Object.keys(obj)) {
74 instanceToUpdate[key] = obj[key]
78 function resetSequelizeInstance (instance: Model<any>, savedFields: object) {
79 Object.keys(savedFields).forEach(key => {
80 instance[key] = savedFields[key]
84 function afterCommitIfTransaction (t: Transaction, fn: Function) {
85 if (t) return t.afterCommit(() => fn())
90 function deleteNonExistingModels <T extends { hasSameUniqueKeysThan (other: T): boolean } & Model<T>> (
95 return fromDatabase.filter(f => !newModels.find(newModel => newModel.hasSameUniqueKeysThan(f)))
96 .map(f => f.destroy({ transaction: t }))
99 // ---------------------------------------------------------------------------
102 resetSequelizeInstance,
103 retryTransactionWrapper,
105 updateInstanceWithAnother,
106 afterCommitIfTransaction,
107 deleteNonExistingModels