server/helpers/database-utils.ts

   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
   6 function retryTransactionWrapper <T, A, B, C> (
   7   functionToRetry: (arg1: A, arg2: B, arg3: C) => Promise<T> | Bluebird<T>,
   8   arg1: A,
   9   arg2: B,
  10   arg3: C
  11 ): Promise<T>
  12
  13 function retryTransactionWrapper <T, A, B> (
  14   functionToRetry: (arg1: A, arg2: B) => Promise<T> | Bluebird<T>,
  15   arg1: A,
  16   arg2: B
  17 ): Promise<T>
  18
  19 function retryTransactionWrapper <T, A> (
  20   functionToRetry: (arg1: A) => Promise<T> | Bluebird<T>,
  21   arg1: A
  22 ): Promise<T>
  23
  24 function retryTransactionWrapper <T> (
  25   functionToRetry: (...args: any[]) => Promise<T> | Bluebird<T>,
  26   ...args: any[]
  27 ): Promise<T> {
  28   return transactionRetryer<T>(callback => {
  29     functionToRetry.apply(this, args)
  30         .then((result: T) => callback(null, result))
  31         .catch(err => callback(err))
  32   })
  33   .catch(err => {
  34     logger.error('Cannot execute %s with many retries.', functionToRetry.toString(), { err })
  35     throw err
  36   })
  37 }
  38
  39 function transactionRetryer <T> (func: (err: any, data: T) => any) {
  40   return new Promise<T>((res, rej) => {
  41     retry(
  42       {
  43         times: 5,
  44
  45         errorFilter: err => {
  46           const willRetry = (err.name === 'SequelizeDatabaseError')
  47           logger.debug('Maybe retrying the transaction function.', { willRetry, err })
  48           return willRetry
  49         }
  50       },
  51       func,
  52       (err, data) => err ? rej(err) : res(data)
  53     )
  54   })
  55 }
  56
  57 function updateInstanceWithAnother <T extends Model<T>> (instanceToUpdate: Model<T>, baseInstance: Model<T>) {
  58   const obj = baseInstance.toJSON()
  59
  60   for (const key of Object.keys(obj)) {
  61     instanceToUpdate.set(key, obj[key])
  62   }
  63 }
  64
  65 // ---------------------------------------------------------------------------
  66
  67 export {
  68   retryTransactionWrapper,
  69   transactionRetryer,
  70   updateInstanceWithAnother
  71 }