]>
Commit | Line | Data |
---|---|---|
df66d815 C |
1 | import * as Sequelize from 'sequelize' |
2 | ||
c8000975 C |
3 | let sequelizes: { [ id: number ]: Sequelize.Sequelize } = {} |
4 | ||
df66d815 | 5 | function getSequelize (serverNumber: number) { |
c8000975 C |
6 | if (sequelizes[serverNumber]) return sequelizes[serverNumber] |
7 | ||
df66d815 C |
8 | const dbname = 'peertube_test' + serverNumber |
9 | const username = 'peertube' | |
10 | const password = 'peertube' | |
11 | const host = 'localhost' | |
12 | const port = 5432 | |
13 | ||
c8000975 | 14 | const seq = new Sequelize(dbname, username, password, { |
df66d815 C |
15 | dialect: 'postgres', |
16 | host, | |
17 | port, | |
18 | operatorsAliases: false, | |
19 | logging: false | |
20 | }) | |
c8000975 C |
21 | |
22 | sequelizes[serverNumber] = seq | |
23 | ||
24 | return seq | |
df66d815 C |
25 | } |
26 | ||
27 | function setActorField (serverNumber: number, to: string, field: string, value: string) { | |
28 | const seq = getSequelize(serverNumber) | |
29 | ||
30 | const options = { type: Sequelize.QueryTypes.UPDATE } | |
31 | ||
32 | return seq.query(`UPDATE actor SET "${field}" = '${value}' WHERE url = '${to}'`, options) | |
33 | } | |
34 | ||
a130f33c C |
35 | function setVideoField (serverNumber: number, uuid: string, field: string, value: string) { |
36 | const seq = getSequelize(serverNumber) | |
37 | ||
38 | const options = { type: Sequelize.QueryTypes.UPDATE } | |
39 | ||
40 | return seq.query(`UPDATE video SET "${field}" = '${value}' WHERE uuid = '${uuid}'`, options) | |
41 | } | |
42 | ||
9f79ade6 C |
43 | function setPlaylistField (serverNumber: number, uuid: string, field: string, value: string) { |
44 | const seq = getSequelize(serverNumber) | |
45 | ||
46 | const options = { type: Sequelize.QueryTypes.UPDATE } | |
47 | ||
48 | return seq.query(`UPDATE "videoPlaylist" SET "${field}" = '${value}' WHERE uuid = '${uuid}'`, options) | |
49 | } | |
50 | ||
c8000975 C |
51 | async function closeAllSequelize (servers: any[]) { |
52 | for (let i = 1; i <= servers.length; i++) { | |
53 | if (sequelizes[ i ]) { | |
54 | await sequelizes[ i ].close() | |
55 | delete sequelizes[ i ] | |
56 | } | |
57 | } | |
58 | } | |
59 | ||
df66d815 | 60 | export { |
a130f33c | 61 | setVideoField, |
9f79ade6 | 62 | setPlaylistField, |
c8000975 C |
63 | setActorField, |
64 | closeAllSequelize | |
df66d815 | 65 | } |