Camilla Schoeser
116 lines
4.0 KiB
JavaScript
116 lines
4.0 KiB
JavaScript
import { ELYSIA_REQUEST_ID } from "./utils.mjs";
|
|
const ELYSIA_TRACE = Symbol("ElysiaTrace"), createProcess = () => {
|
|
const { promise, resolve } = Promise.withResolvers(), { promise: end, resolve: resolveEnd } = Promise.withResolvers(), { promise: error, resolve: resolveError } = Promise.withResolvers(), callbacks = [], callbacksEnd = [];
|
|
return [
|
|
(callback) => (callback && callbacks.push(callback), promise),
|
|
(process) => {
|
|
const processes = [], resolvers = [];
|
|
let groupError = null;
|
|
for (let i = 0; i < (process.total ?? 0); i++) {
|
|
const { promise: promise2, resolve: resolve2 } = Promise.withResolvers(), { promise: end2, resolve: resolveEnd2 } = Promise.withResolvers(), { promise: error2, resolve: resolveError2 } = Promise.withResolvers(), callbacks2 = [], callbacksEnd2 = [];
|
|
processes.push((callback) => (callback && callbacks2.push(callback), promise2)), resolvers.push((process2) => {
|
|
const result2 = {
|
|
...process2,
|
|
end: end2,
|
|
error: error2,
|
|
index: i,
|
|
onStop(callback) {
|
|
return callback && callbacksEnd2.push(callback), end2;
|
|
}
|
|
};
|
|
resolve2(result2);
|
|
for (let i2 = 0; i2 < callbacks2.length; i2++)
|
|
callbacks2[i2](result2);
|
|
return (error3 = null) => {
|
|
const end3 = performance.now();
|
|
error3 && (groupError = error3);
|
|
const detail = {
|
|
end: end3,
|
|
error: error3,
|
|
get elapsed() {
|
|
return end3 - process2.begin;
|
|
}
|
|
};
|
|
for (let i2 = 0; i2 < callbacksEnd2.length; i2++)
|
|
callbacksEnd2[i2](detail);
|
|
resolveEnd2(end3), resolveError2(error3);
|
|
};
|
|
});
|
|
}
|
|
const result = {
|
|
...process,
|
|
end,
|
|
error,
|
|
onEvent(callback) {
|
|
for (let i = 0; i < processes.length; i++)
|
|
processes[i](callback);
|
|
},
|
|
onStop(callback) {
|
|
return callback && callbacksEnd.push(callback), end;
|
|
}
|
|
};
|
|
resolve(result);
|
|
for (let i = 0; i < callbacks.length; i++) callbacks[i](result);
|
|
return {
|
|
resolveChild: resolvers,
|
|
resolve(error2 = null) {
|
|
const end2 = performance.now();
|
|
!error2 && groupError && (error2 = groupError);
|
|
const detail = {
|
|
end: end2,
|
|
error: error2,
|
|
get elapsed() {
|
|
return end2 - process.begin;
|
|
}
|
|
};
|
|
for (let i = 0; i < callbacksEnd.length; i++)
|
|
callbacksEnd[i](detail);
|
|
resolveEnd(end2), resolveError(error2);
|
|
}
|
|
};
|
|
}
|
|
];
|
|
}, createTracer = (traceListener) => (context) => {
|
|
const [onRequest, resolveRequest] = createProcess(), [onParse, resolveParse] = createProcess(), [onTransform, resolveTransform] = createProcess(), [onBeforeHandle, resolveBeforeHandle] = createProcess(), [onHandle, resolveHandle] = createProcess(), [onAfterHandle, resolveAfterHandle] = createProcess(), [onError, resolveError] = createProcess(), [onMapResponse, resolveMapResponse] = createProcess(), [onAfterResponse, resolveAfterResponse] = createProcess();
|
|
return traceListener({
|
|
// @ts-ignore
|
|
id: context[ELYSIA_REQUEST_ID],
|
|
context,
|
|
set: context.set,
|
|
// @ts-ignore
|
|
onRequest,
|
|
// @ts-ignore
|
|
onParse,
|
|
// @ts-ignore
|
|
onTransform,
|
|
// @ts-ignore
|
|
onBeforeHandle,
|
|
// @ts-ignore
|
|
onHandle,
|
|
// @ts-ignore
|
|
onAfterHandle,
|
|
// @ts-ignore
|
|
onMapResponse,
|
|
// @ts-ignore
|
|
onAfterResponse,
|
|
// @ts-ignore
|
|
onError,
|
|
time: Date.now(),
|
|
store: context.store
|
|
}), {
|
|
request: resolveRequest,
|
|
parse: resolveParse,
|
|
transform: resolveTransform,
|
|
beforeHandle: resolveBeforeHandle,
|
|
handle: resolveHandle,
|
|
afterHandle: resolveAfterHandle,
|
|
error: resolveError,
|
|
mapResponse: resolveMapResponse,
|
|
afterResponse: resolveAfterResponse
|
|
};
|
|
};
|
|
export {
|
|
ELYSIA_TRACE,
|
|
createTracer
|
|
};
|