]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/commitdiff
Use saveInTransactionWithRetries helper
authorChocobozzz <me@florianbigard.com>
Fri, 19 May 2023 13:20:12 +0000 (15:20 +0200)
committerChocobozzz <me@florianbigard.com>
Fri, 19 May 2023 13:20:12 +0000 (15:20 +0200)
server/lib/runners/job-handlers/abstract-job-handler.ts

index ca97d08816f3d4c1b2bbcd7814c2a0532ca38c29..329977de19555b2b2689c2b47c4ebe8e810ed0b4 100644 (file)
@@ -1,5 +1,5 @@
 import { throttle } from 'lodash'
-import { retryTransactionWrapper, saveInTransactionWithRetries } from '@server/helpers/database-utils'
+import { saveInTransactionWithRetries } from '@server/helpers/database-utils'
 import { logger, loggerTagsFactory } from '@server/helpers/logger'
 import { RUNNER_JOBS } from '@server/initializers/constants'
 import { sequelizeTypescript } from '@server/initializers/database'
@@ -124,11 +124,7 @@ export abstract class AbstractJobHandler <C, U extends RunnerJobUpdatePayload, S
       return
     }
 
-    await retryTransactionWrapper(() => {
-      return sequelizeTypescript.transaction(async transaction => {
-        return runnerJob.save({ transaction })
-      })
-    })
+    await saveInTransactionWithRetries(runnerJob)
   }
 
   // ---------------------------------------------------------------------------
@@ -184,11 +180,7 @@ export abstract class AbstractJobHandler <C, U extends RunnerJobUpdatePayload, S
 
     runnerJob.setToErrorOrCancel(cancelState)
 
-    await retryTransactionWrapper(() => {
-      return sequelizeTypescript.transaction(async transaction => {
-        await runnerJob.save({ transaction })
-      })
-    })
+    await saveInTransactionWithRetries(runnerJob)
 
     const children = await RunnerJobModel.listChildrenOf(runnerJob)
     for (const child of children) {
@@ -219,11 +211,7 @@ export abstract class AbstractJobHandler <C, U extends RunnerJobUpdatePayload, S
 
     runnerJob.resetToPending()
 
-    await retryTransactionWrapper(() => {
-      return sequelizeTypescript.transaction(async transaction => {
-        await runnerJob.save({ transaction })
-      })
-    })
+    await saveInTransactionWithRetries(runnerJob)
   }
 
   protected setAbortState (runnerJob: MRunnerJob) {
@@ -260,11 +248,7 @@ export abstract class AbstractJobHandler <C, U extends RunnerJobUpdatePayload, S
       runnerJob.resetToPending()
     }
 
-    await retryTransactionWrapper(() => {
-      return sequelizeTypescript.transaction(async transaction => {
-        await runnerJob.save({ transaction })
-      })
-    })
+    await saveInTransactionWithRetries(runnerJob)
 
     if (runnerJob.state === errorState) {
       const children = await RunnerJobModel.listChildrenOf(runnerJob)