]>
Commit | Line | Data |
---|---|---|
65fcc311 | 1 | // TODO: import from ES6 when retry typing file will include errorFilter function |
4d4e5cd4 | 2 | import * as retry from 'async/retry' |
6fcd19ba | 3 | import * as Promise from 'bluebird' |
4df023f2 | 4 | |
65fcc311 | 5 | import { logger } from './logger' |
4145c1c6 | 6 | |
69818c93 | 7 | type RetryTransactionWrapperOptions = { errorMessage: string, arguments?: any[] } |
6fcd19ba | 8 | function retryTransactionWrapper (functionToRetry: (... args) => Promise<any>, options: RetryTransactionWrapperOptions) { |
4df023f2 C |
9 | const args = options.arguments ? options.arguments : [] |
10 | ||
6fcd19ba | 11 | return transactionRetryer( |
4df023f2 | 12 | function (callback) { |
6fcd19ba C |
13 | functionToRetry.apply(this, args) |
14 | .then(result => callback(null, result)) | |
15 | .catch(err => callback(err)) | |
4df023f2 C |
16 | } |
17 | ) | |
6fcd19ba C |
18 | .catch(err => { |
19 | // Do not throw the error, continue the process | |
ad0997ad | 20 | logger.error(options.errorMessage, err) |
6fcd19ba | 21 | }) |
4df023f2 C |
22 | } |
23 | ||
6fcd19ba C |
24 | function transactionRetryer (func: Function) { |
25 | return new Promise((res, rej) => { | |
26 | retry({ | |
27 | times: 5, | |
4df023f2 | 28 | |
6fcd19ba C |
29 | errorFilter: function (err) { |
30 | const willRetry = (err.name === 'SequelizeDatabaseError') | |
31 | logger.debug('Maybe retrying the transaction function.', { willRetry }) | |
32 | return willRetry | |
33 | } | |
34 | }, func, function (err) { | |
35 | err ? rej(err) : res() | |
36 | }) | |
4df023f2 C |
37 | }) |
38 | } | |
39 | ||
40 | // --------------------------------------------------------------------------- | |
41 | ||
65fcc311 | 42 | export { |
65fcc311 | 43 | retryTransactionWrapper, |
65fcc311 C |
44 | transactionRetryer |
45 | } |