archive-gh-me/vitest
1
0
Fork 0
mirror of https://github.com/vitest-dev/vitest.git synced 2026-02-01 17:36:51 +00:00
Code Issues Projects Releases Wiki Activity

refactor: improve benchmark implementation (#4268)

Browse Source
This commit is contained in:
Dunqing 2023-10-16 02:18:33 -05:00 committed by GitHub
parent 9112cc96b4
commit a25f6018b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 32 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
packages/vitest/src/runtime/runners/benchmark.ts
View File

@ -47,14 +47,32 @@ async function runBenchmarkSuite(suite: Suite, runner: VitestRunner) {
if (benchmarkGroup.length) {
const defer = createDefer()
const benchmarkMap: Record<string, Benchmark> = {}
suite.result = {
state: 'run',
startTime: start,
benchmark: createBenchmarkResult(suite.name),
}
updateTask(suite)
benchmarkGroup.forEach((benchmark, idx) => {
const addBenchTaskListener = (task: InstanceType<typeof Task>, benchmark: Benchmark) => {
task.addEventListener('complete', (e) => {
const task = e.task
const taskRes = task.result!
const result = benchmark.result!.benchmark!
Object.assign(result, taskRes)
updateTask(benchmark)
}, {
once: true,
})
task.addEventListener('error', (e) => {
const task = e.task
defer.reject(benchmark ? task.result!.error : e)
}, {
once: true,
})
}
benchmarkGroup.forEach((benchmark) => {
const options = getBenchOptions(benchmark)
const benchmarkInstance = new Bench(options)
@ -65,50 +83,32 @@ async function runBenchmarkSuite(suite: Suite, runner: VitestRunner) {
startTime: start,
benchmark: createBenchmarkResult(benchmark.name),
}
const id = idx.toString()
benchmarkMap[id] = benchmark
const task = new Task(benchmarkInstance, id, benchmarkFn)
const task = new Task(benchmarkInstance, benchmark.name, benchmarkFn)
benchmarkTasks.set(benchmark, task)
addBenchTaskListener(task, benchmark)
updateTask(benchmark)
})
benchmarkGroup.forEach((benchmark) => {
const task = benchmarkTasks.get(benchmark)!
task.addEventListener('complete', (e) => {
const task = e.task
const _benchmark = benchmarkMap[task.name || '']
if (_benchmark) {
const taskRes = task.result!
const result = _benchmark.result!.benchmark!
Object.assign(result, taskRes)
updateTask(_benchmark)
}
})
task.addEventListener('error', (e) => {
const task = e.task
const _benchmark = benchmarkMap[task.name || '']
defer.reject(_benchmark ? task.result!.error : e)
})
})
const tasks: BenchTask[] = []
const { setTimeout } = getSafeTimers()
const tasks: [BenchTask, Benchmark][] = []
for (const benchmark of benchmarkGroup) {
const task = benchmarkTasks.get(benchmark)!
await task.warmup()
const { setTimeout } = getSafeTimers()
tasks.push(await new Promise<BenchTask>(resolve => setTimeout(async () => {
resolve(await task.run())
})))
tasks.push([
await new Promise<BenchTask>(resolve => setTimeout(async () => {
resolve(await task.run())
})),
benchmark,
])
}
suite.result!.duration = performance.now() - start
suite.result!.state = 'pass'
tasks
.sort((a, b) => a.result!.mean - b.result!.mean)
.forEach((cycle, idx) => {
const benchmark = benchmarkMap[cycle.name || '']
.sort(([taskA], [taskB]) => taskA.result!.mean - taskB.result!.mean)
.forEach(([, benchmark], idx) => {
benchmark.result!.state = 'pass'
if (benchmark) {
const result = benchmark.result!.benchmark!