aboutsummaryrefslogtreecommitdiffhomepage
path: root/server/lib/request-video-event-scheduler.js
diff options
context:
space:
mode:
authorChocobozzz <florian.bigard@gmail.com>2017-02-26 18:57:33 +0100
committerChocobozzz <florian.bigard@gmail.com>2017-02-26 20:01:26 +0100
commite4c87ec26962e359d1c70b03ed188a3f19d6a25b (patch)
tree26fe20e6f600bc6f6f569dde2171b0a2346b135c /server/lib/request-video-event-scheduler.js
parent9e167724f7e933f41d9ea2e1c31772bf4c560a28 (diff)
downloadPeerTube-e4c87ec26962e359d1c70b03ed188a3f19d6a25b.tar.gz
PeerTube-e4c87ec26962e359d1c70b03ed188a3f19d6a25b.tar.zst
PeerTube-e4c87ec26962e359d1c70b03ed188a3f19d6a25b.zip
Server: implement video views
Diffstat (limited to 'server/lib/request-video-event-scheduler.js')
-rw-r--r--server/lib/request-video-event-scheduler.js109
1 files changed, 109 insertions, 0 deletions
diff --git a/server/lib/request-video-event-scheduler.js b/server/lib/request-video-event-scheduler.js
new file mode 100644
index 000000000..5ea5631b0
--- /dev/null
+++ b/server/lib/request-video-event-scheduler.js
@@ -0,0 +1,109 @@
1'use strict'
2
3const BaseRequestScheduler = require('./base-request-scheduler')
4const constants = require('../initializers/constants')
5const db = require('../initializers/database')
6
7module.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}