module main

import hono
import net.http
import time

fn main() {
	println('=== 混合路由性能测试 ===')
	
	// 创建应用实例
	mut app := hono.new_hono()
	
	// 添加静态路由
	app.get('/api/users', fn (req hono.Request) http.Response {
		return hono.Response.json('{"users": []}')
	})
	
	app.get('/api/posts', fn (req hono.Request) http.Response {
		return hono.Response.json('{"posts": []}')
	})
	
	app.get('/api/comments', fn (req hono.Request) http.Response {
		return hono.Response.json('{"comments": []}')
	})
	
	// 添加动态路由
	app.get('/api/users/:id', fn (req hono.Request) http.Response {
		return hono.Response.json('{"user_id": "${req.param["id"]}"}')
	})
	
	app.get('/api/posts/:id/comments', fn (req hono.Request) http.Response {
		return hono.Response.json('{"post_id": "${req.param["id"]}", "comments": []}')
	})
	
	app.get('/api/**/search', fn (req hono.Request) http.Response {
		return hono.Response.json('{"search": "wildcard"}')
	})
	
	// 添加100个动态路由
	println('添加100个动态路由...')
	for i := 1; i <= 100; i++ {
		// 用户相关路由
		app.get('/api/users/:id/profile', fn (req hono.Request) http.Response {
			return hono.Response.json('{"user_id": "${req.param["id"]}", "profile": "data"}')
		})
		
		app.get('/api/users/:id/posts/:post_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"user_id": "${req.param["id"]}", "post_id": "${req.param["post_id"]}"}')
		})
		
		app.get('/api/users/:id/comments/:comment_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"user_id": "${req.param["id"]}", "comment_id": "${req.param["comment_id"]}"}')
		})
		
		// 帖子相关路由
		app.get('/api/posts/:id/author/:author_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"post_id": "${req.param["id"]}", "author_id": "${req.param["author_id"]}"}')
		})
		
		app.get('/api/posts/:id/tags/:tag_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"post_id": "${req.param["id"]}", "tag_id": "${req.param["tag_id"]}"}')
		})
		
		app.get('/api/posts/:id/categories/:category_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"post_id": "${req.param["id"]}", "category_id": "${req.param["category_id"]}"}')
		})
		
		// 评论相关路由
		app.get('/api/comments/:id/author/:author_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"comment_id": "${req.param["id"]}", "author_id": "${req.param["author_id"]}"}')
		})
		
		app.get('/api/comments/:id/post/:post_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"comment_id": "${req.param["id"]}", "post_id": "${req.param["post_id"]}"}')
		})
		
		// 分类相关路由
		app.get('/api/categories/:id/posts/:post_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"category_id": "${req.param["id"]}", "post_id": "${req.param["post_id"]}"}')
		})
		
		app.get('/api/categories/:id/tags/:tag_id', fn (req hono.Request) http.Response {
			return hono.Response.json('{"category_id": "${req.param["id"]}", "tag_id": "${req.param["tag_id"]}"}')
		})
	}
	
	// 测试静态路由性能
	println('\n--- 静态路由性能测试 ---')
	mut start := time.now()
	for i := 0; i < 1000000; i++ {
		_ = app.hybrid_router.match_route('GET', '/api/users')
		_ = app.hybrid_router.match_route('GET', '/api/posts')
		_ = app.hybrid_router.match_route('GET', '/api/comments')
	}
	mut end := time.now()
	println('静态路由 1000000次匹配耗时: ${end - start}')
	
	// 测试动态路由性能（包含新增的100个路由）
	println('\n--- 动态路由性能测试（103个动态路由） ---')
	start = time.now()
	for i := 0; i < 1000000; i++ { // 减少测试次数，因为路由数量大幅增加
		_ = app.hybrid_router.match_route('GET', '/api/users/123')
		_ = app.hybrid_router.match_route('GET', '/api/posts/456/comments')
		_ = app.hybrid_router.match_route('GET', '/api/anything/search')
		_ = app.hybrid_router.match_route('GET', '/api/users/789/profile')
		_ = app.hybrid_router.match_route('GET', '/api/posts/101/author/202')
		_ = app.hybrid_router.match_route('GET', '/api/comments/303/post/404')
		_ = app.hybrid_router.match_route('GET', '/api/categories/505/tags/606')
	}
	end = time.now()
	println('动态路由 1000000次匹配耗时: ${end - start}')
	
	// 测试缓存效果
	println('\n--- 缓存效果测试 ---')
	start = time.now()
	for i := 0; i < 1000000; i++ {
		_ = app.hybrid_router.match_route('GET', '/api/users/123')
	}
	end = time.now()
	println('缓存命中 1000000次匹配耗时: ${end - start}')
	
	// 测试路由查找性能（遍历所有动态路由）
	println('\n--- 路由查找性能测试 ---')
	start = time.now()
	for i := 0; i < 10000; i++ {
		// 测试不同路由的匹配
		_ = app.hybrid_router.match_route('GET', '/api/users/${i}')
		_ = app.hybrid_router.match_route('GET', '/api/posts/${i}/comments')
		_ = app.hybrid_router.match_route('GET', '/api/users/${i}/profile')
		_ = app.hybrid_router.match_route('GET', '/api/posts/${i}/author/${i+1}')
		_ = app.hybrid_router.match_route('GET', '/api/comments/${i}/post/${i+1}')
	}
	end = time.now()
	println('路由查找 50000次匹配耗时: ${end - start}')
	
	// 获取统计信息
	static_count, dynamic_count, cache_size, cache_capacity := app.get_router_stats()
	println('\n--- 路由统计信息 ---')
	println('静态路由数量: ${static_count}')
	println('动态路由数量: ${dynamic_count}')
	println('缓存大小: ${cache_size}/${cache_capacity}')
	
	// 添加 TrieRouter 性能测试
	println('\n--- Trie 路由树性能测试 ---')
	start = time.now()
	for i := 0; i < 1000000; i++ {
		_ = app.trie_router.match_route('GET', '/api/users')
		_ = app.trie_router.match_route('GET', '/api/posts')
		_ = app.trie_router.match_route('GET', '/api/comments')
	}
	end = time.now()
	println('Trie静态路由 1000000次匹配耗时: ${end - start}')

	start = time.now()
	for i := 0; i < 1000000; i++ {
		_ = app.trie_router.match_route('GET', '/api/users/123')
		_ = app.trie_router.match_route('GET', '/api/posts/456/comments')
		_ = app.trie_router.match_route('GET', '/api/anything/search')
		_ = app.trie_router.match_route('GET', '/api/users/789/profile')
		_ = app.trie_router.match_route('GET', '/api/posts/101/author/202')
		_ = app.trie_router.match_route('GET', '/api/comments/303/post/404')
		_ = app.trie_router.match_route('GET', '/api/categories/505/tags/606')
	}
	end = time.now()
	println('Trie动态路由 1000000次匹配耗时: ${end - start}')
	
	println('\n=== 测试完成 ===')
}