server/lib/request-video-event-scheduler.js

   1 'use strict'
   2
   3 const BaseRequestScheduler = require('./base-request-scheduler')
   4 const constants = require('../initializers/constants')
   5 const db = require('../initializers/database')
   6
   7 module.exports = class RequestVideoEventScheduler extends BaseRequestScheduler {
   8
   9   constructor () {
  10     super()
  11
  12     // We limit the size of the requests
  13     this.limitPods = constants.REQUESTS_VIDEO_EVENT_LIMIT_PODS
  14     this.limitPerPod = constants.REQUESTS_VIDEO_EVENT_LIMIT_PER_POD
  15
  16     this.description = 'video event requests'
  17   }
  18
  19   getRequestModel () {
  20     return db.RequestVideoEvent
  21   }
  22
  23   getRequestToPodModel () {
  24     return db.RequestVideoEvent
  25   }
  26
  27   buildRequestObjects (eventsToProcess) {
  28     const requestsToMakeGrouped = {}
  29
  30     /* Example:
  31         {
  32           pod1: {
  33             video1: { views: 4, likes: 5 },
  34             video2: { likes: 5 }
  35           }
  36         }
  37     */
  38     const eventsPerVideoPerPod = {}
  39
  40     // We group video events per video and per pod
  41     // We add the counts of the same event types
  42     Object.keys(eventsToProcess).forEach(toPodId => {
  43       eventsToProcess[toPodId].forEach(eventToProcess => {
  44         if (!eventsPerVideoPerPod[toPodId]) eventsPerVideoPerPod[toPodId] = {}
  45
  46         if (!requestsToMakeGrouped[toPodId]) {
  47           requestsToMakeGrouped[toPodId] = {
  48             toPod: eventToProcess.pod,
  49             endpoint: constants.REQUEST_ENDPOINTS.EVENT,
  50             ids: [], // request ids, to delete them from the DB in the future
  51             datas: [] // requests data
  52           }
  53         }
  54         requestsToMakeGrouped[toPodId].ids.push(eventToProcess.id)
  55
  56         const eventsPerVideo = eventsPerVideoPerPod[toPodId]
  57         const remoteId = eventToProcess.video.remoteId
  58         if (!eventsPerVideo[remoteId]) eventsPerVideo[remoteId] = {}
  59
  60         const events = eventsPerVideo[remoteId]
  61         if (!events[eventToProcess.type]) events[eventToProcess.type] = 0
  62
  63         events[eventToProcess.type] += eventToProcess.count
  64       })
  65     })
  66
  67     // Now we build our requests array per pod
  68     Object.keys(eventsPerVideoPerPod).forEach(toPodId => {
  69       const eventsForPod = eventsPerVideoPerPod[toPodId]
  70
  71       Object.keys(eventsForPod).forEach(remoteId => {
  72         const eventsForVideo = eventsForPod[remoteId]
  73
  74         Object.keys(eventsForVideo).forEach(eventType => {
  75           requestsToMakeGrouped[toPodId].datas.push({
  76             data: {
  77               remoteId,
  78               eventType,
  79               count: eventsForVideo[eventType]
  80             }
  81           })
  82         })
  83       })
  84     })
  85
  86     return requestsToMakeGrouped
  87   }
  88
  89   // { type, videoId, count?, transaction? }
  90   createRequest (options, callback) {
  91     const type = options.type
  92     const videoId = options.videoId
  93     const transaction = options.transaction
  94     let count = options.count
  95
  96     if (count === undefined) count = 1
  97
  98     const dbRequestOptions = {}
  99     if (transaction) dbRequestOptions.transaction = transaction
 100
 101     const createQuery = {
 102       type,
 103       count,
 104       videoId
 105     }
 106
 107     return db.RequestVideoEvent.create(createQuery, dbRequestOptions).asCallback(callback)
 108   }
 109 }