server/lib/job-queue/handlers/video-views.ts

   1 import { Redis } from '../../redis'
   2 import { logger } from '../../../helpers/logger'
   3 import { VideoModel } from '../../../models/video/video'
   4 import { VideoViewModel } from '../../../models/video/video-view'
   5 import { isTestInstance } from '../../../helpers/core-utils'
   6 import { federateVideoIfNeeded } from '../../activitypub/videos'
   7
   8 async function processVideosViews () {
   9   const lastHour = new Date()
  10
  11   // In test mode, we run this function multiple times per hour, so we don't want the values of the previous hour
  12   if (!isTestInstance()) lastHour.setHours(lastHour.getHours() - 1)
  13
  14   const hour = lastHour.getHours()
  15   const startDate = lastHour.setMinutes(0, 0, 0)
  16   const endDate = lastHour.setMinutes(59, 59, 999)
  17
  18   const videoIds = await Redis.Instance.getVideosIdViewed(hour)
  19   if (videoIds.length === 0) return
  20
  21   logger.info('Processing videos views in job for hour %d.', hour)
  22
  23   for (const videoId of videoIds) {
  24     try {
  25       const views = await Redis.Instance.getVideoViews(videoId, hour)
  26       await Redis.Instance.deleteVideoViews(videoId, hour)
  27
  28       if (views) {
  29         logger.debug('Adding %d views to video %d in hour %d.', views, videoId, hour)
  30
  31         try {
  32           const video = await VideoModel.loadAndPopulateAccountAndServerAndTags(videoId)
  33           if (!video) {
  34             logger.debug('Video %d does not exist anymore, skipping videos view addition.', videoId)
  35             continue
  36           }
  37
  38           await VideoViewModel.create({
  39             startDate: new Date(startDate),
  40             endDate: new Date(endDate),
  41             views,
  42             videoId
  43           })
  44
  45           if (video.isOwned()) {
  46             // If this is a remote video, the origin instance will send us an update
  47             await VideoModel.incrementViews(videoId, views)
  48
  49             // Send video update
  50             video.views += views
  51             await federateVideoIfNeeded(video, false)
  52           }
  53         } catch (err) {
  54           logger.error('Cannot create video views for video %d in hour %d.', videoId, hour, { err })
  55         }
  56       }
  57     } catch (err) {
  58       logger.error('Cannot update video views of video %d in hour %d.', videoId, hour, { err })
  59     }
  60   }
  61 }
  62
  63 // ---------------------------------------------------------------------------
  64
  65 export {
  66   processVideosViews
  67 }