meiseayoung
3b8d21dd0d
- Change all `import hono` to `import meiseayoung.hono` in code and docs - Ensure compatibility with VPM installation via `v install meiseayoung.hono`
286 lines
7.6 KiB
V
286 lines
7.6 KiB
V
import meiseayoung.hono
|
||
import net.http
|
||
import os
|
||
import time
|
||
import strings
|
||
|
||
fn main() {
|
||
println('=== 文件流式传输功能测试 ===')
|
||
|
||
// 创建测试文件
|
||
create_test_files()
|
||
|
||
// 创建Context进行测试
|
||
test_file_streaming()
|
||
|
||
println('\n功能测试完成!')
|
||
|
||
// 清理测试文件
|
||
cleanup_test_files()
|
||
}
|
||
|
||
fn create_test_files() {
|
||
println('创建测试文件...')
|
||
|
||
// 创建小文件 (1KB)
|
||
small_content := 'Hello World! '.repeat(80)
|
||
os.write_file('test_small.txt', small_content) or {
|
||
println(' ❌ 创建小文件失败: $err')
|
||
return
|
||
}
|
||
println(' ✅ 创建小文件: test_small.txt (${small_content.len} bytes)')
|
||
|
||
// 创建中等文件 (约100KB)
|
||
mut medium_content := strings.new_builder(100 * 1024)
|
||
for i in 0 .. 1000 {
|
||
medium_content.write_string('This is line ${i:04d} with some content to make it longer and test streaming.\n')
|
||
}
|
||
medium_str := medium_content.str()
|
||
os.write_file('test_medium.txt', medium_str) or {
|
||
println(' ❌ 创建中等文件失败: $err')
|
||
return
|
||
}
|
||
println(' ✅ 创建中等文件: test_medium.txt (${medium_str.len} bytes)')
|
||
|
||
// 创建大文件 (约1MB)
|
||
mut large_content := strings.new_builder(1024 * 1024)
|
||
base_line := 'This is a long line of text that will be repeated many times to create a large file for testing streaming functionality. '
|
||
for i in 0 .. 10000 {
|
||
large_content.write_string('${i:05d}: $base_line\n')
|
||
}
|
||
large_str := large_content.str()
|
||
os.write_file('test_large.txt', large_str) or {
|
||
println(' ❌ 创建大文件失败: $err')
|
||
return
|
||
}
|
||
println(' ✅ 创建大文件: test_large.txt (${large_str.len} bytes)')
|
||
}
|
||
|
||
fn test_file_streaming() {
|
||
println('\n开始功能测试...')
|
||
|
||
// 创建模拟请求
|
||
test_req := http.Request{
|
||
method: http.Method.get
|
||
url: '/test'
|
||
data: ''
|
||
header: http.new_header()
|
||
}
|
||
|
||
// 测试1：传统文件服务
|
||
println('\n--- 测试1: 传统文件服务 ---')
|
||
check_traditional_file_serving(test_req)
|
||
|
||
// 测试2：流式文件服务
|
||
println('\n--- 测试2: 流式文件服务 ---')
|
||
check_stream_file_serving(test_req)
|
||
|
||
// 测试3：智能文件服务
|
||
println('\n--- 测试3: 智能文件服务 ---')
|
||
check_smart_file_serving(test_req)
|
||
|
||
// 测试4：Range请求
|
||
println('\n--- 测试4: Range请求测试 ---')
|
||
check_range_requests()
|
||
|
||
// 测试5：自定义选项测试
|
||
println('\n--- 测试5: 自定义选项测试 ---')
|
||
check_custom_options(test_req)
|
||
}
|
||
|
||
fn check_traditional_file_serving(req http.Request) {
|
||
files_to_test := ['test_small.txt', 'test_medium.txt', 'test_large.txt']
|
||
|
||
for file in files_to_test {
|
||
if !os.exists(file) {
|
||
println(' ❌ 文件不存在: $file')
|
||
continue
|
||
}
|
||
|
||
// 创建Context
|
||
mut ctx := hono.Context.new(req, map[string]string{}, map[string]string{}, '')
|
||
|
||
start := time.now()
|
||
response := ctx.file(file)
|
||
duration := time.now() - start
|
||
|
||
mut original_size := i64(0)
|
||
if stat := os.stat(file) {
|
||
original_size = stat.size
|
||
}
|
||
response_size := response.body.len
|
||
|
||
if response.status_code == 200 && response_size == int(original_size) {
|
||
println(' ✅ $file: 成功 (${response_size} bytes, ${duration})')
|
||
} else {
|
||
println(' ❌ $file: 失败 (状态码: ${response.status_code}, 大小: ${response_size}/${original_size})')
|
||
}
|
||
}
|
||
}
|
||
|
||
fn check_stream_file_serving(req http.Request) {
|
||
files_to_test := ['test_small.txt', 'test_medium.txt', 'test_large.txt']
|
||
|
||
for file in files_to_test {
|
||
if !os.exists(file) {
|
||
println(' ❌ 文件不存在: $file')
|
||
continue
|
||
}
|
||
|
||
// 创建Context
|
||
mut ctx := hono.Context.new(req, map[string]string{}, map[string]string{}, '')
|
||
|
||
start := time.now()
|
||
response := ctx.file_stream(file)
|
||
duration := time.now() - start
|
||
|
||
mut original_size := i64(0)
|
||
if stat := os.stat(file) {
|
||
original_size = stat.size
|
||
}
|
||
response_size := response.body.len
|
||
|
||
if response.status_code == 200 && response_size == int(original_size) {
|
||
println(' ✅ $file (流式): 成功 (${response_size} bytes, ${duration})')
|
||
} else {
|
||
println(' ❌ $file (流式): 失败 (状态码: ${response.status_code}, 大小: ${response_size}/${original_size})')
|
||
}
|
||
}
|
||
}
|
||
|
||
fn check_smart_file_serving(req http.Request) {
|
||
files_to_test := ['test_small.txt', 'test_medium.txt', 'test_large.txt']
|
||
|
||
for file in files_to_test {
|
||
if !os.exists(file) {
|
||
println(' ❌ 文件不存在: $file')
|
||
continue
|
||
}
|
||
|
||
// 创建Context
|
||
mut ctx := hono.Context.new(req, map[string]string{}, map[string]string{}, '')
|
||
|
||
start := time.now()
|
||
response := ctx.file_smart(file)
|
||
duration := time.now() - start
|
||
|
||
mut original_size := i64(0)
|
||
if stat := os.stat(file) {
|
||
original_size = stat.size
|
||
}
|
||
response_size := response.body.len
|
||
|
||
// 检查是否正确选择了传输方式
|
||
mut expected_method := "内存"
|
||
if original_size > 50 * 1024 * 1024 { // 默认阈值50MB
|
||
expected_method = "流式"
|
||
}
|
||
|
||
if response.status_code == 200 && response_size == int(original_size) {
|
||
println(' ✅ $file (智能-$expected_method): 成功 (${response_size} bytes, ${duration})')
|
||
} else {
|
||
println(' ❌ $file (智能): 失败 (状态码: ${response.status_code}, 大小: ${response_size}/${original_size})')
|
||
}
|
||
}
|
||
}
|
||
|
||
fn check_range_requests() {
|
||
file := 'test_medium.txt'
|
||
if !os.exists(file) {
|
||
println(' ❌ 测试文件不存在: $file')
|
||
return
|
||
}
|
||
|
||
// 创建带Range头的请求
|
||
mut range_header := http.new_header()
|
||
range_header.add_custom('Range', 'bytes=0-99') or { }
|
||
|
||
range_req := http.Request{
|
||
method: http.Method.get
|
||
url: '/test'
|
||
data: ''
|
||
header: range_header
|
||
}
|
||
|
||
// 创建Context
|
||
mut ctx := hono.Context.new(range_req, map[string]string{}, map[string]string{}, '')
|
||
|
||
// 使用支持Range的选项
|
||
options := hono.FileOptions{
|
||
enable_range: true
|
||
stream_threshold: 1024 // 强制使用流式传输来测试Range功能
|
||
}
|
||
|
||
response := ctx.file_stream_with_options(file, options)
|
||
|
||
if response.status_code == 206 { // Partial Content
|
||
println(' ✅ Range请求: 成功 (状态码: 206, 内容长度: ${response.body.len})')
|
||
|
||
// 检查Content-Range头
|
||
if content_range := response.header.get_custom('Content-Range') {
|
||
println(' 📊 Content-Range: $content_range')
|
||
}
|
||
} else {
|
||
println(' ❌ Range请求: 失败 (状态码: ${response.status_code})')
|
||
}
|
||
}
|
||
|
||
fn check_custom_options(req http.Request) {
|
||
file := 'test_medium.txt'
|
||
if !os.exists(file) {
|
||
println(' ❌ 测试文件不存在: $file')
|
||
return
|
||
}
|
||
|
||
// 测试自定义选项
|
||
custom_options := hono.FileOptions{
|
||
stream_threshold: 10 * 1024 // 10KB阈值
|
||
buffer_size: 2048 // 2KB缓冲区
|
||
enable_range: true
|
||
max_age: 7200
|
||
content_type: 'text/plain; charset=utf-8'
|
||
headers: {
|
||
'X-Custom-Header': 'streaming-test'
|
||
'X-Buffer-Size': '2048'
|
||
}
|
||
}
|
||
|
||
// 创建Context
|
||
mut ctx := hono.Context.new(req, map[string]string{}, map[string]string{}, '')
|
||
|
||
response := ctx.file_stream_with_options(file, custom_options)
|
||
|
||
if response.status_code == 200 {
|
||
println(' ✅ 自定义选项: 成功')
|
||
|
||
// 检查自定义头部
|
||
if custom_header := response.header.get_custom('X-Custom-Header') {
|
||
println(' 📋 自定义头部: $custom_header')
|
||
}
|
||
|
||
if cache_control := response.header.get_custom('Cache-Control') {
|
||
println(' 🕒 缓存控制: $cache_control')
|
||
}
|
||
|
||
if content_type := response.header.get_custom('Content-Type') {
|
||
println(' 📝 内容类型: $content_type')
|
||
}
|
||
} else {
|
||
println(' ❌ 自定义选项: 失败 (状态码: ${response.status_code})')
|
||
}
|
||
}
|
||
|
||
fn cleanup_test_files() {
|
||
println('\n清理测试文件...')
|
||
test_files := ['test_small.txt', 'test_medium.txt', 'test_large.txt']
|
||
|
||
for file in test_files {
|
||
if os.exists(file) {
|
||
os.rm(file) or {
|
||
println(' 警告: 无法删除 $file')
|
||
}
|
||
}
|
||
}
|
||
println('清理完成')
|
||
}
|