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 { RunnerJobState, 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     nextState: RunnerJobState
  33   }) {
  34     if (options.nextState !== RunnerJobState.ERRORED) return
  35
  36     const video = await loadTranscodingRunnerVideo(options.runnerJob, this.lTags)
  37     if (!video) return
  38
  39     await moveToFailedTranscodingState(video)
  40
  41     await VideoJobInfoModel.decrease(video.uuid, 'pendingTranscode')
  42   }
  43
  44   protected async specificCancel (options: {
  45     runnerJob: MRunnerJob
  46   }) {
  47     const { runnerJob } = options
  48
  49     const video = await loadTranscodingRunnerVideo(options.runnerJob, this.lTags)
  50     if (!video) return
  51
  52     const pending = await VideoJobInfoModel.decrease(video.uuid, 'pendingTranscode')
  53
  54     logger.debug(`Pending transcode decreased to ${pending} after cancel`, this.lTags(video.uuid))
  55
  56     if (pending === 0) {
  57       logger.info(
  58         `All transcoding jobs of ${video.uuid} have been processed or canceled, moving it to its next state`,
  59         this.lTags(video.uuid)
  60       )
  61
  62       const privatePayload = runnerJob.privatePayload as RunnerJobVODPrivatePayload
  63       await retryTransactionWrapper(moveToNextState, { video, isNewVideo: privatePayload.isNewVideo })
  64     }
  65   }
  66 }