elysia 2.0.0-beta.21 → 2.0.0-beta.22

package/dist/compile/handler/jit.d.ts

@@ -7,9 +7,15 @@ import { TailMode } from "./utils.js";
 
 //#region src/compile/handler/jit.d.ts
 /**
- * @internal test hook: receives each route's emitted source. A sync-first
- * route's async tail is a helper beside it, so `toString()` of the compiled
- * handler does not show the whole pipeline
+ * @internal one factory per emitted source, each route calls it for its own
+ * closure. JSC stops sharing code between identical `new Function` sources
+ * once its source cache churns, so without this every route links a copy
+ */
+declare const factoryCache: Map<string, Function>;
+declare const FACTORY_CACHE_LIMIT = 256;
+/**
+ * @internal test hook: receives each route's full emitted source, including
+ * the async tail helper that `toString()` of the handler does not show
  */
 declare let onEmit: ((code: string) => void) | undefined;
 declare const setOnEmit: (fn: typeof onEmit) => void;
@@ -27,19 +33,13 @@ interface CompileHandlerJitOptions {
   isHandleFunction: boolean;
   isStaticResponse: boolean;
   isPromiseHandler: boolean;
-  /**
-   * Non-Error classes a returned value is rethrown for, see
-   * `returnedErrorClasses`
-   */
+  /** Non-Error classes a returned value is rethrown for (`returnedErrorClasses`) */
   errorClasses?: Function[];
-  /**
-   * Per-route descriptor + compile artifacts, computed by `describeRoute`.
-   * The JIT no longer re-derives these facts; it names its emissions off them.
-   */
+  /** Per-route descriptor + compile artifacts from `describeRoute` */
   state: RouteCompileState;
   /** `routeShape`, recorded with the captured code */
   shape?: number;
 }
 declare function compileHandlerJit(options: CompileHandlerJitOptions, tailPass?: TailMode): CompiledHandler;
 //#endregion
-export { CompileHandlerJitOptions, compileHandlerJit, createInlineHandler, setAsyncTail, setCaptureHeaderShorthand, setOnEmit };
+export { CompileHandlerJitOptions, FACTORY_CACHE_LIMIT, compileHandlerJit, createInlineHandler, factoryCache, setAsyncTail, setCaptureHeaderShorthand, setOnEmit };

package/dist/compile/handler/jit.js

@@ -17,20 +17,25 @@ const require_compile_jit_probe = require('../jit-probe.js');
 
 //#region src/compile/handler/jit.ts
 const awaitValue = (value, arm = "") => arm ? `await (_av=(${value}),${arm},_av)` : `await ${value}`;
+/**
+* @internal one factory per emitted source, each route calls it for its own
+* closure. JSC stops sharing code between identical `new Function` sources
+* once its source cache churns, so without this every route links a copy
+*/
+const factoryCache = /* @__PURE__ */ new Map();
+const FACTORY_CACHE_LIMIT = 256;
 let captureHeaderShorthand;
 /**
-* @internal test hook: receives each route's emitted source. A sync-first
-* route's async tail is a helper beside it, so `toString()` of the compiled
-* handler does not show the whole pipeline
+* @internal test hook: receives each route's full emitted source, including
+* the async tail helper that `toString()` of the handler does not show
 */
 let onEmit;
 const setOnEmit = (fn) => {
 	onEmit = fn;
 };
 /**
-* @internal test hook: `false` compiles sync-first routes on the `async`
-* lane instead, the reference the async tail is differentially tested
-* against. `undefined` restores the default (see `tail` below)
+* @internal test hook: `false` compiles sync-first routes on the plain
+* `async` lane (the differential reference), `undefined` restores the default
 */
 let asyncTailOn;
 const setAsyncTail = (on) => {
@@ -161,9 +166,22 @@ const createInlineHandler = (map, h) => ((c) => {
 	if (typeof r?.then === "function") return Promise.resolve(r).then((v) => map(require_handler_utils.forwardError(v), c.request, true));
 	return map(r, c.request, true);
 });
+const createInlineSetHandler = (map, h) => ((c) => {
+	const r = h(c);
+	if (r instanceof Error) throw r;
+	if (typeof r?.then === "function") return Promise.resolve(r).then((v) => map(require_handler_utils.forwardError(v), c.set, c.request, true));
+	return map(r, c.set, c.request, true);
+});
+const createInlineDefaultHeaderHandler = (map, h) => ((c) => {
+	require_adapter_utils.materializeSetHeaders(c.set);
+	const r = h(c);
+	if (r instanceof Error) throw r;
+	if (typeof r?.then === "function") return Promise.resolve(r).then((v) => map(require_handler_utils.forwardError(v), c.set, c.request, true));
+	return map(r, c.set, c.request, true);
+});
 function compileHandlerJit(options, tailPass) {
 	const { method, path, handler, root, errorRoot, hook, adapter, isHandleFunction, isStaticResponse, isPromiseHandler, errorClasses, state, shape } = options;
-	const { vali, inference, cookieConfig, beforeHandlePrefix, traceHandlers, tracePhases, hasAnyPhase, traceHandleOn, descriptor: { async: isAsync, tail: tailRoute, responseMode, hasBody, bodyValiIsAsync, headersValiIsAsync, paramsValiIsAsync, queryValiIsAsync, cookieValiIsAsync: cookieValidIsAsync, responseValiAsync, hasCookieSign, syncCookieSign, asyncCookieSign, lazyCookieVerify, hasErrorHook, hasAfterResponse, hasBeforeHandle, hasAfterHandle, hasMapResponse, hasResponseValidator, hasTrace, traceCount, hasLifecycleHook, syncErrorHook, syncAfterResponse } } = state;
+	const { vali, inference, cookieConfig, beforeHandlePrefix, traceHandlers, tracePhases, hasAnyPhase, traceHandleOn, descriptor: { async: isAsync, tail: tailRoute, responseMode, hasBody, bodyValiIsAsync, headersValiIsAsync, paramsValiIsAsync, queryValiIsAsync, cookieValiIsAsync, responseValiAsync, hasCookieSign, syncCookieSign, asyncCookieSign, lazyCookieVerify, hasErrorHook, hasAfterResponse, hasBeforeHandle, hasAfterHandle, hasMapResponse, hasResponseValidator, hasTrace, traceCount, hasLifecycleHook, syncErrorHook, syncAfterResponse } } = state;
 	const hasStaticAfterResponse = !!hook?.afterResponse?.length;
 	const hasDeriveDispose = !!hook?.["~deriveEntries"]?.length;
 	const hasDynamicAfterResponse = !!inference.afterResponse || hasDeriveDispose;
@@ -177,7 +195,7 @@ function compileHandlerJit(options, tailPass) {
 	} : isAsync;
 	const plain = (code) => require_compile_handler_utils.tailPlain(asyncMode, code);
 	const seenKeys = new Set(["rt", "fre"]);
-	const paramValues = [errorRoot, require_handler_utils.finalizeRouteError];
+	const paramValues = [errorRoot, require_handler_utils.finalizeRouteErrorOf(hook)];
 	const aliasKeys = [];
 	function link(v, key) {
 		if (!seenKeys.has(key)) {
@@ -193,8 +211,8 @@ function compileHandlerJit(options, tailPass) {
 	const abortPeek = "c['~sig']?.aborted";
 	const arm = abortOn ? "_as??=(c['~sig']??=c.request.signal)" : "";
 	let abortSchedule = "";
-	let discard = "";
-	const abortCheck = () => abortOn ? plain(abortSchedule || discard ? `if(${abortPeek}){${discard}${abortSchedule}return emp.clone()}\n` : `if(${abortPeek})return emp.clone()\n`) : "";
+	let abortDiscard = "";
+	const abortCheck = () => abortOn ? plain(abortSchedule || abortDiscard ? `if(${abortPeek}){${abortDiscard}${abortSchedule}return emp.clone()}\n` : `if(${abortPeek})return emp.clone()\n`) : "";
 	const abortChainGuard = () => abortOn ? abortPeek : void 0;
 	const abortCatch = abortOn ? `if((${arm}).aborted)return emp.clone()\n` : "";
 	const phaseOn = (phase) => hasTrace && (tracePhases === null || tracePhases.has(phase));
@@ -331,7 +349,7 @@ function compileHandlerJit(options, tailPass) {
 			code += `c.cookie=bcj(c.set,_ck,cc,1)\n`;
 		} else {
 			let deferDecode = false;
-			if (!hasCookieSign && !cookieValidIsAsync) if (!vali?.cookie) {
+			if (!hasCookieSign && !cookieValiIsAsync) if (!vali?.cookie) {
 				link(require_cookie_utils.parseCookieRawDeferred, "pcrd");
 				code += `let _ck=pcrd(${cookieHeaderExpr},cc)\n`;
 				deferDecode = true;
@@ -339,7 +357,7 @@ function compileHandlerJit(options, tailPass) {
 				link(require_cookie_utils.parseCookieRawSync, "pcrs");
 				code += `let _ck=pcrs(${cookieHeaderExpr},cc)\n`;
 			}
-			else if (syncCookieSign && !cookieValidIsAsync) {
+			else if (syncCookieSign && !cookieValiIsAsync) {
 				link(require_cookie_utils.parseCookieRawSigned, "pcrsg");
 				code += `let _ck=pcrsg(${cookieHeaderExpr},cc)\n`;
 			} else {
@@ -349,8 +367,8 @@ function compileHandlerJit(options, tailPass) {
 			if (vali?.cookie) {
 				link(vali, "va");
 				const cookieIsOptional = !!(hook?.cookie)?.["~optional"];
-				const value = `va.cookie.From(_ck,${fromArgs("cookie", !!cookieValidIsAsync)})`;
-				const validateExpr = `_ck=${cookieValidIsAsync ? awaitValue(value, arm) : value}\n`;
+				const value = `va.cookie.From(_ck,${fromArgs("cookie", cookieValiIsAsync)})`;
+				const validateExpr = `_ck=${cookieValiIsAsync ? awaitValue(value, arm) : value}\n`;
 				if (cookieIsOptional) code += `if(Object.keys(_ck).length){${validateExpr}}\n`;
 				else code += validateExpr;
 			}
@@ -388,8 +406,9 @@ function compileHandlerJit(options, tailPass) {
 	abortSchedule = hasDeriveDispose ? syncAfterResponse ? catchSchedule : schedule : "";
 	const freThenSchedule = (arg) => catchSchedule ? `if(c._arf)return fre(rt,c,${arg})\n${deriveDisposeOnly ? `if(c['~dispose'])` : ""}c._arf=true\nconst _fr=fre(rt,c,${arg})\nreturn typeof _fr?.then==='function'?_fr.then((_v)=>{${catchSchedule}return _v\n}):(${catchSchedule.trim()},_fr)\n` : `return fre(rt,c,${arg})\n`;
 	const mapThenSchedule = (s, onReject) => `if(typeof _m?.then==='function')return Promise.resolve(_m).then((_v)=>{\n${s}return _v\n},${onReject})\n` + s + `return _m\n`;
-	const signPrefix = syncCookieSign ? `scv(c.set.cookie,cc)\n` : asyncCookieSign ? `_sg=scv(c.set.cookie,cc)\nif(_sg){${arm ? `${arm}\n` : ""}await _sg}\n` : "";
+	const signPrefix = syncCookieSign ? `scv(c.set.cookie,cc,c.set)\n` : asyncCookieSign ? `_sg=scv(c.set.cookie,cc,c.set)\nif(_sg){${arm ? `${arm}\n` : ""}await _sg}\n` : "";
 	if (syncCookieSign || asyncCookieSign) link(require_cookie_utils.signCookieValues, "scv");
+	const streamSign = signPrefix ? `if(typeof _r?.next==='function')c.set['~sign']=_sgn\n` : "";
 	let factoryHelpers = "";
 	let tailCatch = -1;
 	if (hasBeforeHandle || hasAfterHandle || hasMapResponse || hasAfterResponse || hasResponseValidator || hasCookieSign || hasTrace) {
@@ -443,11 +462,11 @@ function compileHandlerJit(options, tailPass) {
 		} else if (isHandleFunction && require_compile_handler_utils.asyncTail(asyncMode)) code += require_compile_handler_utils.tailStage(require_compile_handler_utils.asyncTail(asyncMode), hasBeforeHandle ? "_r===undefined" : void 0) + callHandler() + teeBlock + "}\n";
 		else if (hasBeforeHandle) code += plain(`if(_r===undefined){\n${callHandler()}${teeBlock}}\n`);
 		else code += plain(callHandler() + teeBlock);
-		if (teeBlock) discard = "_sv?.return()\n";
+		if (teeBlock) abortDiscard = "_sv?.return()\n";
 		code += abortCheck();
 		if (syncAfterResponse) {
 			if (!errorClasses) link(require_handler_utils.forwardError, "fe");
-			factoryHelpers += syncScheduleDecl + `function _fin(c,_r){\n` + rethrow("_r") + "if(_r&&(_r[Symbol.iterator]||_r[Symbol.asyncIterator])&&typeof _r.next==='function'){\nconst _s=tee(_r,2)\nreturn _fin2(c,_s[0],_s[1],_s[0])\n}\nif(_r instanceof ReadableStream){const _o=obs(_r)\nif(_o)return _fin2(c,_o[0],_o[1],_o[2])}\nreturn _fin2(c,_r)\n}\nfunction _fin2(c,_r,_stl,_sv){try{\nc.responseValue=_r\n" + signPrefix + `const _m=${mapValue("_r")}\n` + mapThenSchedule(syncScheduleDecl ? `_scf(c,_stl)\n` : scheduleAfterResponse, `(_e)=>{_sv?.return()\n${freThenSchedule("_e")}}`) + `}catch(_e){_sv?.return();throw _e}}\n`;
+			factoryHelpers += syncScheduleDecl + `function _fin(c,_r){\n` + rethrow("_r") + "if(_r&&(_r[Symbol.iterator]||_r[Symbol.asyncIterator])&&typeof _r.next==='function'){\nconst _s=tee(_r,2)\nreturn _fin2(c,_s[0],_s[1],_s[0])\n}\nif(_r instanceof ReadableStream){const _o=obs(_r)\nif(_o)return _fin2(c,_o[0],_o[1],_o[2])}\nreturn _fin2(c,_r)\n}\nfunction _fin2(c,_r,_stl,_sv){try{\nc.responseValue=_r\n" + signPrefix + streamSign + `const _m=${mapValue("_r")}\n` + mapThenSchedule(syncScheduleDecl ? `_scf(c,_stl)\n` : scheduleAfterResponse, `(_e)=>{_sv?.return()\n${freThenSchedule("_e")}}`) + `}catch(_e){_sv?.return();throw _e}}\n`;
 			code += `if(typeof _r?.then==='function')return Promise.resolve(_r).then(${fwd}).then((_v)=>_fin(c,_v)).catch((_e)=>{${freThenSchedule("_e")}})\nreturn _fin(c,_r)\n`;
 		} else {
 			code += plain(rethrow("_r"));
@@ -477,17 +496,24 @@ function compileHandlerJit(options, tailPass) {
 				if (hasMapResponse) code += abortCheck();
 			}
 			if (hasResponseValidator) {
-				link(vali, "va");
 				link(require_error.ElysiaStatus, "es");
+				link(require_constants.StatusMap, "sm");
+				let pick = "";
+				let i = 0;
+				for (const [status, v] of vali.response) {
+					link(status, `vs${i}`);
+					link(v, `vr${i}`);
+					pick += `_st==vs${i}?vr${i}:`;
+					i++;
+				}
+				pick += "undefined";
 				const encodeStatus = responseValiAsync ? `(_vr.mayReturnPromise?_vr.From(_r.response,'response',true):_vr.EncodeFrom(_r.response,'response'))` : `_vr.EncodeFrom(_r.response,'response')`;
 				const encodeBody = responseValiAsync ? `(_vr.mayReturnPromise?_vr.From(_r,'response',true):_vr.EncodeFrom(_r,'response'))` : `_vr.EncodeFrom(_r,'response')`;
-				code += plain(`if(_r instanceof es){
-const _vr=va.response[_r.status]
-if(_vr)_r.response=${responseValiAsync ? awaitValue(encodeStatus, arm) : encodeStatus}\n}else if(!(_r instanceof Response)&&!(_r instanceof ReadableStream)&&typeof _r?.next!=='function'){\nconst _vr=va.response[c.set.status??200]\nif(_vr)_r=${responseValiAsync ? awaitValue(encodeBody, arm) : encodeBody}\n}\n`);
+				code += plain(`if(_r instanceof es){\nconst _st=_r.status,_vr=${pick}\nif(_vr)_r.response=${responseValiAsync ? awaitValue(encodeStatus, arm) : encodeStatus}\n}else if(!(_r instanceof Response)&&!(_r instanceof ReadableStream)&&typeof _r?.next!=='function'){\nconst _sr=c.set.status??200,_st=typeof _sr==='string'?(sm[_sr]??_sr):_sr,_vr=${pick}\nif(_vr)_r=${responseValiAsync ? awaitValue(encodeBody, arm) : encodeBody}\n}\n`);
 				code += abortCheck();
 			}
 			const deferSchedule = !!schedule;
-			code += plain(signPrefix);
+			code += plain(signPrefix + streamSign);
 			const finalMap = mapValue("_r");
 			const onMapReject = syncErrorHook ? `(_e)=>_ce(_e,c)` : dedupSchedule ? `(_e)=>{${freThenSchedule("_e")}}` : `(_e)=>fre(rt,c,_e)`;
 			if (tail) code += `let _m\n${stageSite(finalMap, "_m")}` + (deferSchedule ? schedule : "") + `return _m\n`;
@@ -558,7 +584,7 @@ if(_vr)_r.response=${responseValiAsync ? awaitValue(encodeStatus, arm) : encodeS
 	}
 	if (syncCookieSign || asyncCookieSign) {
 		code = code.replaceAll("fre(rt,c,", "_sfre(rt,c,");
-		factoryHelpers = factoryHelpers.replaceAll("fre(rt,c,", "_sfre(rt,c,") + `function _sgn(s){return scv(s.cookie,cc)}\nfunction _sfre(rt,c,e){return fre(rt,c,e,_sgn)}\n`;
+		factoryHelpers = factoryHelpers.replaceAll("fre(rt,c,", "_sfre(rt,c,") + `function _sgn(s){return scv(s.cookie,cc,s)}\nfunction _sfre(rt,c,e){return fre(rt,c,e,_sgn)}\n`;
 	}
 	if (factoryHelpers) code = `(function(){\n${factoryHelpers}return ${code}})()`;
 	const alias = aliasKeys.join(",");
@@ -574,33 +600,24 @@ if(_vr)_r.response=${responseValiAsync ? awaitValue(encodeStatus, arm) : encodeS
 	const isGeneratorHandler = isHandleFunction && handler.constructor.name.endsWith("GeneratorFunction");
 	if (!hasTrace && isHandleFunction && !isGeneratorHandler && !inlineUnsafe) {
 		if (alias === "rc" || !isAsync && !syncErrorHook && alias === "rc,fe") return createInlineHandler(responseCompact, handler);
-		else if (alias === "rm" || alias === "msh,rm" || !isAsync && !syncErrorHook && (alias === "rm,fe" || alias === "msh,rm,fe")) {
-			const rmap = responseMap;
-			const h = handler;
-			let inlineHandler;
-			if (responseMode === "set-with-default-headers" && inference.set) inlineHandler = ((c) => {
-				require_adapter_utils.materializeSetHeaders(c.set);
-				const r = h(c);
-				if (r instanceof Error) throw r;
-				if (typeof r?.then === "function") return Promise.resolve(r).then((v) => rmap(require_handler_utils.forwardError(v), c.set, c.request, true));
-				return rmap(r, c.set, c.request, true);
-			});
-			else inlineHandler = ((c) => {
-				const r = h(c);
-				if (r instanceof Error) throw r;
-				if (typeof r?.then === "function") return Promise.resolve(r).then((v) => rmap(require_handler_utils.forwardError(v), c.set, c.request, true));
-				return rmap(r, c.set, c.request, true);
-			});
-			return inlineHandler;
-		}
+		else if (alias === "rm" || alias === "msh,rm" || !isAsync && !syncErrorHook && (alias === "rm,fe" || alias === "msh,rm,fe")) return responseMode === "set-with-default-headers" && inference.set ? createInlineDefaultHeaderHandler(responseMap, handler) : createInlineSetHandler(responseMap, handler);
 	}
 	require_compile_jit_probe.JITProbe.record("handler:new-function");
-	return new Function("h", fullAlias, `return ${code}`)(handler, ...paramValues);
+	const key = fullAlias + "\n" + code;
+	let factory = factoryCache.get(key);
+	if (!factory) {
+		if (factoryCache.size >= 256) require_utils.evictOldestHalf(factoryCache);
+		factory = new Function("h", fullAlias, `return ${code}`);
+		factoryCache.set(key, factory);
+	}
+	return factory(handler, ...paramValues);
 }
 
 //#endregion
+exports.FACTORY_CACHE_LIMIT = FACTORY_CACHE_LIMIT;
 exports.compileHandlerJit = compileHandlerJit;
 exports.createInlineHandler = createInlineHandler;
+exports.factoryCache = factoryCache;
 exports.setAsyncTail = setAsyncTail;
 exports.setCaptureHeaderShorthand = setCaptureHeaderShorthand;
 exports.setOnEmit = setOnEmit;

package/dist/compile/handler/jit.mjs

@@ -1,6 +1,7 @@
+// @bun
 import { hasHeaderShorthand, isBun } from "../../universal/constants.mjs";
-import { isDynamicRegex, traceEventIndex } from "../../constants.mjs";
-import { requestId } from "../../utils.mjs";
+import { StatusMap, isDynamicRegex, traceEventIndex } from "../../constants.mjs";
+import { evictOldestHalf, requestId } from "../../utils.mjs";
 import { isProduction } from "../../universal/is-production.mjs";
 import { ELYSIA_TYPES } from "../../type/constants.mjs";
 import { ElysiaStatus, ParseError, ValidationError } from "../../error.mjs";
@@ -9,27 +10,32 @@ import { parseQueryFromURL } from "../../parse-query.mjs";
 import { Capture } from "../aot.mjs";
 import { resolvedTraceOf, traceCapabilityRequired } from "../../generation.mjs";
 import { buildCookieJar, parseCookieRaw, parseCookieRawDeferred, parseCookieRawLazy, parseCookieRawSigned, parseCookieRawSync, signCookieValues } from "../../cookie/utils.mjs";
-import { drainDisposables, emptyResponse, finalizeRouteError, forwardError, forwardErrorOf } from "../../handler/utils.mjs";
+import { drainDisposables, emptyResponse, finalizeRouteErrorOf, forwardError, forwardErrorOf } from "../../handler/utils.mjs";
 import { fallbackResponse } from "../../handler/error.mjs";
 import { armEntryAbort, asyncTail, awaitGuard, awaitSite, cloneResponse, cloneStaticValue, getQueryParseChannels, hasRequestBody, mapAfterResponse, mapBeforeHandle, mapChainHook, mapError, mapTransform, runBeforeHandlePrefix, runBeforeHandlePrefixAsync, tailPlain, tailStage } from "./utils.mjs";
 import { JITProbe } from "../jit-probe.mjs";
 
 //#region src/compile/handler/jit.ts
 const awaitValue = (value, arm = "") => arm ? `await (_av=(${value}),${arm},_av)` : `await ${value}`;
+/**
+* @internal one factory per emitted source, each route calls it for its own
+* closure. JSC stops sharing code between identical `new Function` sources
+* once its source cache churns, so without this every route links a copy
+*/
+const factoryCache = /* @__PURE__ */ new Map();
+const FACTORY_CACHE_LIMIT = 256;
 let captureHeaderShorthand;
 /**
-* @internal test hook: receives each route's emitted source. A sync-first
-* route's async tail is a helper beside it, so `toString()` of the compiled
-* handler does not show the whole pipeline
+* @internal test hook: receives each route's full emitted source, including
+* the async tail helper that `toString()` of the handler does not show
 */
 let onEmit;
 const setOnEmit = (fn) => {
 	onEmit = fn;
 };
 /**
-* @internal test hook: `false` compiles sync-first routes on the `async`
-* lane instead, the reference the async tail is differentially tested
-* against. `undefined` restores the default (see `tail` below)
+* @internal test hook: `false` compiles sync-first routes on the plain
+* `async` lane (the differential reference), `undefined` restores the default
 */
 let asyncTailOn;
 const setAsyncTail = (on) => {
@@ -160,9 +166,22 @@ const createInlineHandler = (map, h) => ((c) => {
 	if (typeof r?.then === "function") return Promise.resolve(r).then((v) => map(forwardError(v), c.request, true));
 	return map(r, c.request, true);
 });
+const createInlineSetHandler = (map, h) => ((c) => {
+	const r = h(c);
+	if (r instanceof Error) throw r;
+	if (typeof r?.then === "function") return Promise.resolve(r).then((v) => map(forwardError(v), c.set, c.request, true));
+	return map(r, c.set, c.request, true);
+});
+const createInlineDefaultHeaderHandler = (map, h) => ((c) => {
+	materializeSetHeaders(c.set);
+	const r = h(c);
+	if (r instanceof Error) throw r;
+	if (typeof r?.then === "function") return Promise.resolve(r).then((v) => map(forwardError(v), c.set, c.request, true));
+	return map(r, c.set, c.request, true);
+});
 function compileHandlerJit(options, tailPass) {
 	const { method, path, handler, root, errorRoot, hook, adapter, isHandleFunction, isStaticResponse, isPromiseHandler, errorClasses, state, shape } = options;
-	const { vali, inference, cookieConfig, beforeHandlePrefix, traceHandlers, tracePhases, hasAnyPhase, traceHandleOn, descriptor: { async: isAsync, tail: tailRoute, responseMode, hasBody, bodyValiIsAsync, headersValiIsAsync, paramsValiIsAsync, queryValiIsAsync, cookieValiIsAsync: cookieValidIsAsync, responseValiAsync, hasCookieSign, syncCookieSign, asyncCookieSign, lazyCookieVerify, hasErrorHook, hasAfterResponse, hasBeforeHandle, hasAfterHandle, hasMapResponse, hasResponseValidator, hasTrace, traceCount, hasLifecycleHook, syncErrorHook, syncAfterResponse } } = state;
+	const { vali, inference, cookieConfig, beforeHandlePrefix, traceHandlers, tracePhases, hasAnyPhase, traceHandleOn, descriptor: { async: isAsync, tail: tailRoute, responseMode, hasBody, bodyValiIsAsync, headersValiIsAsync, paramsValiIsAsync, queryValiIsAsync, cookieValiIsAsync, responseValiAsync, hasCookieSign, syncCookieSign, asyncCookieSign, lazyCookieVerify, hasErrorHook, hasAfterResponse, hasBeforeHandle, hasAfterHandle, hasMapResponse, hasResponseValidator, hasTrace, traceCount, hasLifecycleHook, syncErrorHook, syncAfterResponse } } = state;
 	const hasStaticAfterResponse = !!hook?.afterResponse?.length;
 	const hasDeriveDispose = !!hook?.["~deriveEntries"]?.length;
 	const hasDynamicAfterResponse = !!inference.afterResponse || hasDeriveDispose;
@@ -176,7 +195,7 @@ function compileHandlerJit(options, tailPass) {
 	} : isAsync;
 	const plain = (code) => tailPlain(asyncMode, code);
 	const seenKeys = new Set(["rt", "fre"]);
-	const paramValues = [errorRoot, finalizeRouteError];
+	const paramValues = [errorRoot, finalizeRouteErrorOf(hook)];
 	const aliasKeys = [];
 	function link(v, key) {
 		if (!seenKeys.has(key)) {
@@ -192,8 +211,8 @@ function compileHandlerJit(options, tailPass) {
 	const abortPeek = "c['~sig']?.aborted";
 	const arm = abortOn ? "_as??=(c['~sig']??=c.request.signal)" : "";
 	let abortSchedule = "";
-	let discard = "";
-	const abortCheck = () => abortOn ? plain(abortSchedule || discard ? `if(${abortPeek}){${discard}${abortSchedule}return emp.clone()}\n` : `if(${abortPeek})return emp.clone()\n`) : "";
+	let abortDiscard = "";
+	const abortCheck = () => abortOn ? plain(abortSchedule || abortDiscard ? `if(${abortPeek}){${abortDiscard}${abortSchedule}return emp.clone()}\n` : `if(${abortPeek})return emp.clone()\n`) : "";
 	const abortChainGuard = () => abortOn ? abortPeek : void 0;
 	const abortCatch = abortOn ? `if((${arm}).aborted)return emp.clone()\n` : "";
 	const phaseOn = (phase) => hasTrace && (tracePhases === null || tracePhases.has(phase));
@@ -330,7 +349,7 @@ function compileHandlerJit(options, tailPass) {
 			code += `c.cookie=bcj(c.set,_ck,cc,1)\n`;
 		} else {
 			let deferDecode = false;
-			if (!hasCookieSign && !cookieValidIsAsync) if (!vali?.cookie) {
+			if (!hasCookieSign && !cookieValiIsAsync) if (!vali?.cookie) {
 				link(parseCookieRawDeferred, "pcrd");
 				code += `let _ck=pcrd(${cookieHeaderExpr},cc)\n`;
 				deferDecode = true;
@@ -338,7 +357,7 @@ function compileHandlerJit(options, tailPass) {
 				link(parseCookieRawSync, "pcrs");
 				code += `let _ck=pcrs(${cookieHeaderExpr},cc)\n`;
 			}
-			else if (syncCookieSign && !cookieValidIsAsync) {
+			else if (syncCookieSign && !cookieValiIsAsync) {
 				link(parseCookieRawSigned, "pcrsg");
 				code += `let _ck=pcrsg(${cookieHeaderExpr},cc)\n`;
 			} else {
@@ -348,8 +367,8 @@ function compileHandlerJit(options, tailPass) {
 			if (vali?.cookie) {
 				link(vali, "va");
 				const cookieIsOptional = !!(hook?.cookie)?.["~optional"];
-				const value = `va.cookie.From(_ck,${fromArgs("cookie", !!cookieValidIsAsync)})`;
-				const validateExpr = `_ck=${cookieValidIsAsync ? awaitValue(value, arm) : value}\n`;
+				const value = `va.cookie.From(_ck,${fromArgs("cookie", cookieValiIsAsync)})`;
+				const validateExpr = `_ck=${cookieValiIsAsync ? awaitValue(value, arm) : value}\n`;
 				if (cookieIsOptional) code += `if(Object.keys(_ck).length){${validateExpr}}\n`;
 				else code += validateExpr;
 			}
@@ -387,8 +406,9 @@ function compileHandlerJit(options, tailPass) {
 	abortSchedule = hasDeriveDispose ? syncAfterResponse ? catchSchedule : schedule : "";
 	const freThenSchedule = (arg) => catchSchedule ? `if(c._arf)return fre(rt,c,${arg})\n${deriveDisposeOnly ? `if(c['~dispose'])` : ""}c._arf=true\nconst _fr=fre(rt,c,${arg})\nreturn typeof _fr?.then==='function'?_fr.then((_v)=>{${catchSchedule}return _v\n}):(${catchSchedule.trim()},_fr)\n` : `return fre(rt,c,${arg})\n`;
 	const mapThenSchedule = (s, onReject) => `if(typeof _m?.then==='function')return Promise.resolve(_m).then((_v)=>{\n${s}return _v\n},${onReject})\n` + s + `return _m\n`;
-	const signPrefix = syncCookieSign ? `scv(c.set.cookie,cc)\n` : asyncCookieSign ? `_sg=scv(c.set.cookie,cc)\nif(_sg){${arm ? `${arm}\n` : ""}await _sg}\n` : "";
+	const signPrefix = syncCookieSign ? `scv(c.set.cookie,cc,c.set)\n` : asyncCookieSign ? `_sg=scv(c.set.cookie,cc,c.set)\nif(_sg){${arm ? `${arm}\n` : ""}await _sg}\n` : "";
 	if (syncCookieSign || asyncCookieSign) link(signCookieValues, "scv");
+	const streamSign = signPrefix ? `if(typeof _r?.next==='function')c.set['~sign']=_sgn\n` : "";
 	let factoryHelpers = "";
 	let tailCatch = -1;
 	if (hasBeforeHandle || hasAfterHandle || hasMapResponse || hasAfterResponse || hasResponseValidator || hasCookieSign || hasTrace) {
@@ -442,11 +462,11 @@ function compileHandlerJit(options, tailPass) {
 		} else if (isHandleFunction && asyncTail(asyncMode)) code += tailStage(asyncTail(asyncMode), hasBeforeHandle ? "_r===undefined" : void 0) + callHandler() + teeBlock + "}\n";
 		else if (hasBeforeHandle) code += plain(`if(_r===undefined){\n${callHandler()}${teeBlock}}\n`);
 		else code += plain(callHandler() + teeBlock);
-		if (teeBlock) discard = "_sv?.return()\n";
+		if (teeBlock) abortDiscard = "_sv?.return()\n";
 		code += abortCheck();
 		if (syncAfterResponse) {
 			if (!errorClasses) link(forwardError, "fe");
-			factoryHelpers += syncScheduleDecl + `function _fin(c,_r){\n` + rethrow("_r") + "if(_r&&(_r[Symbol.iterator]||_r[Symbol.asyncIterator])&&typeof _r.next==='function'){\nconst _s=tee(_r,2)\nreturn _fin2(c,_s[0],_s[1],_s[0])\n}\nif(_r instanceof ReadableStream){const _o=obs(_r)\nif(_o)return _fin2(c,_o[0],_o[1],_o[2])}\nreturn _fin2(c,_r)\n}\nfunction _fin2(c,_r,_stl,_sv){try{\nc.responseValue=_r\n" + signPrefix + `const _m=${mapValue("_r")}\n` + mapThenSchedule(syncScheduleDecl ? `_scf(c,_stl)\n` : scheduleAfterResponse, `(_e)=>{_sv?.return()\n${freThenSchedule("_e")}}`) + `}catch(_e){_sv?.return();throw _e}}\n`;
+			factoryHelpers += syncScheduleDecl + `function _fin(c,_r){\n` + rethrow("_r") + "if(_r&&(_r[Symbol.iterator]||_r[Symbol.asyncIterator])&&typeof _r.next==='function'){\nconst _s=tee(_r,2)\nreturn _fin2(c,_s[0],_s[1],_s[0])\n}\nif(_r instanceof ReadableStream){const _o=obs(_r)\nif(_o)return _fin2(c,_o[0],_o[1],_o[2])}\nreturn _fin2(c,_r)\n}\nfunction _fin2(c,_r,_stl,_sv){try{\nc.responseValue=_r\n" + signPrefix + streamSign + `const _m=${mapValue("_r")}\n` + mapThenSchedule(syncScheduleDecl ? `_scf(c,_stl)\n` : scheduleAfterResponse, `(_e)=>{_sv?.return()\n${freThenSchedule("_e")}}`) + `}catch(_e){_sv?.return();throw _e}}\n`;
 			code += `if(typeof _r?.then==='function')return Promise.resolve(_r).then(${fwd}).then((_v)=>_fin(c,_v)).catch((_e)=>{${freThenSchedule("_e")}})\nreturn _fin(c,_r)\n`;
 		} else {
 			code += plain(rethrow("_r"));
@@ -476,17 +496,24 @@ function compileHandlerJit(options, tailPass) {
 				if (hasMapResponse) code += abortCheck();
 			}
 			if (hasResponseValidator) {
-				link(vali, "va");
 				link(ElysiaStatus, "es");
+				link(StatusMap, "sm");
+				let pick = "";
+				let i = 0;
+				for (const [status, v] of vali.response) {
+					link(status, `vs${i}`);
+					link(v, `vr${i}`);
+					pick += `_st==vs${i}?vr${i}:`;
+					i++;
+				}
+				pick += "undefined";
 				const encodeStatus = responseValiAsync ? `(_vr.mayReturnPromise?_vr.From(_r.response,'response',true):_vr.EncodeFrom(_r.response,'response'))` : `_vr.EncodeFrom(_r.response,'response')`;
 				const encodeBody = responseValiAsync ? `(_vr.mayReturnPromise?_vr.From(_r,'response',true):_vr.EncodeFrom(_r,'response'))` : `_vr.EncodeFrom(_r,'response')`;
-				code += plain(`if(_r instanceof es){
-const _vr=va.response[_r.status]
-if(_vr)_r.response=${responseValiAsync ? awaitValue(encodeStatus, arm) : encodeStatus}\n}else if(!(_r instanceof Response)&&!(_r instanceof ReadableStream)&&typeof _r?.next!=='function'){\nconst _vr=va.response[c.set.status??200]\nif(_vr)_r=${responseValiAsync ? awaitValue(encodeBody, arm) : encodeBody}\n}\n`);
+				code += plain(`if(_r instanceof es){\nconst _st=_r.status,_vr=${pick}\nif(_vr)_r.response=${responseValiAsync ? awaitValue(encodeStatus, arm) : encodeStatus}\n}else if(!(_r instanceof Response)&&!(_r instanceof ReadableStream)&&typeof _r?.next!=='function'){\nconst _sr=c.set.status??200,_st=typeof _sr==='string'?(sm[_sr]??_sr):_sr,_vr=${pick}\nif(_vr)_r=${responseValiAsync ? awaitValue(encodeBody, arm) : encodeBody}\n}\n`);
 				code += abortCheck();
 			}
 			const deferSchedule = !!schedule;
-			code += plain(signPrefix);
+			code += plain(signPrefix + streamSign);
 			const finalMap = mapValue("_r");
 			const onMapReject = syncErrorHook ? `(_e)=>_ce(_e,c)` : dedupSchedule ? `(_e)=>{${freThenSchedule("_e")}}` : `(_e)=>fre(rt,c,_e)`;
 			if (tail) code += `let _m\n${stageSite(finalMap, "_m")}` + (deferSchedule ? schedule : "") + `return _m\n`;
@@ -557,7 +584,7 @@ if(_vr)_r.response=${responseValiAsync ? awaitValue(encodeStatus, arm) : encodeS
 	}
 	if (syncCookieSign || asyncCookieSign) {
 		code = code.replaceAll("fre(rt,c,", "_sfre(rt,c,");
-		factoryHelpers = factoryHelpers.replaceAll("fre(rt,c,", "_sfre(rt,c,") + `function _sgn(s){return scv(s.cookie,cc)}\nfunction _sfre(rt,c,e){return fre(rt,c,e,_sgn)}\n`;
+		factoryHelpers = factoryHelpers.replaceAll("fre(rt,c,", "_sfre(rt,c,") + `function _sgn(s){return scv(s.cookie,cc,s)}\nfunction _sfre(rt,c,e){return fre(rt,c,e,_sgn)}\n`;
 	}
 	if (factoryHelpers) code = `(function(){\n${factoryHelpers}return ${code}})()`;
 	const alias = aliasKeys.join(",");
@@ -573,29 +600,18 @@ if(_vr)_r.response=${responseValiAsync ? awaitValue(encodeStatus, arm) : encodeS
 	const isGeneratorHandler = isHandleFunction && handler.constructor.name.endsWith("GeneratorFunction");
 	if (!hasTrace && isHandleFunction && !isGeneratorHandler && !inlineUnsafe) {
 		if (alias === "rc" || !isAsync && !syncErrorHook && alias === "rc,fe") return createInlineHandler(responseCompact, handler);
-		else if (alias === "rm" || alias === "msh,rm" || !isAsync && !syncErrorHook && (alias === "rm,fe" || alias === "msh,rm,fe")) {
-			const rmap = responseMap;
-			const h = handler;
-			let inlineHandler;
-			if (responseMode === "set-with-default-headers" && inference.set) inlineHandler = ((c) => {
-				materializeSetHeaders(c.set);
-				const r = h(c);
-				if (r instanceof Error) throw r;
-				if (typeof r?.then === "function") return Promise.resolve(r).then((v) => rmap(forwardError(v), c.set, c.request, true));
-				return rmap(r, c.set, c.request, true);
-			});
-			else inlineHandler = ((c) => {
-				const r = h(c);
-				if (r instanceof Error) throw r;
-				if (typeof r?.then === "function") return Promise.resolve(r).then((v) => rmap(forwardError(v), c.set, c.request, true));
-				return rmap(r, c.set, c.request, true);
-			});
-			return inlineHandler;
-		}
+		else if (alias === "rm" || alias === "msh,rm" || !isAsync && !syncErrorHook && (alias === "rm,fe" || alias === "msh,rm,fe")) return responseMode === "set-with-default-headers" && inference.set ? createInlineDefaultHeaderHandler(responseMap, handler) : createInlineSetHandler(responseMap, handler);
 	}
 	JITProbe.record("handler:new-function");
-	return new Function("h", fullAlias, `return ${code}`)(handler, ...paramValues);
+	const key = fullAlias + "\n" + code;
+	let factory = factoryCache.get(key);
+	if (!factory) {
+		if (factoryCache.size >= 256) evictOldestHalf(factoryCache);
+		factory = new Function("h", fullAlias, `return ${code}`);
+		factoryCache.set(key, factory);
+	}
+	return factory(handler, ...paramValues);
 }
 
 //#endregion
-export { compileHandlerJit, createInlineHandler, setAsyncTail, setCaptureHeaderShorthand, setOnEmit };
+export { FACTORY_CACHE_LIMIT, compileHandlerJit, createInlineHandler, factoryCache, setAsyncTail, setCaptureHeaderShorthand, setOnEmit };

package/dist/compile/handler/params.d.ts

@@ -20,6 +20,10 @@ interface HandlerParamContext {
   cookieConfig: unknown;
   tracers: unknown;
 }
+/**
+ * `vs<i>` / `vr<i>`: status and validator of the route's i-th declared response
+ */
+declare const isResponseParam: (name: string) => boolean;
 declare function resolveHandlerParams(names: string[], c: HandlerParamContext): unknown[];
 //#endregion
-export { resolveHandlerParams };
+export { isResponseParam, resolveHandlerParams };

package/dist/compile/handler/params.js

@@ -1,4 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_constants = require('../../constants.js');
 const require_utils = require('../../utils.js');
 const require_error = require('../../error.js');
 const require_adapter_utils = require('../../adapter/utils.js');
@@ -28,6 +29,7 @@ const handlerParams = () => _handlerParams ??= {
 	pq: () => require_parse_query.parseQueryFromURL,
 	pe: () => require_error.ParseError,
 	es: () => require_error.ElysiaStatus,
+	sm: () => require_constants.StatusMap,
 	rdc: () => require_compile_handler_utils.replaceDeriveContext,
 	dsp: () => require_handler_utils.registerDeriveDisposable,
 	dds: () => require_handler_utils.drainDisposables,
@@ -49,10 +51,10 @@ const handlerParams = () => _handlerParams ??= {
 	va: (c) => c.vali,
 	fe: () => require_handler_utils.forwardError,
 	ie: (c) => {
-		const classes = require_handler_utils.returnedErrorClasses(c.hook, c.root);
+		const classes = require_handler_utils.returnedErrorClasses(c.hook);
 		return classes ? require_handler_utils.forwardErrorOf(classes) : require_handler_utils.forwardError;
 	},
-	fre: () => require_handler_utils.finalizeRouteError,
+	fre: (c) => require_handler_utils.finalizeRouteErrorOf(c.hook),
 	fbr: () => require_handler_error.fallbackResponse,
 	aet: () => require_handler_error.adoptErrorType,
 	rt: (c) => c.root,
@@ -66,18 +68,25 @@ const handlerParams = () => _handlerParams ??= {
 	tr: (c) => c.tracers,
 	cc: (c) => c.cookieConfig
 };
+/**
+* `vs<i>` / `vr<i>`: status and validator of the route's i-th declared response
+*/
+const isResponseParam = (name) => /^v[rs]\d+$/.test(name);
 function resolveHandlerParams(names, c) {
 	const length = names.length;
 	if (!length) return [];
 	const out = new Array(length);
 	const params = handlerParams();
 	for (let i = 0; i < length; i++) {
-		const resolve = params[names[i]];
-		if (!resolve) throw new Error(`[elysia-aot]: Fail to reconstruct build, missing "${names[i]}" param`);
-		out[i] = resolve(c);
+		const name = names[i];
+		const entry = isResponseParam(name) && [...c.vali?.response ?? []][+name.slice(2)];
+		const resolve = params[name];
+		if (!entry && !resolve) throw new Error(`[elysia-aot]: Fail to reconstruct build, missing "${name}" param`);
+		out[i] = entry ? entry[name[1] === "s" ? 0 : 1] : resolve(c);
 	}
 	return out;
 }
 
 //#endregion
+exports.isResponseParam = isResponseParam;
 exports.resolveHandlerParams = resolveHandlerParams;

package/dist/compile/handler/params.mjs

@@ -1,9 +1,11 @@
+// @bun
+import { StatusMap } from "../../constants.mjs";
 import { requestId } from "../../utils.mjs";
 import { ElysiaStatus, ParseError, ValidationError } from "../../error.mjs";
 import { materializeSetHeaders, normalizeContentType, observeStream, tee } from "../../adapter/utils.mjs";
 import { parseQueryFromURL } from "../../parse-query.mjs";
 import { buildCookieJar, parseCookieRaw, parseCookieRawDeferred, parseCookieRawLazy, parseCookieRawSigned, parseCookieRawSync, signCookieValues } from "../../cookie/utils.mjs";
-import { drainDisposables, emptyResponse, finalizeRouteError, forwardError, forwardErrorOf, registerDeriveDisposable, returnedErrorClasses } from "../../handler/utils.mjs";
+import { drainDisposables, emptyResponse, finalizeRouteErrorOf, forwardError, forwardErrorOf, registerDeriveDisposable, returnedErrorClasses } from "../../handler/utils.mjs";
 import { adoptErrorType, fallbackResponse } from "../../handler/error.mjs";
 import { armEntryAbort, cloneResponse, getQueryParseChannels, hasRequestBody, replaceDeriveContext } from "./utils.mjs";
 import { staticCloneResolver } from "./static-clone-resolver.mjs";
@@ -27,6 +29,7 @@ const handlerParams = () => _handlerParams ??= {
 	pq: () => parseQueryFromURL,
 	pe: () => ParseError,
 	es: () => ElysiaStatus,
+	sm: () => StatusMap,
 	rdc: () => replaceDeriveContext,
 	dsp: () => registerDeriveDisposable,
 	dds: () => drainDisposables,
@@ -48,10 +51,10 @@ const handlerParams = () => _handlerParams ??= {
 	va: (c) => c.vali,
 	fe: () => forwardError,
 	ie: (c) => {
-		const classes = returnedErrorClasses(c.hook, c.root);
+		const classes = returnedErrorClasses(c.hook);
 		return classes ? forwardErrorOf(classes) : forwardError;
 	},
-	fre: () => finalizeRouteError,
+	fre: (c) => finalizeRouteErrorOf(c.hook),
 	fbr: () => fallbackResponse,
 	aet: () => adoptErrorType,
 	rt: (c) => c.root,
@@ -65,18 +68,24 @@ const handlerParams = () => _handlerParams ??= {
 	tr: (c) => c.tracers,
 	cc: (c) => c.cookieConfig
 };
+/**
+* `vs<i>` / `vr<i>`: status and validator of the route's i-th declared response
+*/
+const isResponseParam = (name) => /^v[rs]\d+$/.test(name);
 function resolveHandlerParams(names, c) {
 	const length = names.length;
 	if (!length) return [];
 	const out = new Array(length);
 	const params = handlerParams();
 	for (let i = 0; i < length; i++) {
-		const resolve = params[names[i]];
-		if (!resolve) throw new Error(`[elysia-aot]: Fail to reconstruct build, missing "${names[i]}" param`);
-		out[i] = resolve(c);
+		const name = names[i];
+		const entry = isResponseParam(name) && [...c.vali?.response ?? []][+name.slice(2)];
+		const resolve = params[name];
+		if (!entry && !resolve) throw new Error(`[elysia-aot]: Fail to reconstruct build, missing "${name}" param`);
+		out[i] = entry ? entry[name[1] === "s" ? 0 : 1] : resolve(c);
 	}
 	return out;
 }
 
 //#endregion
-export { resolveHandlerParams };
+export { isResponseParam, resolveHandlerParams };

package/dist/compile/handler/reconstruct.d.ts

@@ -3,10 +3,10 @@ import { AnyElysia } from "../../base.js";
 import { CompiledCookieConfig } from "../../cookie/config.js";
 
 //#region src/compile/handler/reconstruct.d.ts
-declare abstract class Reconstrct {
+declare abstract class Reconstruct {
   static validator(hook: AnyLocalHook, root: AnyElysia, method: HTTPMethod, path: string, liveOnly?: boolean): any;
   static cookie(hook: AnyLocalHook, root: AnyElysia): CompiledCookieConfig;
   static trace(hook: AnyLocalHook, root: AnyElysia): any;
 }
 //#endregion
-export { Reconstrct };
+export { Reconstruct };