]>
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 | |
90d4bb81 C |
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 | ||
0f91ae62 | 24 | function retryTransactionWrapper <T> ( |
90d4bb81 C |
25 | functionToRetry: (...args: any[]) => Promise<T> | Bluebird<T>, |
26 | ...args: any[] | |
0f91ae62 | 27 | ): Promise<T> { |
0f91ae62 | 28 | return transactionRetryer<T>(callback => { |
075f16ca | 29 | functionToRetry.apply(this, args) |
0f91ae62 | 30 | .then((result: T) => callback(null, result)) |
6fcd19ba | 31 | .catch(err => callback(err)) |
075f16ca | 32 | }) |
6fcd19ba | 33 | .catch(err => { |
90d4bb81 | 34 | logger.error('Cannot execute %s with many retries.', functionToRetry.toString(), { err }) |
20494f12 | 35 | throw err |
6fcd19ba | 36 | }) |
4df023f2 C |
37 | } |
38 | ||
0f91ae62 C |
39 | function transactionRetryer <T> (func: (err: any, data: T) => any) { |
40 | return new Promise<T>((res, rej) => { | |
90d4bb81 C |
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 | ) | |
4df023f2 C |
54 | }) |
55 | } | |
56 | ||
c39ea24b | 57 | function updateInstanceWithAnother <T extends Model<T>> (instanceToUpdate: Model<T>, baseInstance: Model<T>) { |
a5625b41 C |
58 | const obj = baseInstance.toJSON() |
59 | ||
60 | for (const key of Object.keys(obj)) { | |
61 | instanceToUpdate.set(key, obj[key]) | |
62 | } | |
63 | } | |
64 | ||
4df023f2 C |
65 | // --------------------------------------------------------------------------- |
66 | ||
65fcc311 | 67 | export { |
65fcc311 | 68 | retryTransactionWrapper, |
a5625b41 C |
69 | transactionRetryer, |
70 | updateInstanceWithAnother | |
65fcc311 | 71 | } |