mirror of https://github.com/Chocobozzz/PeerTube
131 lines
3.9 KiB
TypeScript
131 lines
3.9 KiB
TypeScript
import * as Sequelize from 'sequelize'
|
|
|
|
import { database as db } from '../../initializers/database'
|
|
import { AbstractRequestScheduler } from './abstract-request-scheduler'
|
|
import { logger } from '../../helpers'
|
|
import {
|
|
REQUESTS_VIDEO_QADU_LIMIT_PODS,
|
|
REQUESTS_VIDEO_QADU_LIMIT_PER_POD,
|
|
REQUEST_VIDEO_QADU_ENDPOINT,
|
|
REQUEST_VIDEO_QADU_TYPES
|
|
} from '../../initializers'
|
|
import { RequestVideoQaduType } from '../../../shared'
|
|
|
|
export type RequestVideoQaduSchedulerOptions = {
|
|
type: RequestVideoQaduType
|
|
videoId: string
|
|
transaction?: Sequelize.Transaction
|
|
}
|
|
|
|
class RequestVideoQaduScheduler extends AbstractRequestScheduler {
|
|
constructor () {
|
|
super()
|
|
|
|
// We limit the size of the requests
|
|
this.limitPods = REQUESTS_VIDEO_QADU_LIMIT_PODS
|
|
this.limitPerPod = REQUESTS_VIDEO_QADU_LIMIT_PER_POD
|
|
|
|
this.description = 'video QADU requests'
|
|
}
|
|
|
|
getRequestModel () {
|
|
return db.RequestVideoQadu
|
|
}
|
|
|
|
getRequestToPodModel () {
|
|
return db.RequestVideoQadu
|
|
}
|
|
|
|
buildRequestObjects (requests: { [ toPodId: number ]: any }[]) {
|
|
const requestsToMakeGrouped = {}
|
|
|
|
Object.keys(requests).forEach(toPodId => {
|
|
requests[toPodId].forEach(data => {
|
|
const request = data.request
|
|
const video = data.video
|
|
const pod = data.pod
|
|
const hashKey = toPodId
|
|
|
|
if (!requestsToMakeGrouped[hashKey]) {
|
|
requestsToMakeGrouped[hashKey] = {
|
|
toPod: pod,
|
|
endpoint: REQUEST_VIDEO_QADU_ENDPOINT,
|
|
ids: [], // request ids, to delete them from the DB in the future
|
|
datas: [], // requests data
|
|
videos: {}
|
|
}
|
|
}
|
|
|
|
// Maybe another attribute was filled for this video
|
|
let videoData = requestsToMakeGrouped[hashKey].videos[video.id]
|
|
if (!videoData) videoData = {}
|
|
|
|
switch (request.type) {
|
|
case REQUEST_VIDEO_QADU_TYPES.LIKES:
|
|
videoData.likes = video.likes
|
|
break
|
|
|
|
case REQUEST_VIDEO_QADU_TYPES.DISLIKES:
|
|
videoData.dislikes = video.dislikes
|
|
break
|
|
|
|
case REQUEST_VIDEO_QADU_TYPES.VIEWS:
|
|
videoData.views = video.views
|
|
break
|
|
|
|
default:
|
|
logger.error('Unknown request video QADU type %s.', request.type)
|
|
return
|
|
}
|
|
|
|
// Do not forget the remoteId so the remote pod can identify the video
|
|
videoData.remoteId = video.id
|
|
requestsToMakeGrouped[hashKey].ids.push(request.id)
|
|
|
|
// Maybe there are multiple quick and dirty update for the same video
|
|
// We use this hashmap to dedupe them
|
|
requestsToMakeGrouped[hashKey].videos[video.id] = videoData
|
|
})
|
|
})
|
|
|
|
// Now we deduped similar quick and dirty updates, we can build our requests datas
|
|
Object.keys(requestsToMakeGrouped).forEach(hashKey => {
|
|
Object.keys(requestsToMakeGrouped[hashKey].videos).forEach(videoId => {
|
|
const videoData = requestsToMakeGrouped[hashKey].videos[videoId]
|
|
|
|
requestsToMakeGrouped[hashKey].datas.push({
|
|
data: videoData
|
|
})
|
|
})
|
|
|
|
// We don't need it anymore, it was just to build our datas array
|
|
delete requestsToMakeGrouped[hashKey].videos
|
|
})
|
|
|
|
return requestsToMakeGrouped
|
|
}
|
|
|
|
createRequest ({ type, videoId, transaction }: RequestVideoQaduSchedulerOptions, callback: (err: Error) => void) {
|
|
const dbRequestOptions: Sequelize.BulkCreateOptions = {}
|
|
if (transaction) dbRequestOptions.transaction = transaction
|
|
|
|
// Send the update to all our friends
|
|
db.Pod.listAllIds(transaction, function (err, podIds) {
|
|
if (err) return callback(err)
|
|
|
|
const queries = []
|
|
podIds.forEach(podId => {
|
|
queries.push({ type, videoId, podId })
|
|
})
|
|
|
|
return db.RequestVideoQadu.bulkCreate(queries, dbRequestOptions).asCallback(callback)
|
|
})
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
export {
|
|
RequestVideoQaduScheduler
|
|
}
|