84 lines
1.9 KiB
TypeScript
84 lines
1.9 KiB
TypeScript
// Copyright 2022 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
/* eslint-disable no-await-in-loop, no-console */
|
|
|
|
import createDebug from 'debug';
|
|
import fs from 'fs/promises';
|
|
import path from 'path';
|
|
|
|
import { Bootstrap } from '../bootstrap';
|
|
|
|
export const debug = createDebug('mock:benchmarks');
|
|
|
|
export { Bootstrap };
|
|
export { App } from '../playwright';
|
|
|
|
export type StatsType = {
|
|
mean: number;
|
|
stddev: number;
|
|
[key: string]: number;
|
|
};
|
|
|
|
export const RUN_COUNT = process.env.RUN_COUNT
|
|
? parseInt(process.env.RUN_COUNT, 10)
|
|
: 100;
|
|
|
|
export const GROUP_SIZE = process.env.GROUP_SIZE
|
|
? parseInt(process.env.GROUP_SIZE, 10)
|
|
: 8;
|
|
|
|
export const DISCARD_COUNT = process.env.DISCARD_COUNT
|
|
? parseInt(process.env.DISCARD_COUNT, 10)
|
|
: 5;
|
|
|
|
export function stats(
|
|
list: ReadonlyArray<number>,
|
|
percentiles: ReadonlyArray<number> = []
|
|
): StatsType {
|
|
if (list.length === 0) {
|
|
throw new Error('Empty list given to stats');
|
|
}
|
|
|
|
let mean = 0;
|
|
let stddev = 0;
|
|
|
|
for (const value of list) {
|
|
mean += value;
|
|
stddev += value ** 2;
|
|
}
|
|
mean /= list.length;
|
|
stddev /= list.length;
|
|
|
|
stddev -= mean ** 2;
|
|
stddev = Math.sqrt(stddev);
|
|
|
|
const sorted = list.slice().sort((a, b) => a - b);
|
|
|
|
const result: StatsType = { mean, stddev };
|
|
|
|
for (const p of percentiles) {
|
|
result[`p${p}`] = sorted[Math.floor((sorted.length * p) / 100)];
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
export async function saveLogs(bootstrap: Bootstrap): Promise<void> {
|
|
const { ARTIFACTS_DIR } = process.env;
|
|
if (!ARTIFACTS_DIR) {
|
|
console.error('Not saving logs. Please set ARTIFACTS_DIR env variable');
|
|
return;
|
|
}
|
|
|
|
await fs.mkdir(ARTIFACTS_DIR, { recursive: true });
|
|
|
|
const { logsDir } = bootstrap;
|
|
await fs.rename(logsDir, path.join(ARTIFACTS_DIR, 'logs'));
|
|
}
|
|
|
|
// Can happen if electron exits prematurely
|
|
process.on('unhandledRejection', reason => {
|
|
console.error('Unhandled rejection:');
|
|
console.error(reason);
|
|
process.exit(1);
|
|
});
|