]>
Commit | Line | Data |
---|---|---|
69818c93 C |
1 | import * as Sequelize from 'sequelize' |
2 | ||
e02643f3 | 3 | import { database as db } from '../../initializers/database' |
4771e000 | 4 | import { AbstractRequestScheduler, RequestsObjects } from './abstract-request-scheduler' |
65fcc311 C |
5 | import { |
6 | REQUESTS_VIDEO_EVENT_LIMIT_PODS, | |
7 | REQUESTS_VIDEO_EVENT_LIMIT_PER_POD, | |
8 | REQUEST_VIDEO_EVENT_ENDPOINT | |
9 | } from '../../initializers' | |
6fcd19ba | 10 | import { RequestsVideoEventGrouped } from '../../models' |
4771e000 | 11 | import { RequestVideoEventType, RemoteVideoEventRequest, RemoteVideoEventType } from '../../../shared' |
65fcc311 | 12 | |
69818c93 | 13 | export type RequestVideoEventSchedulerOptions = { |
ee9e7b61 | 14 | type: RequestVideoEventType |
0a6658fd | 15 | videoId: number |
69818c93 C |
16 | count?: number |
17 | transaction?: Sequelize.Transaction | |
18 | } | |
19 | ||
6fcd19ba | 20 | class RequestVideoEventScheduler extends AbstractRequestScheduler<RequestsVideoEventGrouped> { |
e4c87ec2 C |
21 | constructor () { |
22 | super() | |
23 | ||
24 | // We limit the size of the requests | |
65fcc311 C |
25 | this.limitPods = REQUESTS_VIDEO_EVENT_LIMIT_PODS |
26 | this.limitPerPod = REQUESTS_VIDEO_EVENT_LIMIT_PER_POD | |
e4c87ec2 C |
27 | |
28 | this.description = 'video event requests' | |
29 | } | |
30 | ||
31 | getRequestModel () { | |
32 | return db.RequestVideoEvent | |
33 | } | |
34 | ||
35 | getRequestToPodModel () { | |
36 | return db.RequestVideoEvent | |
37 | } | |
38 | ||
4771e000 C |
39 | buildRequestsObjects (eventRequests: RequestsVideoEventGrouped) { |
40 | const requestsToMakeGrouped: RequestsObjects<RemoteVideoEventRequest> = {} | |
e4c87ec2 C |
41 | |
42 | /* Example: | |
43 | { | |
44 | pod1: { | |
45 | video1: { views: 4, likes: 5 }, | |
46 | video2: { likes: 5 } | |
47 | } | |
48 | } | |
49 | */ | |
4771e000 C |
50 | const eventsPerVideoPerPod: { |
51 | [ podId: string ]: { | |
0a6658fd | 52 | [ videoUUID: string ]: { |
4771e000 C |
53 | views?: number |
54 | likes?: number | |
55 | dislikes?: number | |
56 | } | |
57 | } | |
58 | } = {} | |
e4c87ec2 C |
59 | |
60 | // We group video events per video and per pod | |
61 | // We add the counts of the same event types | |
6fcd19ba C |
62 | Object.keys(eventRequests).forEach(toPodId => { |
63 | eventRequests[toPodId].forEach(eventToProcess => { | |
e4c87ec2 C |
64 | if (!eventsPerVideoPerPod[toPodId]) eventsPerVideoPerPod[toPodId] = {} |
65 | ||
66 | if (!requestsToMakeGrouped[toPodId]) { | |
67 | requestsToMakeGrouped[toPodId] = { | |
68 | toPod: eventToProcess.pod, | |
65fcc311 | 69 | endpoint: REQUEST_VIDEO_EVENT_ENDPOINT, |
e4c87ec2 C |
70 | ids: [], // request ids, to delete them from the DB in the future |
71 | datas: [] // requests data | |
72 | } | |
73 | } | |
74 | requestsToMakeGrouped[toPodId].ids.push(eventToProcess.id) | |
75 | ||
76 | const eventsPerVideo = eventsPerVideoPerPod[toPodId] | |
0a6658fd C |
77 | const uuid = eventToProcess.video.uuid |
78 | if (!eventsPerVideo[uuid]) eventsPerVideo[uuid] = {} | |
e4c87ec2 | 79 | |
0a6658fd | 80 | const events = eventsPerVideo[uuid] |
e4c87ec2 C |
81 | if (!events[eventToProcess.type]) events[eventToProcess.type] = 0 |
82 | ||
83 | events[eventToProcess.type] += eventToProcess.count | |
84 | }) | |
85 | }) | |
86 | ||
87 | // Now we build our requests array per pod | |
88 | Object.keys(eventsPerVideoPerPod).forEach(toPodId => { | |
89 | const eventsForPod = eventsPerVideoPerPod[toPodId] | |
90 | ||
0a6658fd C |
91 | Object.keys(eventsForPod).forEach(uuid => { |
92 | const eventsForVideo = eventsForPod[uuid] | |
e4c87ec2 C |
93 | |
94 | Object.keys(eventsForVideo).forEach(eventType => { | |
95 | requestsToMakeGrouped[toPodId].datas.push({ | |
96 | data: { | |
0a6658fd | 97 | uuid, |
4771e000 C |
98 | eventType: eventType as RemoteVideoEventType, |
99 | count: +eventsForVideo[eventType] | |
e4c87ec2 C |
100 | } |
101 | }) | |
102 | }) | |
103 | }) | |
104 | }) | |
105 | ||
106 | return requestsToMakeGrouped | |
107 | } | |
108 | ||
6fcd19ba | 109 | createRequest ({ type, videoId, count, transaction }: RequestVideoEventSchedulerOptions) { |
e4c87ec2 C |
110 | if (count === undefined) count = 1 |
111 | ||
69818c93 | 112 | const dbRequestOptions: Sequelize.CreateOptions = {} |
e4c87ec2 C |
113 | if (transaction) dbRequestOptions.transaction = transaction |
114 | ||
115 | const createQuery = { | |
116 | type, | |
117 | count, | |
118 | videoId | |
119 | } | |
120 | ||
6fcd19ba | 121 | return db.RequestVideoEvent.create(createQuery, dbRequestOptions) |
e4c87ec2 C |
122 | } |
123 | } | |
65fcc311 C |
124 | |
125 | // --------------------------------------------------------------------------- | |
126 | ||
127 | export { | |
128 | RequestVideoEventScheduler | |
129 | } |