]>
Commit | Line | Data |
---|---|---|
41fb13c3 C |
1 | import retry from 'async/retry' |
2 | import Bluebird from 'bluebird' | |
fa47956e | 3 | import { Transaction } from 'sequelize' |
06215f15 | 4 | import { Model } from 'sequelize-typescript' |
e024fd6a | 5 | import { sequelizeTypescript } from '@server/initializers/database' |
65fcc311 | 6 | import { logger } from './logger' |
4145c1c6 | 7 | |
77d7e851 | 8 | function retryTransactionWrapper <T, A, B, C, D> ( |
883a9019 | 9 | functionToRetry: (arg1: A, arg2: B, arg3: C, arg4: D) => Promise<T>, |
77d7e851 C |
10 | arg1: A, |
11 | arg2: B, | |
12 | arg3: C, | |
13 | arg4: D, | |
14 | ): Promise<T> | |
15 | ||
90d4bb81 | 16 | function retryTransactionWrapper <T, A, B, C> ( |
883a9019 | 17 | functionToRetry: (arg1: A, arg2: B, arg3: C) => Promise<T>, |
90d4bb81 C |
18 | arg1: A, |
19 | arg2: B, | |
20 | arg3: C | |
21 | ): Promise<T> | |
22 | ||
23 | function retryTransactionWrapper <T, A, B> ( | |
883a9019 | 24 | functionToRetry: (arg1: A, arg2: B) => Promise<T>, |
90d4bb81 C |
25 | arg1: A, |
26 | arg2: B | |
27 | ): Promise<T> | |
28 | ||
29 | function retryTransactionWrapper <T, A> ( | |
883a9019 | 30 | functionToRetry: (arg1: A) => Promise<T>, |
90d4bb81 C |
31 | arg1: A |
32 | ): Promise<T> | |
33 | ||
2baea0c7 C |
34 | function retryTransactionWrapper <T> ( |
35 | functionToRetry: () => Promise<T> | Bluebird<T> | |
36 | ): Promise<T> | |
37 | ||
0f91ae62 | 38 | function retryTransactionWrapper <T> ( |
883a9019 | 39 | functionToRetry: (...args: any[]) => Promise<T>, |
90d4bb81 | 40 | ...args: any[] |
0f91ae62 | 41 | ): Promise<T> { |
0f91ae62 | 42 | return transactionRetryer<T>(callback => { |
2baea0c7 | 43 | functionToRetry.apply(null, args) |
0f91ae62 | 44 | .then((result: T) => callback(null, result)) |
6fcd19ba | 45 | .catch(err => callback(err)) |
075f16ca | 46 | }) |
6fcd19ba | 47 | .catch(err => { |
e0f49a19 | 48 | logger.warn(`Cannot execute ${functionToRetry.name} with many retries.`, { err }) |
20494f12 | 49 | throw err |
6fcd19ba | 50 | }) |
4df023f2 C |
51 | } |
52 | ||
0f91ae62 C |
53 | function transactionRetryer <T> (func: (err: any, data: T) => any) { |
54 | return new Promise<T>((res, rej) => { | |
90d4bb81 C |
55 | retry( |
56 | { | |
57 | times: 5, | |
58 | ||
59 | errorFilter: err => { | |
60 | const willRetry = (err.name === 'SequelizeDatabaseError') | |
28dfb44b | 61 | logger.debug('Maybe retrying the transaction function.', { willRetry, err, tags: [ 'sql', 'retry' ] }) |
90d4bb81 C |
62 | return willRetry |
63 | } | |
64 | }, | |
65 | func, | |
66 | (err, data) => err ? rej(err) : res(data) | |
67 | ) | |
4df023f2 C |
68 | }) |
69 | } | |
70 | ||
28dfb44b C |
71 | // --------------------------------------------------------------------------- |
72 | ||
45657746 | 73 | function resetSequelizeInstance <T> (instance: Model<T>) { |
823c34c0 | 74 | return instance.reload() |
06215f15 C |
75 | } |
76 | ||
764b1a14 | 77 | function filterNonExistingModels <T extends { hasSameUniqueKeysThan (other: T): boolean }> ( |
d7a25329 | 78 | fromDatabase: T[], |
764b1a14 | 79 | newModels: T[] |
d7a25329 C |
80 | ) { |
81 | return fromDatabase.filter(f => !newModels.find(newModel => newModel.hasSameUniqueKeysThan(f))) | |
764b1a14 C |
82 | } |
83 | ||
84 | function deleteAllModels <T extends Pick<Model, 'destroy'>> (models: T[], transaction: Transaction) { | |
85 | return Promise.all(models.map(f => f.destroy({ transaction }))) | |
d7a25329 C |
86 | } |
87 | ||
4df023f2 C |
88 | // --------------------------------------------------------------------------- |
89 | ||
28dfb44b | 90 | function runInReadCommittedTransaction <T> (fn: (t: Transaction) => Promise<T>) { |
a6a12dae C |
91 | const options = { isolationLevel: Transaction.ISOLATION_LEVELS.READ_COMMITTED } |
92 | ||
93 | return sequelizeTypescript.transaction(options, t => fn(t)) | |
28dfb44b C |
94 | } |
95 | ||
96 | function afterCommitIfTransaction (t: Transaction, fn: Function) { | |
97 | if (t) return t.afterCommit(() => fn()) | |
98 | ||
99 | return fn() | |
100 | } | |
101 | ||
102 | // --------------------------------------------------------------------------- | |
103 | ||
65fcc311 | 104 | export { |
06215f15 | 105 | resetSequelizeInstance, |
65fcc311 | 106 | retryTransactionWrapper, |
a5625b41 | 107 | transactionRetryer, |
d7a25329 | 108 | afterCommitIfTransaction, |
764b1a14 C |
109 | filterNonExistingModels, |
110 | deleteAllModels, | |
fa47956e | 111 | runInReadCommittedTransaction |
65fcc311 | 112 | } |