aboutsummaryrefslogtreecommitdiffhomepage
path: root/scripts/simulate-many-viewers.ts
blob: fb666c3182c91466aa89419d1e5835e75cf973da (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import Bluebird from 'bluebird'
import { wait } from '@shared/core-utils'
import {
  createSingleServer,
  doubleFollow,
  killallServers,
  PeerTubeServer,
  setAccessTokensToServers,
  waitJobs
} from '@shared/server-commands'

let servers: PeerTubeServer[]
const viewers: { xForwardedFor: string }[] = []
let videoId: string

run()
  .then(() => process.exit(0))
  .catch(err => console.error(err))
  .finally(() => killallServers(servers))

async function run () {
  await prepare()

  while (true) {
    await runViewers()
  }
}

async function prepare () {
  console.log('Preparing servers...')

  const config = {
    log: {
      level: 'info'
    },
    rates_limit: {
      api: {
        max: 5_000_000
      }
    },
    views: {
      videos: {
        local_buffer_update_interval: '30 minutes',
        ip_view_expiration: '1 hour'
      }
    }
  }

  servers = await Promise.all([
    createSingleServer(1, config, { nodeArgs: [ '--inspect' ] }),
    createSingleServer(2, config),
    createSingleServer(3, config)
  ])

  await setAccessTokensToServers(servers)
  await doubleFollow(servers[0], servers[1])
  await doubleFollow(servers[0], servers[2])

  const { uuid } = await servers[0].videos.quickUpload({ name: 'video' })
  videoId = uuid

  await waitJobs(servers)

  const THOUSAND_VIEWERS = 2

  for (let i = 2; i < 252; i++) {
    for (let j = 2; j < 6; j++) {
      for (let k = 2; k < THOUSAND_VIEWERS + 2; k++) {
        viewers.push({ xForwardedFor: `0.${k}.${j}.${i},127.0.0.1` })
      }
    }
  }

  console.log('Servers preparation finished.')
}

async function runViewers () {
  console.log('Will run views of %d viewers.', viewers.length)

  const before = new Date().getTime()

  await Bluebird.map(viewers, viewer => {
    return servers[0].views.simulateView({ id: videoId, xForwardedFor: viewer.xForwardedFor })
  }, { concurrency: 100 })

  console.log('Finished to run views in %d seconds.', (new Date().getTime() - before) / 1000)

  await wait(5000)
}