]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - server/lib/schedulers/update-videos-scheduler.ts
Optimize videos list SQL queries workflow
[github/Chocobozzz/PeerTube.git] / server / lib / schedulers / update-videos-scheduler.ts
1 import { logger } from '../../helpers/logger'
2 import { AbstractScheduler } from './abstract-scheduler'
3 import { ScheduleVideoUpdateModel } from '../../models/video/schedule-video-update'
4 import { retryTransactionWrapper } from '../../helpers/database-utils'
5 import { federateVideoIfNeeded } from '../activitypub'
6 import { SCHEDULER_INTERVALS_MS } from '../../initializers/constants'
7 import { VideoPrivacy } from '../../../shared/models/videos'
8 import { Notifier } from '../notifier'
9 import { sequelizeTypescript } from '../../initializers/database'
10 import { MVideoFullLight } from '@server/typings/models'
11
12 export class UpdateVideosScheduler extends AbstractScheduler {
13
14 private static instance: AbstractScheduler
15
16 protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.updateVideos
17
18 private constructor () {
19 super()
20 }
21
22 protected async internalExecute () {
23 return retryTransactionWrapper(this.updateVideos.bind(this))
24 }
25
26 private async updateVideos () {
27 if (!await ScheduleVideoUpdateModel.areVideosToUpdate()) return undefined
28
29 const publishedVideos = await sequelizeTypescript.transaction(async t => {
30 const schedules = await ScheduleVideoUpdateModel.listVideosToUpdate(t)
31 const publishedVideos: MVideoFullLight[] = []
32
33 for (const schedule of schedules) {
34 const video = schedule.Video
35 logger.info('Executing scheduled video update on %s.', video.uuid)
36
37 if (schedule.privacy) {
38 const wasConfidentialVideo = video.isConfidential()
39 const isNewVideo = video.isNewVideo(schedule.privacy)
40
41 video.setPrivacy(schedule.privacy)
42 await video.save({ transaction: t })
43 await federateVideoIfNeeded(video, isNewVideo, t)
44
45 if (wasConfidentialVideo) {
46 const videoToPublish: MVideoFullLight = Object.assign(video, { ScheduleVideoUpdate: schedule, UserVideoHistories: [] })
47 publishedVideos.push(videoToPublish)
48 }
49 }
50
51 await schedule.destroy({ transaction: t })
52 }
53
54 return publishedVideos
55 })
56
57 for (const v of publishedVideos) {
58 Notifier.Instance.notifyOnNewVideoIfNeeded(v)
59 Notifier.Instance.notifyOnVideoPublishedAfterScheduledUpdate(v)
60 }
61 }
62
63 static get Instance () {
64 return this.instance || (this.instance = new this())
65 }
66 }