Fix benchmark preparation

pull/6527/head
Chocobozzz 2024-07-15 11:12:50 +02:00
parent 5f0f4ba7de
commit 8bd8b33780
No known key found for this signature in database
GPG Key ID: 583A612D890159BE
1 changed files with 16 additions and 7 deletions

View File

@ -1,14 +1,15 @@
import autocannon, { printResult } from 'autocannon'
import { program } from 'commander'
import { writeJson } from 'fs-extra/esm'
import { Video, VideoPrivacy } from '@peertube/peertube-models' import { Video, VideoPrivacy } from '@peertube/peertube-models'
import { import {
createMultipleServers, createSingleServer,
doubleFollow, doubleFollow,
killallServers, killallServers,
PeerTubeServer, PeerTubeServer,
setAccessTokensToServers setAccessTokensToServers,
waitJobs
} from '@peertube/peertube-server-commands' } from '@peertube/peertube-server-commands'
import autocannon, { printResult } from 'autocannon'
import { program } from 'commander'
import { writeJson } from 'fs-extra/esm'
let servers: PeerTubeServer[] let servers: PeerTubeServer[]
// First server // First server
@ -239,7 +240,7 @@ function runBenchmark (options: {
} }
async function prepare () { async function prepare () {
servers = await createMultipleServers(3, { const config = {
rates_limit: { rates_limit: {
api: { api: {
max: 5_000_000 max: 5_000_000
@ -272,7 +273,13 @@ async function prepare () {
max: 5_000_000 max: 5_000_000
} }
} }
}, { nodeArgs: [ '--inspect' ] }) }
servers = await Promise.all([
createSingleServer(1, config, { nodeArgs: [ '--inspect' ] }),
createSingleServer(2, config),
createSingleServer(3, config)
])
server = servers[0] server = servers[0]
await setAccessTokensToServers(servers) await setAccessTokensToServers(servers)
@ -320,4 +327,6 @@ async function prepare () {
fixture: 'subtitle-good2.vtt' fixture: 'subtitle-good2.vtt'
}) })
} }
await waitJobs(servers)
} }