server/lib/runners/job-handlers/abstract-vod-transcoding-job-handler.ts

   1
   2 import { retryTransactionWrapper } from '@server/helpers/database-utils'
   3 import { logger } from '@server/helpers/logger'
   4 import { moveToFailedTranscodingState, moveToNextState } from '@server/lib/video-state'
   5 import { VideoJobInfoModel } from '@server/models/video/video-job-info'
   6 import { MRunnerJob } from '@server/types/models/runners'
   7 import { RunnerJobSuccessPayload, RunnerJobUpdatePayload, RunnerJobVODPrivatePayload } from '@shared/models'
   8 import { AbstractJobHandler } from './abstract-job-handler'
   9 import { loadTranscodingRunnerVideo } from './shared'
  10
  11 // eslint-disable-next-line max-len
  12 export abstract class AbstractVODTranscodingJobHandler <C, U extends RunnerJobUpdatePayload, S extends RunnerJobSuccessPayload> extends AbstractJobHandler<C, U, S> {
  13
  14   protected isAbortSupported () {
  15     return true
  16   }
  17
  18   protected specificUpdate (_options: {
  19     runnerJob: MRunnerJob
  20   }) {
  21     // empty
  22   }
  23
  24   protected specificAbort (_options: {
  25     runnerJob: MRunnerJob
  26   }) {
  27     // empty
  28   }
  29
  30   protected async specificError (options: {
  31     runnerJob: MRunnerJob
  32   }) {
  33     const video = await loadTranscodingRunnerVideo(options.runnerJob, this.lTags)
  34     if (!video) return
  35
  36     await moveToFailedTranscodingState(video)
  37
  38     await VideoJobInfoModel.decrease(video.uuid, 'pendingTranscode')
  39   }
  40
  41   protected async specificCancel (options: {
  42     runnerJob: MRunnerJob
  43   }) {
  44     const { runnerJob } = options
  45
  46     const video = await loadTranscodingRunnerVideo(options.runnerJob, this.lTags)
  47     if (!video) return
  48
  49     const pending = await VideoJobInfoModel.decrease(video.uuid, 'pendingTranscode')
  50
  51     logger.debug(`Pending transcode decreased to ${pending} after cancel`, this.lTags(video.uuid))
  52
  53     if (pending === 0) {
  54       logger.info(
  55         `All transcoding jobs of ${video.uuid} have been processed or canceled, moving it to its next state`,
  56         this.lTags(video.uuid)
  57       )
  58
  59       const privatePayload = runnerJob.privatePayload as RunnerJobVODPrivatePayload
  60       await retryTransactionWrapper(moveToNextState, { video, isNewVideo: privatePayload.isNewVideo })
  61     }
  62   }
  63 }