20 lines
698 B
JavaScript
20 lines
698 B
JavaScript
import { eventHandler } from "h3";
|
|
export const globalTiming = globalThis.__timing__ || {
|
|
start: () => 0,
|
|
end: () => 0,
|
|
metrics: []
|
|
};
|
|
export const timingMiddleware = eventHandler((event) => {
|
|
const start = globalTiming.start();
|
|
const _end = event.res.end;
|
|
event.res.end = function(chunk, encoding, cb) {
|
|
const metrics = [["Generate", globalTiming.end(start)], ...globalTiming.metrics];
|
|
const serverTiming = metrics.map((m) => `-;dur=${m[1]};desc="${encodeURIComponent(m[0])}"`).join(", ");
|
|
if (!event.res.headersSent) {
|
|
event.res.setHeader("Server-Timing", serverTiming);
|
|
}
|
|
_end.call(event.res, chunk, encoding, cb);
|
|
return this;
|
|
}.bind(event.res);
|
|
});
|