geniURL-Mirror/test/latency-test.ts

// NOTE:
// requires the env vars HTTP_PORT and AUTH_TOKENS to be set

import "dotenv/config";
import _axios from "axios";
import percentile from "percentile";

const settings = {
    amount: 50,
    url: `http://127.0.0.1:${process.env.HTTP_PORT}/search/top?q=pink guy - dog festival directions`,
};


const axios = _axios.create({ timeout: 10_000 });

async function run() {
    console.log(`\n\n>>> Running latency test with ${settings.amount} requests...\n`);
    const startTs = Date.now();

    const times = [];
    for(let i = 0; i < settings.amount; i++) {
        const start = Date.now();
        await axios.get(settings.url, {
            headers: {
                "Cache-Control": "no-cache",
                Authorization: `Bearer ${process.env.AUTH_TOKENS!.split(",")[0]}`,
            },
        });
        times.push(Date.now() - start);

        i % 10 === 0 && i !== 0 && console.log(`Sent ${i} of ${settings.amount} requests`);
    }

    const avg = (times.reduce((a, c) => a + c, 0) / times.length).toFixed(0);
    const perc80 = percentile(80, times);
    const perc95 = percentile(95, times);
    const perc99 = percentile(99, times);

    console.log(`\n>>> Latency test finished after ${((Date.now() - startTs) / 1000).toFixed(2)}s`);
    console.log(`avg:\t${avg}\tms`);
    console.log(`80th%:\t${perc80}\tms`);
    console.log(`95th%:\t${perc95}\tms`);
    console.log(`99th%:\t${perc99}\tms`);
    console.log();
}

run();