🎉 feat: hook compression

Author: SaltyAom

CHANGELOG.md

@@ -1,6 +1,7 @@
 # 1.2.11
 Feature:
 - Compressed lifecycle event
+- Lazily build radix tree for dynamic router
 
 # 1.2.10 - 5 Jan 2025
 Feature:

bun.lockb

@@ -1,6 +1,6 @@
 import { Elysia, t } from '../src'
 import { req } from '../test/utils'
 
-const app = new Elysia().get('/', 'Static Content')
+const app = new Elysia().get('/', () => 'Static Content')
 
-console.log(app.router.static.http.static['/'])
+console.dir(app, { depth: 10 })

example/a.ts

@@ -1,18 +1,32 @@
 import { Elysia, t } from '../../src'
 
-const total = 10_000
+const total = 100
 
-const setup = () => new Elysia()
+const setup = (n = 0) => {
+	const app = new Elysia()
 
+	for(let i = 0; i < 10; i++) {
+		app.get(`/:a/${i + (n * total)}`, () => i)
+	}
+
+	return app
+}
+
+const app = new Elysia()
+
+const memory = process.memoryUsage().heapTotal / 1024 / 1024
 const t1 = performance.now()
-for (let i = 0; i < total; i++) setup()
 
+for (let i = 0; i < total; i++) app.use(setup(i))
+
+const memoryAfter = process.memoryUsage().heapTotal / 1024 / 1024
 const took = performance.now() - t1
 
 console.log(
 	Intl.NumberFormat().format(total),
-	'instances took',
+	'routes took',
 	+took.toFixed(4),
 	'ms'
 )
-console.log('Average', +(took / total).toFixed(4), 'ms / instance')
+console.log('Average', +(took / total).toFixed(4), 'ms / route')
+console.log(memoryAfter - memory, 'MB memory used')

example/stress/instance.ts

@@ -1,29 +1,26 @@
 import { Elysia } from '../../src'
 
-const app = new Elysia({ precompile: true })
+const total = 1000
 
-const total = 200
-const t = performance.now()
+{
+	console.log('Elysia')
 
-const memory = process.memoryUsage().heapTotal / 1024 / 1024
+	const app = new Elysia({ precompile: true })
+	const t = performance.now()
+	const memory = process.memoryUsage().heapTotal / 1024 / 1024
 
-for (let i = 0; i < total; i++) app.get(`/id/${i}`, () => 'hello')
+	for (let i = 0; i < total; i++) app.get(`/id/${i}`, () => 'hello')
 
-const memoryAfter = process.memoryUsage().heapTotal / 1024 / 1024
+	const memoryAfter = process.memoryUsage().heapTotal / 1024 / 1024
+	const took = performance.now() - t
 
-const took = performance.now() - t
+	console.log(
+		Intl.NumberFormat().format(total),
+		'routes took',
+		+took.toFixed(4),
+		'ms'
+	)
+	console.log('Average', +(took / total).toFixed(4), 'ms / route')
 
-console.log(
-	Intl.NumberFormat().format(total),
-	'routes took',
-	+took.toFixed(4),
-	'ms'
-)
-console.log('Average', +(took / total).toFixed(4), 'ms / route')
-
-console.log({ memory, memoryAfter })
-console.log(memoryAfter - memory, 'MB memory used')
-
-app.listen(3000)
-
-// console.log(app.router.history)
+	console.log(memoryAfter - memory, 'MB memory used')
+}

example/stress/multiple-routes.ts

@@ -1,7 +1,7 @@
 {
 	"name": "elysia",
 	"description": "Ergonomic Framework for Human",
-	"version": "1.2.10",
+	"version": "1.2.11-exp.1",
 	"author": {
 		"name": "saltyAom",
 		"url": "https://github.com/SaltyAom",
@@ -160,7 +160,7 @@
 	"dependencies": {
 		"@sinclair/typebox": "^0.34.13",
 		"cookie": "^1.0.2",
-		"memoirist": "^0.2.0",
+		"memoirist": "^0.3.0",
 		"openapi-types": "^12.1.3"
 	},
 	"devDependencies": {

package.json

@@ -2052,6 +2052,7 @@ export const composeGeneralHandler = (
 	{ asManifest = false }: { asManifest?: false } = {}
 ) => {
 	const adapter = app['~adapter'].composeGeneralHandler
+	app.router.http.build()
 	const error404 = adapter.error404(
 		!!app.event.request?.length,
 		!!app.event.error?.length

src/compose.ts

@@ -254,7 +254,7 @@ export default class Elysia<
 		http: new Memoirist<{
 			compile: Function
 			handler?: ComposedHandler
-		}>(),
+		}>({ lazy: true }),
 		ws: new Memoirist<{
 			compile: Function
 			handler?: ComposedHandler
@@ -3474,9 +3474,16 @@ export default class Elysia<
 						if (plugin.constructor.name === 'Elysia')
 							return this._use(plugin.default)
 
-						throw new Error(
-							'Invalid plugin type. Expected Elysia instance, function, or module with "default" as Elysia instance or function that returns Elysia instance.'
-						)
+						if (plugin.constructor.name === '_Elysia')
+							return this._use(plugin.default)
+
+						try {
+							return this._use(plugin.default)
+						} catch {
+							throw new Error(
+								'Invalid plugin type. Expected Elysia instance, function, or module with "default" as Elysia instance or function that returns Elysia instance.'
+							)
+						}
 					})
 					.then((x) => x.compile())
 			)

src/index.ts

@@ -96,15 +96,13 @@ export const mergeCookie = <const A extends Object, const B extends Object>(
 	a: A,
 	b: B
 ): A & B => {
-	// @ts-ignore
-	// eslint-disable-next-line @typescript-eslint/no-unused-vars
-	const { properties: _, ...target } = a ?? {}
+	const v = mergeDeep(Object.assign({}, a), b, {
+		skipKeys: ['properties']
+	}) as A & B
 
-	// @ts-ignore
-	// eslint-disable-next-line @typescript-eslint/no-unused-vars
-	const { properties: __, ...source } = b ?? {}
+	if ('properties' in v) delete v.properties
 
-	return mergeDeep(target, source) as A & B
+	return v
 }
 
 export const mergeObjectArray = <T extends HookContainer>(
@@ -231,7 +229,13 @@ export const mergeHook = (
 	// 		customBStore[union]
 	// 	)
 
-	const hook = Object.assign({}, a, b, {
+	// @ts-expect-error
+	const { resolve: resolveA, ...restA } = a ?? {}
+	const { resolve: resolveB, ...restB } = b ?? {}
+
+	return {
+		...restA,
+		...restB,
 		// Merge local hook first
 		// @ts-ignore
 		body: b?.body ?? a?.body,
@@ -261,12 +265,11 @@ export const mergeHook = (
 		transform: mergeObjectArray(a?.transform, b?.transform),
 		beforeHandle: mergeObjectArray(
 			mergeObjectArray(
-				// @ts-expect-error
-				fnToContainer(a?.resolve, 'resolve'),
+				fnToContainer(resolveA, 'resolve'),
 				a?.beforeHandle
 			),
 			mergeObjectArray(
-				fnToContainer(b?.resolve, 'resolve'),
+				fnToContainer(resolveB, 'resolve'),
 				b?.beforeHandle
 			)
 		),
@@ -278,11 +281,7 @@ export const mergeHook = (
 		) as any,
 		trace: mergeObjectArray(a?.trace, b?.trace) as any,
 		error: mergeObjectArray(a?.error, b?.error)
-	})
-
-	if (hook.resolve) delete hook.resolve
-
-	return hook
+	}
 }
 
 interface ReplaceSchemaTypeOptions {
@@ -969,7 +968,7 @@ export const mergeLifeCycle = (
 	b: Partial<LifeCycleStore | AnyLocalHook>,
 	checksum?: number
 ): LifeCycleStore => {
-	return Object.assign({}, a, b, {
+	return {
 		start: mergeObjectArray(
 			a.start,
 			injectChecksum(checksum, b?.start)
@@ -1025,7 +1024,7 @@ export const mergeLifeCycle = (
 			a.stop,
 			injectChecksum(checksum, b?.stop)
 		) as HookContainer<GracefulHandler<any>>[]
-	})
+	}
 }
 
 export const asHookType = (