Fix avatar transaction retry

[github/Chocobozzz/PeerTube.git] / server / lib / avatar.ts

diff --git a/server/lib/avatar.ts b/server/lib/avatar.ts
index 14f0a05f58829f5d93a175c2e616392ce9cde1ec..4b6bc318579ffeb4cd29ad01c9b1c83591b3779f 100644 (file)
--- a/server/lib/avatar.ts
+++ b/server/lib/avatar.ts
@@ -6,6 +6,7 @@ import { processImage } from '../helpers/image-utils'
 import { AccountModel } from '../models/account/account'
 import { VideoChannelModel } from '../models/video/video-channel'
 import { extname, join } from 'path'
+import { retryTransactionWrapper } from '../helpers/database-utils'
 
 async function updateActorAvatarFile (avatarPhysicalFile: Express.Multer.File, accountOrChannel: AccountModel | VideoChannelModel) {
   const extension = extname(avatarPhysicalFile.filename)
@@ -13,13 +14,15 @@ async function updateActorAvatarFile (avatarPhysicalFile: Express.Multer.File, a
   const destination = join(CONFIG.STORAGE.AVATARS_DIR, avatarName)
   await processImage(avatarPhysicalFile, destination, AVATARS_SIZE)
 
-  return sequelizeTypescript.transaction(async t => {
-    const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarName, t)
-    await updatedActor.save({ transaction: t })
+  return retryTransactionWrapper(() => {
+    return sequelizeTypescript.transaction(async t => {
+      const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarName, t)
+      await updatedActor.save({ transaction: t })
 
-    await sendUpdateActor(accountOrChannel, t)
+      await sendUpdateActor(accountOrChannel, t)
 
-    return updatedActor.Avatar
+      return updatedActor.Avatar
+    })
   })
 }