]>
Commit | Line | Data |
---|---|---|
4d4e5cd4 | 1 | import * as retry from 'async/retry' |
20494f12 | 2 | import * as Bluebird from 'bluebird' |
a5625b41 | 3 | import { Model } from 'sequelize-typescript' |
65fcc311 | 4 | import { logger } from './logger' |
4145c1c6 | 5 | |
69818c93 | 6 | type RetryTransactionWrapperOptions = { errorMessage: string, arguments?: any[] } |
0f91ae62 C |
7 | function retryTransactionWrapper <T> ( |
8 | functionToRetry: (...args) => Promise<T> | Bluebird<T>, | |
9 | options: RetryTransactionWrapperOptions | |
10 | ): Promise<T> { | |
4df023f2 C |
11 | const args = options.arguments ? options.arguments : [] |
12 | ||
0f91ae62 | 13 | return transactionRetryer<T>(callback => { |
075f16ca | 14 | functionToRetry.apply(this, args) |
0f91ae62 | 15 | .then((result: T) => callback(null, result)) |
6fcd19ba | 16 | .catch(err => callback(err)) |
075f16ca | 17 | }) |
6fcd19ba | 18 | .catch(err => { |
d5b7d911 | 19 | logger.error(options.errorMessage, { err }) |
20494f12 | 20 | throw err |
6fcd19ba | 21 | }) |
4df023f2 C |
22 | } |
23 | ||
0f91ae62 C |
24 | function transactionRetryer <T> (func: (err: any, data: T) => any) { |
25 | return new Promise<T>((res, rej) => { | |
6fcd19ba C |
26 | retry({ |
27 | times: 5, | |
4df023f2 | 28 | |
075f16ca | 29 | errorFilter: err => { |
6fcd19ba | 30 | const willRetry = (err.name === 'SequelizeDatabaseError') |
9d3ef9fe | 31 | logger.debug('Maybe retrying the transaction function.', { willRetry, err }) |
6fcd19ba C |
32 | return willRetry |
33 | } | |
20494f12 | 34 | }, func, (err, data) => err ? rej(err) : res(data)) |
4df023f2 C |
35 | }) |
36 | } | |
37 | ||
c39ea24b | 38 | function updateInstanceWithAnother <T extends Model<T>> (instanceToUpdate: Model<T>, baseInstance: Model<T>) { |
a5625b41 C |
39 | const obj = baseInstance.toJSON() |
40 | ||
41 | for (const key of Object.keys(obj)) { | |
42 | instanceToUpdate.set(key, obj[key]) | |
43 | } | |
44 | } | |
45 | ||
4df023f2 C |
46 | // --------------------------------------------------------------------------- |
47 | ||
65fcc311 | 48 | export { |
65fcc311 | 49 | retryTransactionWrapper, |
a5625b41 C |
50 | transactionRetryer, |
51 | updateInstanceWithAnother | |
65fcc311 | 52 | } |