refactor(cli): move the benchmarkd command to folder

pull/90/head
moonrailgun 2 years ago
parent 3f6c3b08d8
commit 48145a01ee

@ -0,0 +1,11 @@
import { CommandModule } from 'yargs';
import { benchMessageCommand } from './message';
// https://docs.docker.com/engine/api/v1.41/
export const benchmarkCommand: CommandModule = {
command: 'benchmark',
describe: 'Tailchat Benchmark Test',
builder: (yargs) => yargs.command(benchMessageCommand).demandCommand(),
handler(args) {},
};

@ -10,15 +10,11 @@ import ora from 'ora';
import prettyMs from 'pretty-ms'; import prettyMs from 'pretty-ms';
import filesize from 'filesize'; import filesize from 'filesize';
export const benchCommand: CommandModule = { export const benchMessageCommand: CommandModule = {
command: 'bench', command: 'message',
describe: 'Benchmark', describe:
builder: (yargs) =>
yargs
.command(
'message',
'Stress testing through Tailchat network requests (suitable for pure business testing)', 'Stress testing through Tailchat network requests (suitable for pure business testing)',
(yargs) => builder: (yargs) =>
yargs yargs
.option('groupId', { .option('groupId', {
describe: 'Group ID', describe: 'Group ID',
@ -50,8 +46,7 @@ export const benchCommand: CommandModule = {
type: 'number', type: 'number',
default: Infinity, default: Infinity,
}), }),
async handler(args) {
async (args) => {
config(); // 加载环境变量 config(); // 加载环境变量
const broker = new TcBroker({ const broker = new TcBroker({
@ -66,9 +61,9 @@ export const benchCommand: CommandModule = {
console.log('==============='); console.log('===============');
await startBenchmark<number>({ await startBenchmark<number>({
parallel: args.parallel, parallel: args.parallel as boolean,
parallelLimit: args.parallelLimit, parallelLimit: args.parallelLimit as number,
number: args.num, number: args.num as number,
task: async (i) => { task: async (i) => {
const start = process.hrtime(); const start = process.hrtime();
await broker.call( await broker.call(
@ -97,10 +92,7 @@ export const benchCommand: CommandModule = {
}); });
await broker.stop(); await broker.stop();
} },
)
.demandCommand(),
handler() {},
}; };
/** /**

@ -4,7 +4,7 @@ import { createCommand } from './commands/create';
import { connectCommand } from './commands/connect'; import { connectCommand } from './commands/connect';
import { appCommand } from './commands/app'; import { appCommand } from './commands/app';
import { declarationCommand } from './commands/declaration'; import { declarationCommand } from './commands/declaration';
import { benchCommand } from './commands/bench'; import { benchmarkCommand } from './commands/benchmark';
import { dockerCommand } from './commands/docker'; import { dockerCommand } from './commands/docker';
import { usageCommand } from './commands/usage'; import { usageCommand } from './commands/usage';
import { registryCommand } from './commands/registry'; import { registryCommand } from './commands/registry';
@ -15,7 +15,7 @@ yargs
.command(createCommand) .command(createCommand)
.command(connectCommand) .command(connectCommand)
.command(appCommand) .command(appCommand)
.command(benchCommand) .command(benchmarkCommand)
.command(declarationCommand) .command(declarationCommand)
.command(dockerCommand) .command(dockerCommand)
.command(registryCommand) .command(registryCommand)

Loading…
Cancel
Save