You've already forked node-redis
mirror of
https://github.com/redis/node-redis.git
synced 2025-08-04 15:02:09 +03:00
improve encodeCommand performance, add set-get-delete-string benchmark
This commit is contained in:
76
benchmark/lib/runner.js
Normal file
76
benchmark/lib/runner.js
Normal file
@@ -0,0 +1,76 @@
|
||||
import yargs from 'yargs';
|
||||
import { hideBin } from 'yargs/helpers';
|
||||
import { basename } from 'path';
|
||||
|
||||
const { path, times, concurrency } = yargs(hideBin(process.argv))
|
||||
.option('path', {
|
||||
type: 'string',
|
||||
demandOption: true
|
||||
})
|
||||
.option('times', {
|
||||
type: 'number',
|
||||
default: 1_000_000,
|
||||
demandOption: true
|
||||
})
|
||||
.option('concurrency', {
|
||||
type: 'number',
|
||||
default: 100,
|
||||
demandOption: true
|
||||
})
|
||||
.parseSync();
|
||||
|
||||
async function setup() {
|
||||
const module = await import(path);
|
||||
await module.setup();
|
||||
return module;
|
||||
}
|
||||
|
||||
function getMetadata() {
|
||||
return new Promise(resolve => {
|
||||
process.once('message', resolve);
|
||||
process.send('ready');
|
||||
});
|
||||
}
|
||||
|
||||
const [ { benchmark, teardown }, metadata ] = await Promise.all([
|
||||
setup(),
|
||||
getMetadata()
|
||||
]);
|
||||
|
||||
async function run(times) {
|
||||
return new Promise(resolve => {
|
||||
let num = 0,
|
||||
inProgress = 0;
|
||||
|
||||
function run() {
|
||||
++inProgress;
|
||||
++num;
|
||||
benchmark(metadata)
|
||||
.catch(err => console.error(err))
|
||||
.finally(() => {
|
||||
--inProgress;
|
||||
|
||||
if (num < times) {
|
||||
run();
|
||||
} else if (inProgress === 0) {
|
||||
resolve();
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
for (let i = 0; i < concurrency; i++) {
|
||||
run();
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// warmup
|
||||
await run(Math.min(times * 0.1, 10_000));
|
||||
|
||||
// benchmark
|
||||
const start = process.hrtime.bigint();
|
||||
await run(times);
|
||||
const took = (process.hrtime.bigint() - start);
|
||||
console.log(`[${basename(path)}]: took ${took / 1_000_000n}ms, ${took / BigInt(times)}ns per operation`);
|
||||
|
||||
await teardown();
|
Reference in New Issue
Block a user