1 import Bluebird from 'bluebird'
2 import { wait } from '@shared/core-utils'
8 setAccessTokensToServers,
10 } from '@shared/server-commands'
12 let servers: PeerTubeServer[]
13 const viewers: { xForwardedFor: string }[] = []
17 .then(() => process.exit(0))
18 .catch(err => console.error(err))
19 .finally(() => killallServers(servers))
21 async function run () {
29 async function prepare () {
30 console.log('Preparing servers...')
43 local_buffer_update_interval: '30 minutes',
44 ip_view_expiration: '1 hour'
49 const env = { PRODUCTION_CONSTANTS: 'true' }
51 servers = await Promise.all([
52 createSingleServer(1, config, { env, nodeArgs: [ '--inspect' ] }),
53 createSingleServer(2, config, { env }),
54 createSingleServer(3, config, { env })
57 await setAccessTokensToServers(servers)
58 await doubleFollow(servers[0], servers[1])
59 await doubleFollow(servers[0], servers[2])
61 const { uuid } = await servers[0].videos.quickUpload({ name: 'video' })
64 await waitJobs(servers)
66 const THOUSAND_VIEWERS = 2
68 for (let i = 2; i < 252; i++) {
69 for (let j = 2; j < 6; j++) {
70 for (let k = 2; k < THOUSAND_VIEWERS + 2; k++) {
71 viewers.push({ xForwardedFor: `0.${k}.${j}.${i},127.0.0.1` })
76 console.log('Servers preparation finished.')
79 async function runViewers () {
80 console.log('Will run views of %d viewers.', viewers.length)
82 const before = new Date().getTime()
84 await Bluebird.map(viewers, viewer => {
85 return servers[0].views.simulateView({ id: videoId, xForwardedFor: viewer.xForwardedFor })
86 }, { concurrency: 500 })
88 console.log('Finished to run views in %d seconds.', (new Date().getTime() - before) / 1000)