server/lib/jobs/transcoding-job-scheduler/video-file-optimizer-handler.ts

   1 import * as Bluebird from 'bluebird'
   2 import { computeResolutionsToTranscode, logger } from '../../../helpers'
   3 import { sequelizeTypescript } from '../../../initializers'
   4 import { VideoModel } from '../../../models/video/video'
   5 import { shareVideoByServer } from '../../activitypub'
   6 import { sendAddVideo } from '../../activitypub/send'
   7 import { JobScheduler } from '../job-scheduler'
   8 import { TranscodingJobPayload } from './transcoding-job-scheduler'
   9
  10 async function process (data: TranscodingJobPayload, jobId: number) {
  11   const video = await VideoModel.loadByUUIDAndPopulateAccountAndServerAndTags(data.videoUUID)
  12   // No video, maybe deleted?
  13   if (!video) {
  14     logger.info('Do not process job %d, video does not exist.', jobId, { videoUUID: video.uuid })
  15     return undefined
  16   }
  17
  18   await video.optimizeOriginalVideofile()
  19
  20   return video
  21 }
  22
  23 function onError (err: Error, jobId: number) {
  24   logger.error('Error when optimized video file in job %d.', jobId, err)
  25   return Promise.resolve()
  26 }
  27
  28 async function onSuccess (jobId: number, video: VideoModel, jobScheduler: JobScheduler<TranscodingJobPayload, VideoModel>) {
  29   if (video === undefined) return undefined
  30
  31   logger.info('Job %d is a success.', jobId)
  32
  33   // Maybe the video changed in database, refresh it
  34   const videoDatabase = await VideoModel.loadByUUIDAndPopulateAccountAndServerAndTags(video.uuid)
  35   // Video does not exist anymore
  36   if (!videoDatabase) return undefined
  37
  38   // Now we'll add the video's meta data to our followers
  39   await sendAddVideo(video, undefined)
  40   await shareVideoByServer(video, undefined)
  41
  42   const originalFileHeight = await videoDatabase.getOriginalFileHeight()
  43
  44   // Create transcoding jobs if there are enabled resolutions
  45   const resolutionsEnabled = computeResolutionsToTranscode(originalFileHeight)
  46   logger.info(
  47     'Resolutions computed for video %s and origin file height of %d.', videoDatabase.uuid, originalFileHeight,
  48     { resolutions: resolutionsEnabled }
  49   )
  50
  51   if (resolutionsEnabled.length !== 0) {
  52     try {
  53       await sequelizeTypescript.transaction(async t => {
  54         const tasks: Bluebird<any>[] = []
  55
  56         for (const resolution of resolutionsEnabled) {
  57           const dataInput = {
  58             videoUUID: videoDatabase.uuid,
  59             resolution
  60           }
  61
  62           const p = jobScheduler.createJob(t, 'videoFileTranscoder', dataInput)
  63           tasks.push(p)
  64         }
  65
  66         await Promise.all(tasks)
  67       })
  68
  69       logger.info('Transcoding jobs created for uuid %s.', videoDatabase.uuid, { resolutionsEnabled })
  70     } catch (err) {
  71       logger.warn('Cannot transcode the video.', err)
  72     }
  73   } else {
  74     logger.info('No transcoding jobs created for video %s (no resolutions enabled).')
  75     return undefined
  76   }
  77 }
  78
  79 // ---------------------------------------------------------------------------
  80
  81 export {
  82   process,
  83   onError,
  84   onSuccess
  85 }