tachybase_todo/packages/core/utils/src/perf-hooks.ts

33 lines
1.1 KiB
TypeScript
Raw Normal View History

import { RecordableHistogram, performance } from 'perf_hooks';
export const prePerfHooksWrap = (handler: any, options?: { name?: string }) => {
const { name } = options || {};
return async (ctx: any, next: any) => {
if (!ctx.getPerfHistogram) {
return await handler(ctx, next);
}
const histogram = ctx.getPerfHistogram(name || handler) as RecordableHistogram;
const start = performance.now();
await handler(ctx, async () => {
const duration = performance.now() - start;
histogram.record(Math.ceil(duration * 1e6));
await next();
});
};
};
export const postPerfHooksWrap = (handler: any, options: { name?: string }) => {
const { name } = options || {};
return async (ctx: any, next: any) => {
if (!ctx.getPerfHistogram) {
return await handler(ctx, next);
}
await next();
const histogram = ctx.getPerfHistogram(name || handler) as RecordableHistogram;
const start = performance.now();
await handler(ctx, async () => {});
const duration = performance.now() - start;
histogram.record(Math.ceil(duration * 1e6));
};
};