perf: 重构优化路由性能，添加日志系统，优化大文件内存占用，加强输入验证及路径安全等
This commit is contained in:
meiseayoung
+157
-29
@@ -27,6 +27,8 @@ pub mut:
|
||||
context_router ContextRouter = ContextRouter{}
|
||||
context_hybrid_router ContextHybridRouter
|
||||
context_trie_router ContextTrieRouter
|
||||
fast_router FastRouter // 新增：快速路由器
|
||||
use_fast_router bool = true // 新增：是否使用快速路由器
|
||||
context_middlewares []ContextMiddleware
|
||||
}
|
||||
|
||||
@@ -39,7 +41,17 @@ pub fn (mut app Hono) get(path string, handler fn (mut Context) http.Response) {
|
||||
path: path
|
||||
handler: handler
|
||||
}
|
||||
app.context_hybrid_router.add_route('GET', h, '')
|
||||
|
||||
// 添加到快速路由器
|
||||
if app.use_fast_router {
|
||||
app.fast_router.add_route('GET', h, '') or {
|
||||
println('[WARNING] FastRouter failed to add route ${path}, falling back to hybrid router')
|
||||
app.context_hybrid_router.add_route('GET', h, '')
|
||||
}
|
||||
} else {
|
||||
app.context_hybrid_router.add_route('GET', h, '')
|
||||
}
|
||||
|
||||
app.context_trie_router.add_route('GET', path, h)
|
||||
app.context_router.handlers.get << h
|
||||
}
|
||||
@@ -49,43 +61,103 @@ pub fn (mut app Hono) post(path string, handler fn (mut Context) http.Response)
|
||||
path: path
|
||||
handler: handler
|
||||
}
|
||||
app.context_hybrid_router.add_route('POST', h, '')
|
||||
|
||||
// 添加到快速路由器
|
||||
if app.use_fast_router {
|
||||
app.fast_router.add_route('POST', h, '') or {
|
||||
println('[WARNING] FastRouter failed to add route ${path}, falling back to hybrid router')
|
||||
app.context_hybrid_router.add_route('POST', h, '')
|
||||
}
|
||||
} else {
|
||||
app.context_hybrid_router.add_route('POST', h, '')
|
||||
}
|
||||
|
||||
app.context_trie_router.add_route('POST', path, h)
|
||||
app.context_router.handlers.post << h
|
||||
}
|
||||
|
||||
pub fn (mut app Hono) put(path string, handler fn (mut Context) http.Response) {
|
||||
h := ContextHandler{path, handler}
|
||||
|
||||
// 添加到快速路由器
|
||||
if app.use_fast_router {
|
||||
app.fast_router.add_route('PUT', h, app.base_path) or {
|
||||
println('[WARNING] FastRouter failed to add route ${path}, falling back to hybrid router')
|
||||
app.context_hybrid_router.add_route('PUT', h, app.base_path)
|
||||
}
|
||||
} else {
|
||||
app.context_hybrid_router.add_route('PUT', h, app.base_path)
|
||||
}
|
||||
|
||||
app.context_router.handlers.put << h
|
||||
app.context_hybrid_router.add_route('PUT', h, app.base_path)
|
||||
app.context_trie_router.add_route('PUT', path, h)
|
||||
}
|
||||
|
||||
pub fn (mut app Hono) delete(path string, handler fn (mut Context) http.Response) {
|
||||
h := ContextHandler{path, handler}
|
||||
|
||||
// 添加到快速路由器
|
||||
if app.use_fast_router {
|
||||
app.fast_router.add_route('DELETE', h, app.base_path) or {
|
||||
println('[WARNING] FastRouter failed to add route ${path}, falling back to hybrid router')
|
||||
app.context_hybrid_router.add_route('DELETE', h, app.base_path)
|
||||
}
|
||||
} else {
|
||||
app.context_hybrid_router.add_route('DELETE', h, app.base_path)
|
||||
}
|
||||
|
||||
app.context_router.handlers.delete << h
|
||||
app.context_hybrid_router.add_route('DELETE', h, app.base_path)
|
||||
app.context_trie_router.add_route('DELETE', path, h)
|
||||
}
|
||||
|
||||
pub fn (mut app Hono) patch(path string, handler fn (mut Context) http.Response) {
|
||||
h := ContextHandler{path, handler}
|
||||
|
||||
// 添加到快速路由器
|
||||
if app.use_fast_router {
|
||||
app.fast_router.add_route('PATCH', h, app.base_path) or {
|
||||
println('[WARNING] FastRouter failed to add route ${path}, falling back to hybrid router')
|
||||
app.context_hybrid_router.add_route('PATCH', h, app.base_path)
|
||||
}
|
||||
} else {
|
||||
app.context_hybrid_router.add_route('PATCH', h, app.base_path)
|
||||
}
|
||||
|
||||
app.context_router.handlers.patch << h
|
||||
app.context_hybrid_router.add_route('PATCH', h, app.base_path)
|
||||
app.context_trie_router.add_route('PATCH', path, h)
|
||||
}
|
||||
|
||||
pub fn (mut app Hono) head(path string, handler fn (mut Context) http.Response) {
|
||||
h := ContextHandler{path, handler}
|
||||
|
||||
// 添加到快速路由器
|
||||
if app.use_fast_router {
|
||||
app.fast_router.add_route('HEAD', h, app.base_path) or {
|
||||
println('[WARNING] FastRouter failed to add route ${path}, falling back to hybrid router')
|
||||
app.context_hybrid_router.add_route('HEAD', h, app.base_path)
|
||||
}
|
||||
} else {
|
||||
app.context_hybrid_router.add_route('HEAD', h, app.base_path)
|
||||
}
|
||||
|
||||
app.context_router.handlers.head << h
|
||||
app.context_hybrid_router.add_route('HEAD', h, app.base_path)
|
||||
app.context_trie_router.add_route('HEAD', path, h)
|
||||
}
|
||||
|
||||
pub fn (mut app Hono) options(path string, handler fn (mut Context) http.Response) {
|
||||
h := ContextHandler{path, handler}
|
||||
|
||||
// 添加到快速路由器
|
||||
if app.use_fast_router {
|
||||
app.fast_router.add_route('OPTIONS', h, app.base_path) or {
|
||||
println('[WARNING] FastRouter failed to add route ${path}, falling back to hybrid router')
|
||||
app.context_hybrid_router.add_route('OPTIONS', h, app.base_path)
|
||||
}
|
||||
} else {
|
||||
app.context_hybrid_router.add_route('OPTIONS', h, app.base_path)
|
||||
}
|
||||
|
||||
app.context_router.handlers.options << h
|
||||
app.context_hybrid_router.add_route('OPTIONS', h, app.base_path)
|
||||
app.context_trie_router.add_route('OPTIONS', path, h)
|
||||
}
|
||||
|
||||
@@ -126,26 +198,56 @@ fn (mut s ServerHanler) handle(req http.Request) http.Response {
|
||||
}
|
||||
|
||||
// 尝试 Context 路由
|
||||
if route_match := s.app.context_hybrid_router.match_route(req.method.str(), url.path) {
|
||||
// param 由路由匹配结果提供
|
||||
param_map := route_match.params.clone()
|
||||
// body
|
||||
body := req.data
|
||||
// 构造 Context
|
||||
mut ctx := Context.new(req, param_map, query_map, body)
|
||||
// 洋葱模型递归执行中间件
|
||||
return s.exec_context_middlewares(0, mut ctx, fn [route_match] (mut c Context) http.Response {
|
||||
return route_match.handler.handle(mut c)
|
||||
})
|
||||
// 优先使用快速路由器
|
||||
if s.app.use_fast_router {
|
||||
if route_match := s.app.fast_router.match_route(req.method.str(), url.path) {
|
||||
// param 由路由匹配结果提供
|
||||
param_map := route_match.params.clone()
|
||||
// body
|
||||
body := req.data
|
||||
// 构造 Context
|
||||
mut ctx := Context.new(req, param_map, query_map, body)
|
||||
// 洋葱模型递归执行中间件
|
||||
return s.exec_context_middlewares(0, mut ctx, fn [route_match] (mut c Context) http.Response {
|
||||
return route_match.handler.handle(mut c)
|
||||
})
|
||||
}
|
||||
|
||||
// 如果快速路由器没有匹配，回退到混合路由器
|
||||
if route_match := s.app.context_hybrid_router.match_route(req.method.str(), url.path) {
|
||||
// param 由路由匹配结果提供
|
||||
param_map := route_match.params.clone()
|
||||
// body
|
||||
body := req.data
|
||||
// 构造 Context
|
||||
mut ctx := Context.new(req, param_map, query_map, body)
|
||||
// 洋葱模型递归执行中间件
|
||||
return s.exec_context_middlewares(0, mut ctx, fn [route_match] (mut c Context) http.Response {
|
||||
return route_match.handler.handle(mut c)
|
||||
})
|
||||
}
|
||||
} else {
|
||||
// 如果没有匹配的路由，也要执行中间件
|
||||
param_map := map[string]string{}
|
||||
body := req.data
|
||||
mut ctx := Context.new(req, param_map, query_map, body)
|
||||
return s.exec_context_middlewares(0, mut ctx, fn [res] (mut c Context) http.Response {
|
||||
return res
|
||||
})
|
||||
if route_match := s.app.context_hybrid_router.match_route(req.method.str(), url.path) {
|
||||
// param 由路由匹配结果提供
|
||||
param_map := route_match.params.clone()
|
||||
// body
|
||||
body := req.data
|
||||
// 构造 Context
|
||||
mut ctx := Context.new(req, param_map, query_map, body)
|
||||
// 洋葱模型递归执行中间件
|
||||
return s.exec_context_middlewares(0, mut ctx, fn [route_match] (mut c Context) http.Response {
|
||||
return route_match.handler.handle(mut c)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// 如果没有匹配的路由，也要执行中间件
|
||||
param_map := map[string]string{}
|
||||
body := req.data
|
||||
mut ctx := Context.new(req, param_map, query_map, body)
|
||||
return s.exec_context_middlewares(0, mut ctx, fn [res] (mut c Context) http.Response {
|
||||
return res
|
||||
})
|
||||
}
|
||||
|
||||
// Context 版本的中间件执行函数
|
||||
@@ -200,14 +302,38 @@ pub fn (mut app Hono) set_base_path(base_path string) {
|
||||
|
||||
// 路由统计信息
|
||||
pub fn (app Hono) get_router_stats() (int, int, int, int) {
|
||||
static_paths, dynamic_paths := app.context_hybrid_router.get_all_routes()
|
||||
cache_size, cache_capacity := app.context_hybrid_router.get_cache_stats()
|
||||
return static_paths.len, dynamic_paths.len, cache_size, cache_capacity
|
||||
if app.use_fast_router {
|
||||
static_count, dynamic_count, cache_count := app.fast_router.get_stats()
|
||||
return static_count, dynamic_count, cache_count, 0
|
||||
} else {
|
||||
static_paths, dynamic_paths := app.context_hybrid_router.get_all_routes()
|
||||
cache_size, cache_capacity := app.context_hybrid_router.get_cache_stats()
|
||||
return static_paths.len, dynamic_paths.len, cache_size, cache_capacity
|
||||
}
|
||||
}
|
||||
|
||||
// 清理缓存
|
||||
pub fn (mut app Hono) clear_cache() {
|
||||
app.context_hybrid_router.clear_cache()
|
||||
if app.use_fast_router {
|
||||
app.fast_router.clear_cache()
|
||||
} else {
|
||||
app.context_hybrid_router.clear_cache()
|
||||
}
|
||||
}
|
||||
|
||||
// 启用/禁用快速路由器
|
||||
pub fn (mut app Hono) set_fast_router_enabled(enabled bool) {
|
||||
app.use_fast_router = enabled
|
||||
println('[INFO] FastRouter ${if enabled { 'enabled' } else { 'disabled' }}')
|
||||
}
|
||||
|
||||
// 获取路由器性能分析
|
||||
pub fn (mut app Hono) analyze_router_performance() {
|
||||
if app.use_fast_router {
|
||||
app.fast_router.analyze_performance()
|
||||
} else {
|
||||
app.context_hybrid_router.analyze_router_performance()
|
||||
}
|
||||
}
|
||||
|
||||
pub fn Hono.new() Hono {
|
||||
@@ -218,5 +344,7 @@ pub fn Hono.new() Hono {
|
||||
context_router: ContextRouter{}
|
||||
context_hybrid_router: ContextHybridRouter.new()
|
||||
context_trie_router: ContextTrieRouter.new()
|
||||
fast_router: FastRouter.new()
|
||||
use_fast_router: true
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user