]>
Commit | Line | Data |
---|---|---|
4d4e5cd4 | 1 | import * as retry from 'async/retry' |
20494f12 | 2 | import * as Bluebird from 'bluebird' |
06215f15 | 3 | import { Model } from 'sequelize-typescript' |
65fcc311 | 4 | import { logger } from './logger' |
2284f202 | 5 | import { Transaction } from 'sequelize' |
4145c1c6 | 6 | |
77d7e851 C |
7 | function retryTransactionWrapper <T, A, B, C, D> ( |
8 | functionToRetry: (arg1: A, arg2: B, arg3: C, arg4: D) => Promise<T> | Bluebird<T>, | |
9 | arg1: A, | |
10 | arg2: B, | |
11 | arg3: C, | |
12 | arg4: D, | |
13 | ): Promise<T> | |
14 | ||
90d4bb81 C |
15 | function retryTransactionWrapper <T, A, B, C> ( |
16 | functionToRetry: (arg1: A, arg2: B, arg3: C) => Promise<T> | Bluebird<T>, | |
17 | arg1: A, | |
18 | arg2: B, | |
19 | arg3: C | |
20 | ): Promise<T> | |
21 | ||
22 | function retryTransactionWrapper <T, A, B> ( | |
23 | functionToRetry: (arg1: A, arg2: B) => Promise<T> | Bluebird<T>, | |
24 | arg1: A, | |
25 | arg2: B | |
26 | ): Promise<T> | |
27 | ||
28 | function retryTransactionWrapper <T, A> ( | |
29 | functionToRetry: (arg1: A) => Promise<T> | Bluebird<T>, | |
30 | arg1: A | |
31 | ): Promise<T> | |
32 | ||
2baea0c7 C |
33 | function retryTransactionWrapper <T> ( |
34 | functionToRetry: () => Promise<T> | Bluebird<T> | |
35 | ): Promise<T> | |
36 | ||
0f91ae62 | 37 | function retryTransactionWrapper <T> ( |
90d4bb81 C |
38 | functionToRetry: (...args: any[]) => Promise<T> | Bluebird<T>, |
39 | ...args: any[] | |
0f91ae62 | 40 | ): Promise<T> { |
0f91ae62 | 41 | return transactionRetryer<T>(callback => { |
2baea0c7 | 42 | functionToRetry.apply(null, args) |
0f91ae62 | 43 | .then((result: T) => callback(null, result)) |
6fcd19ba | 44 | .catch(err => callback(err)) |
075f16ca | 45 | }) |
6fcd19ba | 46 | .catch(err => { |
1e11f67b | 47 | logger.error(`Cannot execute ${functionToRetry.name} with many retries.`, { err }) |
20494f12 | 48 | throw err |
6fcd19ba | 49 | }) |
4df023f2 C |
50 | } |
51 | ||
0f91ae62 C |
52 | function transactionRetryer <T> (func: (err: any, data: T) => any) { |
53 | return new Promise<T>((res, rej) => { | |
90d4bb81 C |
54 | retry( |
55 | { | |
56 | times: 5, | |
57 | ||
58 | errorFilter: err => { | |
59 | const willRetry = (err.name === 'SequelizeDatabaseError') | |
60 | logger.debug('Maybe retrying the transaction function.', { willRetry, err }) | |
61 | return willRetry | |
62 | } | |
63 | }, | |
64 | func, | |
65 | (err, data) => err ? rej(err) : res(data) | |
66 | ) | |
4df023f2 C |
67 | }) |
68 | } | |
69 | ||
c39ea24b | 70 | function updateInstanceWithAnother <T extends Model<T>> (instanceToUpdate: Model<T>, baseInstance: Model<T>) { |
a5625b41 C |
71 | const obj = baseInstance.toJSON() |
72 | ||
73 | for (const key of Object.keys(obj)) { | |
1735c825 | 74 | instanceToUpdate[key] = obj[key] |
a5625b41 C |
75 | } |
76 | } | |
77 | ||
06215f15 C |
78 | function resetSequelizeInstance (instance: Model<any>, savedFields: object) { |
79 | Object.keys(savedFields).forEach(key => { | |
1735c825 | 80 | instance[key] = savedFields[key] |
06215f15 C |
81 | }) |
82 | } | |
83 | ||
2284f202 C |
84 | function afterCommitIfTransaction (t: Transaction, fn: Function) { |
85 | if (t) return t.afterCommit(() => fn()) | |
86 | ||
87 | return fn() | |
88 | } | |
89 | ||
d7a25329 C |
90 | function deleteNonExistingModels <T extends { hasSameUniqueKeysThan (other: T): boolean } & Model<T>> ( |
91 | fromDatabase: T[], | |
92 | newModels: T[], | |
93 | t: Transaction | |
94 | ) { | |
95 | return fromDatabase.filter(f => !newModels.find(newModel => newModel.hasSameUniqueKeysThan(f))) | |
96 | .map(f => f.destroy({ transaction: t })) | |
97 | } | |
98 | ||
4df023f2 C |
99 | // --------------------------------------------------------------------------- |
100 | ||
65fcc311 | 101 | export { |
06215f15 | 102 | resetSequelizeInstance, |
65fcc311 | 103 | retryTransactionWrapper, |
a5625b41 | 104 | transactionRetryer, |
2284f202 | 105 | updateInstanceWithAnother, |
d7a25329 C |
106 | afterCommitIfTransaction, |
107 | deleteNonExistingModels | |
65fcc311 | 108 | } |