56 lines
982 B
JavaScript
56 lines
982 B
JavaScript
|
import { performance } from "node:perf_hooks";
|
||
|
|
|
||
|
|
class Benchmark {
|
||
|
|
constructor() {
|
||
|
|
// TypeScript slop
|
||
|
|
this.timeSpent = 0;
|
||
|
|
this.timesCalled = 0;
|
||
|
|
this.beforeTimers = [];
|
||
|
|
}
|
||
|
|
|
||
|
|
reset() {
|
||
|
|
this.timeSpent = 0;
|
||
|
|
this.timesCalled = 0;
|
||
|
|
this.beforeTimers = [];
|
||
|
|
}
|
||
|
|
|
||
|
|
getNewTimestamp() {
|
||
|
|
if (performance) {
|
||
|
|
return performance.now();
|
||
|
|
}
|
||
|
|
return new Date().getTime();
|
||
|
|
}
|
||
|
|
|
||
|
|
incrementCount() {
|
||
|
|
this.timesCalled++;
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO(slightlyoff):
|
||
|
|
// disable all of these hrtime requests when not benchmarking
|
||
|
|
before() {
|
||
|
|
this.timesCalled++;
|
||
|
|
this.beforeTimers.push(this.getNewTimestamp());
|
||
|
|
}
|
||
|
|
|
||
|
|
after() {
|
||
|
|
if (!this.beforeTimers.length) {
|
||
|
|
throw new Error("You called Benchmark after() without a before().");
|
||
|
|
}
|
||
|
|
|
||
|
|
let before = this.beforeTimers.pop();
|
||
|
|
if (!this.beforeTimers.length) {
|
||
|
|
this.timeSpent += this.getNewTimestamp() - before;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
getTimesCalled() {
|
||
|
|
return this.timesCalled;
|
||
|
|
}
|
||
|
|
|
||
|
|
getTotal() {
|
||
|
|
return this.timeSpent;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export default Benchmark;
|
