mirror of
https://github.com/ZeroHawkeye/wordZero.git
synced 2025-09-27 04:05:56 +08:00
591 lines
16 KiB
Go
591 lines
16 KiB
Go
package benchmark
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"runtime"
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/ZeroHawkeye/wordZero/pkg/document"
|
||
"github.com/ZeroHawkeye/wordZero/pkg/style"
|
||
)
|
||
|
||
// 统一的测试配置，与JavaScript和Python保持一致
|
||
var testIterations = map[string]int{
|
||
"basic": 50, // 基础文档创建
|
||
"complex": 30, // 复杂格式化
|
||
"table": 20, // 表格操作
|
||
"largeTable": 10, // 大表格处理
|
||
"largeDoc": 5, // 大型文档
|
||
"memory": 10, // 内存使用测试
|
||
}
|
||
|
||
// BenchmarkCreateBasicDocument 基础文档创建性能测试
|
||
func BenchmarkCreateBasicDocument(b *testing.B) {
|
||
outputDir := "../results/golang"
|
||
os.MkdirAll(outputDir, 0755)
|
||
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
doc := document.New()
|
||
doc.AddParagraph("这是一个基础性能测试文档")
|
||
doc.AddParagraph("测试内容包括基本的文本添加功能")
|
||
|
||
filename := filepath.Join(outputDir, fmt.Sprintf("basic_doc_%d.docx", i))
|
||
doc.Save(filename)
|
||
}
|
||
}
|
||
|
||
// BenchmarkComplexFormatting 复杂格式化性能测试
|
||
func BenchmarkComplexFormatting(b *testing.B) {
|
||
outputDir := "../results/golang"
|
||
os.MkdirAll(outputDir, 0755)
|
||
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
doc := document.New()
|
||
|
||
// 添加标题
|
||
doc.AddParagraph("性能测试报告").SetStyle(style.StyleHeading1)
|
||
doc.AddParagraph("测试概述").SetStyle(style.StyleHeading2)
|
||
|
||
// 添加格式化文本
|
||
para := doc.AddParagraph("")
|
||
para.AddFormattedText("粗体文本", &document.TextFormat{Bold: true})
|
||
para.AddFormattedText(" ", &document.TextFormat{})
|
||
para.AddFormattedText("斜体文本", &document.TextFormat{Italic: true})
|
||
para.AddFormattedText(" ", &document.TextFormat{})
|
||
para.AddFormattedText("彩色文本", &document.TextFormat{FontColor: "FF0000"})
|
||
|
||
// 添加不同样式的段落
|
||
for j := 0; j < 10; j++ {
|
||
para2 := doc.AddParagraph(fmt.Sprintf("这是第%d个段落，包含复杂格式化", j+1))
|
||
para2.SetAlignment(document.AlignCenter)
|
||
}
|
||
|
||
filename := filepath.Join(outputDir, fmt.Sprintf("complex_formatting_%d.docx", i))
|
||
doc.Save(filename)
|
||
}
|
||
}
|
||
|
||
// BenchmarkTableOperations 表格操作性能测试
|
||
func BenchmarkTableOperations(b *testing.B) {
|
||
outputDir := "../results/golang"
|
||
os.MkdirAll(outputDir, 0755)
|
||
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
doc := document.New()
|
||
doc.AddParagraph("表格性能测试").SetStyle(style.StyleHeading1)
|
||
|
||
// 创建10行5列的表格
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 10,
|
||
Cols: 5,
|
||
Width: 7200, // 5英寸 = 7200磅
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
|
||
// 填充表格数据
|
||
for row := 0; row < 10; row++ {
|
||
for col := 0; col < 5; col++ {
|
||
cellText := fmt.Sprintf("R%dC%d", row+1, col+1)
|
||
table.SetCellText(row, col, cellText)
|
||
}
|
||
}
|
||
|
||
filename := filepath.Join(outputDir, fmt.Sprintf("table_operations_%d.docx", i))
|
||
doc.Save(filename)
|
||
}
|
||
}
|
||
|
||
// BenchmarkLargeTable 大表格性能测试
|
||
func BenchmarkLargeTable(b *testing.B) {
|
||
outputDir := "../results/golang"
|
||
os.MkdirAll(outputDir, 0755)
|
||
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
doc := document.New()
|
||
doc.AddParagraph("大表格性能测试").SetStyle(style.StyleHeading1)
|
||
|
||
// 创建100行10列的大表格
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 100,
|
||
Cols: 10,
|
||
Width: 14400, // 10英寸 = 14400磅
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
|
||
// 填充表格数据
|
||
for row := 0; row < 100; row++ {
|
||
for col := 0; col < 10; col++ {
|
||
cellText := fmt.Sprintf("数据_%d_%d", row+1, col+1)
|
||
table.SetCellText(row, col, cellText)
|
||
}
|
||
}
|
||
|
||
filename := filepath.Join(outputDir, fmt.Sprintf("large_table_%d.docx", i))
|
||
doc.Save(filename)
|
||
}
|
||
}
|
||
|
||
// BenchmarkLargeDocument 大型文档性能测试
|
||
func BenchmarkLargeDocument(b *testing.B) {
|
||
outputDir := "../results/golang"
|
||
os.MkdirAll(outputDir, 0755)
|
||
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
doc := document.New()
|
||
doc.AddParagraph("大型文档性能测试").SetStyle(style.StyleHeading1)
|
||
|
||
// 添加1000个段落
|
||
for j := 0; j < 1000; j++ {
|
||
if j%10 == 0 {
|
||
// 每10个段落添加一个标题
|
||
doc.AddParagraph(fmt.Sprintf("章节 %d", j/10+1)).SetStyle(style.StyleHeading2)
|
||
}
|
||
|
||
doc.AddParagraph(fmt.Sprintf("这是第%d个段落。Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", j+1))
|
||
}
|
||
|
||
// 添加一个中等大小的表格
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 20,
|
||
Cols: 8,
|
||
Width: 11520, // 8英寸 = 11520磅
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
for row := 0; row < 20; row++ {
|
||
for col := 0; col < 8; col++ {
|
||
table.SetCellText(row, col, fmt.Sprintf("表格数据%d-%d", row+1, col+1))
|
||
}
|
||
}
|
||
|
||
filename := filepath.Join(outputDir, fmt.Sprintf("large_document_%d.docx", i))
|
||
doc.Save(filename)
|
||
}
|
||
}
|
||
|
||
// BenchmarkMemoryUsage 内存使用测试
|
||
func BenchmarkMemoryUsage(b *testing.B) {
|
||
outputDir := "../results/golang"
|
||
os.MkdirAll(outputDir, 0755)
|
||
|
||
var m1, m2 runtime.MemStats
|
||
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
runtime.GC()
|
||
runtime.ReadMemStats(&m1)
|
||
|
||
doc := document.New()
|
||
|
||
// 创建复杂内容
|
||
for j := 0; j < 100; j++ {
|
||
doc.AddParagraph(fmt.Sprintf("段落%d: 这是一个测试段落，用于测试内存使用情况", j+1))
|
||
}
|
||
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 50,
|
||
Cols: 6,
|
||
Width: 8640, // 6英寸 = 8640磅
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
for row := 0; row < 50; row++ {
|
||
for col := 0; col < 6; col++ {
|
||
table.SetCellText(row, col, fmt.Sprintf("单元格%d-%d", row+1, col+1))
|
||
}
|
||
}
|
||
|
||
runtime.ReadMemStats(&m2)
|
||
|
||
// 输出内存使用情况（可选）
|
||
if i == 0 {
|
||
b.Logf("内存使用: %d KB", (m2.Alloc-m1.Alloc)/1024)
|
||
}
|
||
|
||
filename := filepath.Join(outputDir, fmt.Sprintf("memory_test_%d.docx", i))
|
||
doc.Save(filename)
|
||
}
|
||
}
|
||
|
||
// === 新增：固定迭代次数的测试函数，与其他语言保持一致 ===
|
||
|
||
// TestFixedIterationsPerformance 固定迭代次数的性能测试，与JavaScript和Python保持一致
|
||
func TestFixedIterationsPerformance(t *testing.T) {
|
||
outputDir := "../results/golang"
|
||
os.MkdirAll(outputDir, 0755)
|
||
|
||
tests := []struct {
|
||
name string
|
||
iterations int
|
||
testFunc func(int) time.Duration
|
||
}{
|
||
{"基础文档创建", testIterations["basic"], testBasicDocumentCreationFixed},
|
||
{"复杂格式化", testIterations["complex"], testComplexFormattingFixed},
|
||
{"表格操作", testIterations["table"], testTableOperationsFixed},
|
||
{"大表格处理", testIterations["largeTable"], testLargeTableProcessingFixed},
|
||
{"大型文档", testIterations["largeDoc"], testLargeDocumentFixed},
|
||
{"内存使用测试", testIterations["memory"], testMemoryUsageFixed},
|
||
}
|
||
|
||
results := make([]map[string]interface{}, 0, len(tests))
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
t.Logf("开始测试: %s (迭代次数: %d)", tt.name, tt.iterations)
|
||
|
||
times := make([]float64, 0, tt.iterations)
|
||
|
||
for i := 0; i < tt.iterations; i++ {
|
||
duration := tt.testFunc(i)
|
||
times = append(times, float64(duration.Nanoseconds())/1e6) // 转换为毫秒
|
||
|
||
if i%max(1, tt.iterations/10) == 0 {
|
||
t.Logf(" 进度: %d/%d", i+1, tt.iterations)
|
||
}
|
||
}
|
||
|
||
// 计算统计数据
|
||
var total float64
|
||
minTime := times[0]
|
||
maxTime := times[0]
|
||
|
||
for _, time := range times {
|
||
total += time
|
||
if time < minTime {
|
||
minTime = time
|
||
}
|
||
if time > maxTime {
|
||
maxTime = time
|
||
}
|
||
}
|
||
|
||
avgTime := total / float64(len(times))
|
||
|
||
result := map[string]interface{}{
|
||
"name": tt.name,
|
||
"avgTime": fmt.Sprintf("%.2f", avgTime),
|
||
"minTime": fmt.Sprintf("%.2f", minTime),
|
||
"maxTime": fmt.Sprintf("%.2f", maxTime),
|
||
"iterations": tt.iterations,
|
||
}
|
||
|
||
results = append(results, result)
|
||
|
||
t.Logf(" 平均耗时: %.2fms", avgTime)
|
||
t.Logf(" 最小耗时: %.2fms", minTime)
|
||
t.Logf(" 最大耗时: %.2fms", maxTime)
|
||
})
|
||
}
|
||
|
||
// 生成性能报告（JSON格式，与其他语言保持一致）
|
||
report := map[string]interface{}{
|
||
"timestamp": time.Now().Format(time.RFC3339),
|
||
"platform": "Golang",
|
||
"goVersion": runtime.Version(),
|
||
"results": results,
|
||
}
|
||
|
||
// 保存报告
|
||
reportPath := filepath.Join(outputDir, "performance_report.json")
|
||
file, err := os.Create(reportPath)
|
||
if err != nil {
|
||
t.Fatalf("无法创建报告文件: %v", err)
|
||
}
|
||
defer file.Close()
|
||
|
||
fmt.Fprintf(file, "{\n")
|
||
fmt.Fprintf(file, " \"timestamp\": \"%s\",\n", report["timestamp"])
|
||
fmt.Fprintf(file, " \"platform\": \"%s\",\n", report["platform"])
|
||
fmt.Fprintf(file, " \"goVersion\": \"%s\",\n", report["goVersion"])
|
||
fmt.Fprintf(file, " \"results\": [\n")
|
||
|
||
for i, result := range results {
|
||
fmt.Fprintf(file, " {\n")
|
||
fmt.Fprintf(file, " \"name\": \"%s\",\n", result["name"])
|
||
fmt.Fprintf(file, " \"avgTime\": \"%s\",\n", result["avgTime"])
|
||
fmt.Fprintf(file, " \"minTime\": \"%s\",\n", result["minTime"])
|
||
fmt.Fprintf(file, " \"maxTime\": \"%s\",\n", result["maxTime"])
|
||
fmt.Fprintf(file, " \"iterations\": %d\n", result["iterations"])
|
||
if i < len(results)-1 {
|
||
fmt.Fprintf(file, " },\n")
|
||
} else {
|
||
fmt.Fprintf(file, " }\n")
|
||
}
|
||
}
|
||
|
||
fmt.Fprintf(file, " ]\n")
|
||
fmt.Fprintf(file, "}\n")
|
||
|
||
t.Logf("\n=== Golang 性能测试报告 ===")
|
||
t.Logf("详细报告已保存到: %s", reportPath)
|
||
}
|
||
|
||
// max 辅助函数
|
||
func max(a, b int) int {
|
||
if a > b {
|
||
return a
|
||
}
|
||
return b
|
||
}
|
||
|
||
// === 固定迭代次数的测试实现函数 ===
|
||
|
||
func testBasicDocumentCreationFixed(index int) time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
doc.AddParagraph("这是一个基础性能测试文档")
|
||
doc.AddParagraph("测试内容包括基本的文本添加功能")
|
||
doc.Save(fmt.Sprintf("../results/golang/fixed_basic_doc_%d.docx", index))
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
func testComplexFormattingFixed(index int) time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
doc.AddParagraph("性能测试报告").SetStyle(style.StyleHeading1)
|
||
doc.AddParagraph("测试概述").SetStyle(style.StyleHeading2)
|
||
|
||
para := doc.AddParagraph("")
|
||
para.AddFormattedText("粗体文本", &document.TextFormat{Bold: true})
|
||
para.AddFormattedText(" ", &document.TextFormat{})
|
||
para.AddFormattedText("斜体文本", &document.TextFormat{Italic: true})
|
||
para.AddFormattedText(" ", &document.TextFormat{})
|
||
para.AddFormattedText("彩色文本", &document.TextFormat{FontColor: "FF0000"})
|
||
|
||
// 添加不同样式的段落
|
||
for j := 0; j < 10; j++ {
|
||
para2 := doc.AddParagraph(fmt.Sprintf("这是第%d个段落，包含复杂格式化", j+1))
|
||
para2.SetAlignment(document.AlignCenter)
|
||
}
|
||
|
||
doc.Save(fmt.Sprintf("../results/golang/fixed_complex_formatting_%d.docx", index))
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
func testTableOperationsFixed(index int) time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
doc.AddParagraph("表格性能测试").SetStyle(style.StyleHeading1)
|
||
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 10,
|
||
Cols: 5,
|
||
Width: 7200, // 5英寸
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
|
||
for row := 0; row < 10; row++ {
|
||
for col := 0; col < 5; col++ {
|
||
table.SetCellText(row, col, fmt.Sprintf("R%dC%d", row+1, col+1))
|
||
}
|
||
}
|
||
|
||
doc.Save(fmt.Sprintf("../results/golang/fixed_table_operations_%d.docx", index))
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
func testLargeTableProcessingFixed(index int) time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
doc.AddParagraph("大表格性能测试").SetStyle(style.StyleHeading1)
|
||
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 100,
|
||
Cols: 10,
|
||
Width: 14400, // 10英寸
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
|
||
for row := 0; row < 100; row++ {
|
||
for col := 0; col < 10; col++ {
|
||
table.SetCellText(row, col, fmt.Sprintf("数据_%d_%d", row+1, col+1))
|
||
}
|
||
}
|
||
|
||
doc.Save(fmt.Sprintf("../results/golang/fixed_large_table_%d.docx", index))
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
func testLargeDocumentFixed(index int) time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
doc.AddParagraph("大型文档性能测试").SetStyle(style.StyleHeading1)
|
||
|
||
// 添加1000个段落
|
||
for j := 0; j < 1000; j++ {
|
||
if j%10 == 0 {
|
||
// 每10个段落添加一个标题
|
||
doc.AddParagraph(fmt.Sprintf("章节 %d", j/10+1)).SetStyle(style.StyleHeading2)
|
||
}
|
||
|
||
doc.AddParagraph(fmt.Sprintf("这是第%d个段落。Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", j+1))
|
||
}
|
||
|
||
// 添加一个中等大小的表格
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 20,
|
||
Cols: 8,
|
||
Width: 11520, // 8英寸
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
for row := 0; row < 20; row++ {
|
||
for col := 0; col < 8; col++ {
|
||
table.SetCellText(row, col, fmt.Sprintf("表格数据%d-%d", row+1, col+1))
|
||
}
|
||
}
|
||
|
||
doc.Save(fmt.Sprintf("../results/golang/fixed_large_document_%d.docx", index))
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
func testMemoryUsageFixed(index int) time.Duration {
|
||
start := time.Now()
|
||
|
||
var m1, m2 runtime.MemStats
|
||
runtime.GC()
|
||
runtime.ReadMemStats(&m1)
|
||
|
||
doc := document.New()
|
||
|
||
// 创建复杂内容
|
||
for j := 0; j < 100; j++ {
|
||
doc.AddParagraph(fmt.Sprintf("段落%d: 这是一个测试段落，用于测试内存使用情况", j+1))
|
||
}
|
||
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 50,
|
||
Cols: 6,
|
||
Width: 8640, // 6英寸
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
for row := 0; row < 50; row++ {
|
||
for col := 0; col < 6; col++ {
|
||
table.SetCellText(row, col, fmt.Sprintf("单元格%d-%d", row+1, col+1))
|
||
}
|
||
}
|
||
|
||
runtime.ReadMemStats(&m2)
|
||
|
||
doc.Save(fmt.Sprintf("../results/golang/fixed_memory_test_%d.docx", index))
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
// TestPerformanceComparison 性能对比测试（非基准测试，用于详细分析）
|
||
func TestPerformanceComparison(t *testing.T) {
|
||
outputDir := "../results/golang"
|
||
os.MkdirAll(outputDir, 0755)
|
||
|
||
tests := []struct {
|
||
name string
|
||
testFunc func() time.Duration
|
||
}{
|
||
{"基础文档创建", testBasicDocumentCreation},
|
||
{"复杂格式化", testComplexFormatting},
|
||
{"表格操作", testTableOperations},
|
||
{"大表格处理", testLargeTableProcessing},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
// 运行3次取平均值
|
||
var total time.Duration
|
||
for i := 0; i < 3; i++ {
|
||
duration := tt.testFunc()
|
||
total += duration
|
||
}
|
||
avg := total / 3
|
||
t.Logf("%s 平均耗时: %v", tt.name, avg)
|
||
})
|
||
}
|
||
}
|
||
|
||
func testBasicDocumentCreation() time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
doc.AddParagraph("基础文档测试")
|
||
doc.AddParagraph("这是一个性能测试文档")
|
||
doc.Save("../results/golang/perf_basic.docx")
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
func testComplexFormatting() time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
doc.AddParagraph("复杂格式化测试").SetStyle(style.StyleHeading1)
|
||
|
||
para := doc.AddParagraph("")
|
||
para.AddFormattedText("粗体", &document.TextFormat{Bold: true})
|
||
para.AddFormattedText(" ", &document.TextFormat{})
|
||
para.AddFormattedText("斜体", &document.TextFormat{Italic: true})
|
||
para.AddFormattedText(" ", &document.TextFormat{})
|
||
para.AddFormattedText("红色", &document.TextFormat{FontColor: "FF0000"})
|
||
|
||
doc.Save("../results/golang/perf_complex.docx")
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
func testTableOperations() time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 20,
|
||
Cols: 5,
|
||
Width: 7200, // 5英寸
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
|
||
for row := 0; row < 20; row++ {
|
||
for col := 0; col < 5; col++ {
|
||
table.SetCellText(row, col, fmt.Sprintf("R%dC%d", row+1, col+1))
|
||
}
|
||
}
|
||
|
||
doc.Save("../results/golang/perf_table.docx")
|
||
|
||
return time.Since(start)
|
||
}
|
||
|
||
func testLargeTableProcessing() time.Duration {
|
||
start := time.Now()
|
||
|
||
doc := document.New()
|
||
tableConfig := &document.TableConfig{
|
||
Rows: 100,
|
||
Cols: 8,
|
||
Width: 11520, // 8英寸
|
||
}
|
||
table := doc.AddTable(tableConfig)
|
||
|
||
for row := 0; row < 100; row++ {
|
||
for col := 0; col < 8; col++ {
|
||
table.SetCellText(row, col, fmt.Sprintf("数据%d-%d", row+1, col+1))
|
||
}
|
||
}
|
||
|
||
doc.Save("../results/golang/perf_large_table.docx")
|
||
|
||
return time.Since(start)
|
||
}
|