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

   1 import * as Bluebird from 'bluebird'
   2 import { computeResolutionsToTranscode, logger } from '../../../helpers'
   3
   4 import { database as db } from '../../../initializers/database'
   5 import { VideoInstance } from '../../../models'
   6 import { sendAddVideo } from '../../activitypub/send-request'
   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 db.Video.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: VideoInstance, jobScheduler: JobScheduler<TranscodingJobPayload, VideoInstance>) {
  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 db.Video.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
  41   const originalFileHeight = await videoDatabase.getOriginalFileHeight()
  42   // Create transcoding jobs if there are enabled resolutions
  43
  44   const resolutionsEnabled = computeResolutionsToTranscode(originalFileHeight)
  45   logger.info(
  46     'Resolutions computed for video %s and origin file height of %d.', videoDatabase.uuid, originalFileHeight,
  47     { resolutions: resolutionsEnabled }
  48   )
  49
  50   if (resolutionsEnabled.length !== 0) {
  51     try {
  52       await db.sequelize.transaction(async t => {
  53         const tasks: Bluebird<any>[] = []
  54
  55         for (const resolution of resolutionsEnabled) {
  56           const dataInput = {
  57             videoUUID: videoDatabase.uuid,
  58             resolution
  59           }
  60
  61           const p = jobScheduler.createJob(t, 'videoFileTranscoder', dataInput)
  62           tasks.push(p)
  63         }
  64
  65         await Promise.all(tasks)
  66       })
  67
  68       logger.info('Transcoding jobs created for uuid %s.', videoDatabase.uuid, { resolutionsEnabled })
  69     } catch (err) {
  70       logger.warn('Cannot transcode the video.', err)
  71     }
  72   } else {
  73     logger.info('No transcoding jobs created for video %s (no resolutions enabled).')
  74     return undefined
  75   }
  76 }
  77
  78 // ---------------------------------------------------------------------------
  79
  80 export {
  81   process,
  82   onError,
  83   onSuccess
  84 }