server/models/request-video-qadu.ts

   1 /*
   2   Request Video for Quick And Dirty Updates like:
   3    - views
   4    - likes
   5    - dislikes
   6
   7   We can't put it in the same system than basic requests for efficiency.
   8   Moreover we don't want to slow down the basic requests with a lot of views/likes/dislikes requests.
   9   So we put it an independant request scheduler.
  10 */
  11
  12 import { values } from 'lodash'
  13 import * as Sequelize from 'sequelize'
  14
  15 import { REQUEST_VIDEO_QADU_TYPES } from '../initializers'
  16
  17 import { addMethodsToModel } from './utils'
  18 import {
  19   RequestVideoQaduClass,
  20   RequestVideoQaduInstance,
  21   RequestVideoQaduAttributes,
  22
  23   RequestVideoQaduMethods
  24 } from './request-video-qadu-interface'
  25
  26 let RequestVideoQadu: Sequelize.Model<RequestVideoQaduInstance, RequestVideoQaduAttributes>
  27 let countTotalRequests: RequestVideoQaduMethods.CountTotalRequests
  28 let listWithLimitAndRandom: RequestVideoQaduMethods.ListWithLimitAndRandom
  29 let removeByRequestIdsAndPod: RequestVideoQaduMethods.RemoveByRequestIdsAndPod
  30 let removeAll: RequestVideoQaduMethods.RemoveAll
  31
  32 export default function (sequelize, DataTypes) {
  33   RequestVideoQadu = sequelize.define('RequestVideoQadu',
  34     {
  35       type: {
  36         type: DataTypes.ENUM(values(REQUEST_VIDEO_QADU_TYPES)),
  37         allowNull: false
  38       }
  39     },
  40     {
  41       timestamps: false,
  42       indexes: [
  43         {
  44           fields: [ 'podId' ]
  45         },
  46         {
  47           fields: [ 'videoId' ]
  48         }
  49       ]
  50     }
  51   )
  52
  53   const classMethods = [
  54     associate,
  55
  56     listWithLimitAndRandom,
  57     countTotalRequests,
  58     removeAll,
  59     removeByRequestIdsAndPod
  60   ]
  61   addMethodsToModel(RequestVideoQadu, classMethods)
  62
  63   return RequestVideoQadu
  64 }
  65
  66 // ------------------------------ STATICS ------------------------------
  67
  68 function associate (models) {
  69   RequestVideoQadu.belongsTo(models.Pod, {
  70     foreignKey: {
  71       name: 'podId',
  72       allowNull: false
  73     },
  74     onDelete: 'CASCADE'
  75   })
  76
  77   RequestVideoQadu.belongsTo(models.Video, {
  78     foreignKey: {
  79       name: 'videoId',
  80       allowNull: false
  81     },
  82     onDelete: 'CASCADE'
  83   })
  84 }
  85
  86 countTotalRequests = function (callback) {
  87   const query = {}
  88   return RequestVideoQadu.count(query).asCallback(callback)
  89 }
  90
  91 listWithLimitAndRandom = function (limitPods, limitRequestsPerPod, callback) {
  92   const Pod = RequestVideoQadu['sequelize'].models.Pod
  93
  94   Pod.listRandomPodIdsWithRequest(limitPods, 'RequestVideoQadus', function (err, podIds) {
  95     if (err) return callback(err)
  96
  97     // We don't have friends that have requests
  98     if (podIds.length === 0) return callback(null, [])
  99
 100     const query = {
 101       include: [
 102         {
 103           model: RequestVideoQadu['sequelize'].models.Pod,
 104           where: {
 105             id: {
 106               $in: podIds
 107             }
 108           }
 109         },
 110         {
 111           model: RequestVideoQadu['sequelize'].models.Video
 112         }
 113       ]
 114     }
 115
 116     RequestVideoQadu.findAll(query).asCallback(function (err, requests) {
 117       if (err) return callback(err)
 118
 119       const requestsGrouped = groupAndTruncateRequests(requests, limitRequestsPerPod)
 120       return callback(err, requestsGrouped)
 121     })
 122   })
 123 }
 124
 125 removeByRequestIdsAndPod = function (ids, podId, callback) {
 126   const query = {
 127     where: {
 128       id: {
 129         $in: ids
 130       },
 131       podId
 132     }
 133   }
 134
 135   RequestVideoQadu.destroy(query).asCallback(callback)
 136 }
 137
 138 removeAll = function (callback) {
 139   // Delete all requests
 140   RequestVideoQadu.truncate({ cascade: true }).asCallback(callback)
 141 }
 142
 143 // ---------------------------------------------------------------------------
 144
 145 function groupAndTruncateRequests (requests, limitRequestsPerPod) {
 146   const requestsGrouped = {}
 147
 148   requests.forEach(function (request) {
 149     const pod = request.Pod
 150
 151     if (!requestsGrouped[pod.id]) requestsGrouped[pod.id] = []
 152
 153     if (requestsGrouped[pod.id].length < limitRequestsPerPod) {
 154       requestsGrouped[pod.id].push({
 155         request: request,
 156         video: request.Video,
 157         pod
 158       })
 159     }
 160   })
 161
 162   return requestsGrouped
 163 }