]>
Commit | Line | Data |
---|---|---|
4d4e5cd4 | 1 | import * as retry from 'async/retry' |
20494f12 | 2 | import * as Bluebird from 'bluebird' |
e024fd6a | 3 | import { QueryTypes, 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 C |
8 | function retryTransactionWrapper <T, A, B, C, D> ( |
9 | functionToRetry: (arg1: A, arg2: B, arg3: C, arg4: D) => Promise<T> | Bluebird<T>, | |
10 | arg1: A, | |
11 | arg2: B, | |
12 | arg3: C, | |
13 | arg4: D, | |
14 | ): Promise<T> | |
15 | ||
90d4bb81 C |
16 | function retryTransactionWrapper <T, A, B, C> ( |
17 | functionToRetry: (arg1: A, arg2: B, arg3: C) => Promise<T> | Bluebird<T>, | |
18 | arg1: A, | |
19 | arg2: B, | |
20 | arg3: C | |
21 | ): Promise<T> | |
22 | ||
23 | function retryTransactionWrapper <T, A, B> ( | |
24 | functionToRetry: (arg1: A, arg2: B) => Promise<T> | Bluebird<T>, | |
25 | arg1: A, | |
26 | arg2: B | |
27 | ): Promise<T> | |
28 | ||
29 | function retryTransactionWrapper <T, A> ( | |
30 | functionToRetry: (arg1: A) => Promise<T> | Bluebird<T>, | |
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> ( |
90d4bb81 C |
39 | functionToRetry: (...args: any[]) => Promise<T> | Bluebird<T>, |
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 => { |
1e11f67b | 48 | logger.error(`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') | |
61 | logger.debug('Maybe retrying the transaction function.', { willRetry, err }) | |
62 | return willRetry | |
63 | } | |
64 | }, | |
65 | func, | |
66 | (err, data) => err ? rej(err) : res(data) | |
67 | ) | |
4df023f2 C |
68 | }) |
69 | } | |
70 | ||
c39ea24b | 71 | function updateInstanceWithAnother <T extends Model<T>> (instanceToUpdate: Model<T>, baseInstance: Model<T>) { |
a5625b41 C |
72 | const obj = baseInstance.toJSON() |
73 | ||
74 | for (const key of Object.keys(obj)) { | |
1735c825 | 75 | instanceToUpdate[key] = obj[key] |
a5625b41 C |
76 | } |
77 | } | |
78 | ||
06215f15 C |
79 | function resetSequelizeInstance (instance: Model<any>, savedFields: object) { |
80 | Object.keys(savedFields).forEach(key => { | |
1735c825 | 81 | instance[key] = savedFields[key] |
06215f15 C |
82 | }) |
83 | } | |
84 | ||
2284f202 C |
85 | function afterCommitIfTransaction (t: Transaction, fn: Function) { |
86 | if (t) return t.afterCommit(() => fn()) | |
87 | ||
88 | return fn() | |
89 | } | |
90 | ||
d7a25329 C |
91 | function deleteNonExistingModels <T extends { hasSameUniqueKeysThan (other: T): boolean } & Model<T>> ( |
92 | fromDatabase: T[], | |
93 | newModels: T[], | |
94 | t: Transaction | |
95 | ) { | |
96 | return fromDatabase.filter(f => !newModels.find(newModel => newModel.hasSameUniqueKeysThan(f))) | |
97 | .map(f => f.destroy({ transaction: t })) | |
98 | } | |
99 | ||
e024fd6a C |
100 | // Sequelize always skip the update if we only update updatedAt field |
101 | function setAsUpdated (table: string, id: number, transaction?: Transaction) { | |
102 | return sequelizeTypescript.query( | |
103 | `UPDATE "${table}" SET "updatedAt" = :updatedAt WHERE id = :id`, | |
104 | { | |
105 | replacements: { table, id, updatedAt: new Date() }, | |
106 | type: QueryTypes.UPDATE, | |
107 | transaction | |
108 | } | |
109 | ) | |
110 | } | |
111 | ||
4df023f2 C |
112 | // --------------------------------------------------------------------------- |
113 | ||
65fcc311 | 114 | export { |
06215f15 | 115 | resetSequelizeInstance, |
65fcc311 | 116 | retryTransactionWrapper, |
a5625b41 | 117 | transactionRetryer, |
2284f202 | 118 | updateInstanceWithAnother, |
d7a25329 | 119 | afterCommitIfTransaction, |
e024fd6a C |
120 | deleteNonExistingModels, |
121 | setAsUpdated | |
65fcc311 | 122 | } |