diff options
author | Chocobozzz <florian.bigard@gmail.com> | 2017-11-09 17:51:58 +0100 |
---|---|---|
committer | Chocobozzz <florian.bigard@gmail.com> | 2017-11-27 19:40:51 +0100 |
commit | e4f97babf701481b55cc10fb3448feab5f97c867 (patch) | |
tree | af37402a594dc5ff09f71ecb0687e8cfe4cdb471 /server/lib/jobs/transcoding-job-scheduler/video-file-optimizer-handler.ts | |
parent | 343ad675f2a26c15b86150a9a3552e619d5d44f4 (diff) | |
download | PeerTube-e4f97babf701481b55cc10fb3448feab5f97c867.tar.gz PeerTube-e4f97babf701481b55cc10fb3448feab5f97c867.tar.zst PeerTube-e4f97babf701481b55cc10fb3448feab5f97c867.zip |
Begin activitypub
Diffstat (limited to 'server/lib/jobs/transcoding-job-scheduler/video-file-optimizer-handler.ts')
-rw-r--r-- | server/lib/jobs/transcoding-job-scheduler/video-file-optimizer-handler.ts | 85 |
1 files changed, 85 insertions, 0 deletions
diff --git a/server/lib/jobs/transcoding-job-scheduler/video-file-optimizer-handler.ts b/server/lib/jobs/transcoding-job-scheduler/video-file-optimizer-handler.ts new file mode 100644 index 000000000..ccded4721 --- /dev/null +++ b/server/lib/jobs/transcoding-job-scheduler/video-file-optimizer-handler.ts | |||
@@ -0,0 +1,85 @@ | |||
1 | import * as Bluebird from 'bluebird' | ||
2 | |||
3 | import { database as db } from '../../../initializers/database' | ||
4 | import { logger, computeResolutionsToTranscode } from '../../../helpers' | ||
5 | import { VideoInstance } from '../../../models' | ||
6 | import { addVideoToFriends } from '../../friends' | ||
7 | import { JobScheduler } from '../job-scheduler' | ||
8 | |||
9 | async function process (data: { videoUUID: string }, jobId: number) { | ||
10 | const video = await db.Video.loadByUUIDAndPopulateAuthorAndPodAndTags(data.videoUUID) | ||
11 | // No video, maybe deleted? | ||
12 | if (!video) { | ||
13 | logger.info('Do not process job %d, video does not exist.', jobId, { videoUUID: video.uuid }) | ||
14 | return undefined | ||
15 | } | ||
16 | |||
17 | await video.optimizeOriginalVideofile() | ||
18 | |||
19 | return video | ||
20 | } | ||
21 | |||
22 | function onError (err: Error, jobId: number) { | ||
23 | logger.error('Error when optimized video file in job %d.', jobId, err) | ||
24 | return Promise.resolve() | ||
25 | } | ||
26 | |||
27 | async function onSuccess (jobId: number, video: VideoInstance) { | ||
28 | if (video === undefined) return undefined | ||
29 | |||
30 | logger.info('Job %d is a success.', jobId) | ||
31 | |||
32 | // Maybe the video changed in database, refresh it | ||
33 | const videoDatabase = await db.Video.loadByUUIDAndPopulateAuthorAndPodAndTags(video.uuid) | ||
34 | // Video does not exist anymore | ||
35 | if (!videoDatabase) return undefined | ||
36 | |||
37 | const remoteVideo = await videoDatabase.toAddRemoteJSON() | ||
38 | |||
39 | // Now we'll add the video's meta data to our friends | ||
40 | await addVideoToFriends(remoteVideo, null) | ||
41 | |||
42 | const originalFileHeight = await videoDatabase.getOriginalFileHeight() | ||
43 | // Create transcoding jobs if there are enabled resolutions | ||
44 | |||
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 db.sequelize.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.Instance.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 | } | ||