perf: 添加文件流式输入性能优化
This commit is contained in:
yangshangbo
@@ -0,0 +1,282 @@
|
||||
import hono
|
||||
import net.http
|
||||
import os
|
||||
import time
|
||||
|
||||
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 := ''
|
||||
for i in 0 .. 1000 {
|
||||
medium_content += 'This is line ${i:04d} with some content to make it longer and test streaming.\n'
|
||||
}
|
||||
os.write_file('test_medium.txt', medium_content) or {
|
||||
println(' ❌ 创建中等文件失败: $err')
|
||||
return
|
||||
}
|
||||
println(' ✅ 创建中等文件: test_medium.txt (${medium_content.len} bytes)')
|
||||
|
||||
// 创建大文件 (约1MB)
|
||||
mut large_content := ''
|
||||
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 += '${i:05d}: $base_line\n'
|
||||
}
|
||||
os.write_file('test_large.txt', large_content) or {
|
||||
println(' ❌ 创建大文件失败: $err')
|
||||
return
|
||||
}
|
||||
println(' ✅ 创建大文件: test_large.txt (${large_content.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('清理完成')
|
||||
}
|
||||
Reference in New Issue
Block a user