]>
Commit | Line | Data |
---|---|---|
1 | import { QueryTypes, Sequelize } from 'sequelize' | |
2 | ||
3 | let sequelizes: { [ id: number ]: Sequelize } = {} | |
4 | ||
5 | function getSequelize (serverNumber: number) { | |
6 | if (sequelizes[serverNumber]) return sequelizes[serverNumber] | |
7 | ||
8 | const dbname = 'peertube_test' + serverNumber | |
9 | const username = 'peertube' | |
10 | const password = 'peertube' | |
11 | const host = 'localhost' | |
12 | const port = 5432 | |
13 | ||
14 | const seq = new Sequelize(dbname, username, password, { | |
15 | dialect: 'postgres', | |
16 | host, | |
17 | port, | |
18 | logging: false | |
19 | }) | |
20 | ||
21 | sequelizes[serverNumber] = seq | |
22 | ||
23 | return seq | |
24 | } | |
25 | ||
26 | function setActorField (serverNumber: number, to: string, field: string, value: string) { | |
27 | const seq = getSequelize(serverNumber) | |
28 | ||
29 | const options = { type: QueryTypes.UPDATE } | |
30 | ||
31 | return seq.query(`UPDATE actor SET "${field}" = '${value}' WHERE url = '${to}'`, options) | |
32 | } | |
33 | ||
34 | function setVideoField (serverNumber: number, uuid: string, field: string, value: string) { | |
35 | const seq = getSequelize(serverNumber) | |
36 | ||
37 | const options = { type: QueryTypes.UPDATE } | |
38 | ||
39 | return seq.query(`UPDATE video SET "${field}" = '${value}' WHERE uuid = '${uuid}'`, options) | |
40 | } | |
41 | ||
42 | function setPlaylistField (serverNumber: number, uuid: string, field: string, value: string) { | |
43 | const seq = getSequelize(serverNumber) | |
44 | ||
45 | const options = { type: QueryTypes.UPDATE } | |
46 | ||
47 | return seq.query(`UPDATE "videoPlaylist" SET "${field}" = '${value}' WHERE uuid = '${uuid}'`, options) | |
48 | } | |
49 | ||
50 | async function countVideoViewsOf (serverNumber: number, uuid: string) { | |
51 | const seq = getSequelize(serverNumber) | |
52 | ||
53 | // tslint:disable | |
54 | const query = `SELECT SUM("videoView"."views") AS "total" FROM "videoView" INNER JOIN "video" ON "video"."id" = "videoView"."videoId" WHERE "video"."uuid" = '${uuid}'` | |
55 | ||
56 | const options = { type: QueryTypes.SELECT as QueryTypes.SELECT } | |
57 | const [ { total } ] = await seq.query<{ total: number }>(query, options) | |
58 | ||
59 | if (!total) return 0 | |
60 | ||
61 | // FIXME: check if we really need parseInt | |
62 | return parseInt(total + '', 10) | |
63 | } | |
64 | ||
65 | async function closeAllSequelize (servers: any[]) { | |
66 | for (let i = 1; i <= servers.length; i++) { | |
67 | if (sequelizes[ i ]) { | |
68 | await sequelizes[ i ].close() | |
69 | delete sequelizes[ i ] | |
70 | } | |
71 | } | |
72 | } | |
73 | ||
74 | export { | |
75 | setVideoField, | |
76 | setPlaylistField, | |
77 | setActorField, | |
78 | countVideoViewsOf, | |
79 | closeAllSequelize | |
80 | } |