server/lib/schedulers/videos-redundancy-scheduler.ts

   1 import { AbstractScheduler } from './abstract-scheduler'
   2 import { CONFIG, JOB_TTL, REDUNDANCY } from '../../initializers'
   3 import { logger } from '../../helpers/logger'
   4 import { VideosRedundancy } from '../../../shared/models/redundancy'
   5 import { VideoRedundancyModel } from '../../models/redundancy/video-redundancy'
   6 import { VideoFileModel } from '../../models/video/video-file'
   7 import { downloadWebTorrentVideo } from '../../helpers/webtorrent'
   8 import { join } from 'path'
   9 import { rename } from 'fs-extra'
  10 import { getServerActor } from '../../helpers/utils'
  11 import { sendCreateCacheFile, sendUpdateCacheFile } from '../activitypub/send'
  12 import { VideoModel } from '../../models/video/video'
  13 import { getVideoCacheFileActivityPubUrl } from '../activitypub/url'
  14 import { removeVideoRedundancy } from '../redundancy'
  15 import { getOrCreateVideoAndAccountAndChannel } from '../activitypub'
  16
  17 export class VideosRedundancyScheduler extends AbstractScheduler {
  18
  19   private static instance: AbstractScheduler
  20   private executing = false
  21
  22   protected schedulerIntervalMs = CONFIG.REDUNDANCY.VIDEOS.CHECK_INTERVAL
  23
  24   private constructor () {
  25     super()
  26   }
  27
  28   async execute () {
  29     if (this.executing) return
  30
  31     this.executing = true
  32
  33     for (const obj of CONFIG.REDUNDANCY.VIDEOS.STRATEGIES) {
  34       logger.info('Running redundancy scheduler for strategy %s.', obj.strategy)
  35
  36       try {
  37         const videoToDuplicate = await this.findVideoToDuplicate(obj)
  38         if (!videoToDuplicate) continue
  39
  40         const videoFiles = videoToDuplicate.VideoFiles
  41         videoFiles.forEach(f => f.Video = videoToDuplicate)
  42
  43         await this.purgeCacheIfNeeded(obj, videoFiles)
  44
  45         if (await this.isTooHeavy(obj, videoFiles)) {
  46           logger.info('Video %s is too big for our cache, skipping.', videoToDuplicate.url)
  47           continue
  48         }
  49
  50         logger.info('Will duplicate video %s in redundancy scheduler "%s".', videoToDuplicate.url, obj.strategy)
  51
  52         await this.createVideoRedundancy(obj, videoFiles)
  53       } catch (err) {
  54         logger.error('Cannot run videos redundancy %s.', obj.strategy, { err })
  55       }
  56     }
  57
  58     await this.extendsLocalExpiration()
  59
  60     await this.purgeRemoteExpired()
  61
  62     this.executing = false
  63   }
  64
  65   static get Instance () {
  66     return this.instance || (this.instance = new this())
  67   }
  68
  69   private async extendsLocalExpiration () {
  70     const expired = await VideoRedundancyModel.listLocalExpired()
  71
  72     for (const redundancyModel of expired) {
  73       try {
  74         const redundancy = CONFIG.REDUNDANCY.VIDEOS.STRATEGIES.find(s => s.strategy === redundancyModel.strategy)
  75         await this.extendsExpirationOf(redundancyModel, redundancy.minLifetime)
  76       } catch (err) {
  77         logger.error('Cannot extend expiration of %s video from our redundancy system.', this.buildEntryLogId(redundancyModel))
  78       }
  79     }
  80   }
  81
  82   private async purgeRemoteExpired () {
  83     const expired = await VideoRedundancyModel.listRemoteExpired()
  84
  85     for (const redundancyModel of expired) {
  86       try {
  87         await removeVideoRedundancy(redundancyModel)
  88       } catch (err) {
  89         logger.error('Cannot remove redundancy %s from our redundancy system.', this.buildEntryLogId(redundancyModel))
  90       }
  91     }
  92   }
  93
  94   private findVideoToDuplicate (cache: VideosRedundancy) {
  95     if (cache.strategy === 'most-views') {
  96       return VideoRedundancyModel.findMostViewToDuplicate(REDUNDANCY.VIDEOS.RANDOMIZED_FACTOR)
  97     }
  98
  99     if (cache.strategy === 'trending') {
 100       return VideoRedundancyModel.findTrendingToDuplicate(REDUNDANCY.VIDEOS.RANDOMIZED_FACTOR)
 101     }
 102
 103     if (cache.strategy === 'recently-added') {
 104       const minViews = cache.minViews
 105       return VideoRedundancyModel.findRecentlyAddedToDuplicate(REDUNDANCY.VIDEOS.RANDOMIZED_FACTOR, minViews)
 106     }
 107   }
 108
 109   private async createVideoRedundancy (redundancy: VideosRedundancy, filesToDuplicate: VideoFileModel[]) {
 110     const serverActor = await getServerActor()
 111
 112     for (const file of filesToDuplicate) {
 113       // We need more attributes and check if the video still exists
 114       const getVideoOptions = {
 115         videoObject: file.Video.url,
 116         syncParam: { likes: false, dislikes: false, shares: false, comments: false, thumbnail: false, refreshVideo: true },
 117         fetchType: 'only-video' as 'only-video'
 118       }
 119       const { video } = await getOrCreateVideoAndAccountAndChannel(getVideoOptions)
 120
 121       const existing = await VideoRedundancyModel.loadLocalByFileId(file.id)
 122       if (existing) {
 123         if (video) {
 124           await this.extendsExpirationOf(existing, redundancy.minLifetime)
 125         } else {
 126           logger.info('Destroying existing redundancy %s, because the associated video does not exist anymore.', existing.url)
 127
 128           await existing.destroy()
 129         }
 130
 131         continue
 132       }
 133
 134       if (!video) {
 135         logger.info('Video %s we want to duplicate does not existing anymore, skipping.', file.Video.url)
 136
 137         continue
 138       }
 139
 140       logger.info('Duplicating %s - %d in videos redundancy with "%s" strategy.', video.url, file.resolution, redundancy.strategy)
 141
 142       const { baseUrlHttp, baseUrlWs } = video.getBaseUrls()
 143       const magnetUri = video.generateMagnetUri(file, baseUrlHttp, baseUrlWs)
 144
 145       const tmpPath = await downloadWebTorrentVideo({ magnetUri }, JOB_TTL['video-import'])
 146
 147       const destPath = join(CONFIG.STORAGE.VIDEOS_DIR, video.getVideoFilename(file))
 148       await rename(tmpPath, destPath)
 149
 150       const createdModel = await VideoRedundancyModel.create({
 151         expiresOn: this.buildNewExpiration(redundancy.minLifetime),
 152         url: getVideoCacheFileActivityPubUrl(file),
 153         fileUrl: video.getVideoFileUrl(file, CONFIG.WEBSERVER.URL),
 154         strategy: redundancy.strategy,
 155         videoFileId: file.id,
 156         actorId: serverActor.id
 157       })
 158       createdModel.VideoFile = file
 159
 160       await sendCreateCacheFile(serverActor, createdModel)
 161     }
 162   }
 163
 164   private async extendsExpirationOf (redundancy: VideoRedundancyModel, expiresAfterMs: number) {
 165     logger.info('Extending expiration of %s.', redundancy.url)
 166
 167     const serverActor = await getServerActor()
 168
 169     redundancy.expiresOn = this.buildNewExpiration(expiresAfterMs)
 170     await redundancy.save()
 171
 172     await sendUpdateCacheFile(serverActor, redundancy)
 173   }
 174
 175   private async purgeCacheIfNeeded (redundancy: VideosRedundancy, filesToDuplicate: VideoFileModel[]) {
 176     while (this.isTooHeavy(redundancy, filesToDuplicate)) {
 177       const toDelete = await VideoRedundancyModel.loadOldestLocalThatAlreadyExpired(redundancy.strategy, redundancy.minLifetime)
 178       if (!toDelete) return
 179
 180       await removeVideoRedundancy(toDelete)
 181     }
 182   }
 183
 184   private async isTooHeavy (redundancy: VideosRedundancy, filesToDuplicate: VideoFileModel[]) {
 185     const maxSize = redundancy.size - this.getTotalFileSizes(filesToDuplicate)
 186
 187     const totalDuplicated = await VideoRedundancyModel.getTotalDuplicated(redundancy.strategy)
 188
 189     return totalDuplicated > maxSize
 190   }
 191
 192   private buildNewExpiration (expiresAfterMs: number) {
 193     return new Date(Date.now() + expiresAfterMs)
 194   }
 195
 196   private buildEntryLogId (object: VideoRedundancyModel) {
 197     return `${object.VideoFile.Video.url}-${object.VideoFile.resolution}`
 198   }
 199
 200   private getTotalFileSizes (files: VideoFileModel[]) {
 201     const fileReducer = (previous: number, current: VideoFileModel) => previous + current.size
 202
 203     return files.reduce(fileReducer, 0)
 204   }
 205 }