139 lines
3.3 KiB
Go
139 lines
3.3 KiB
Go
package benchmark
|
||
|
||
import (
|
||
"sync"
|
||
"testing"
|
||
|
||
"github.com/landaiqing/go-pixelnebula"
|
||
"github.com/landaiqing/go-pixelnebula/style"
|
||
)
|
||
|
||
// BenchmarkConcurrentGeneration 测试并发生成头像的性能
|
||
func BenchmarkConcurrentGeneration(b *testing.B) {
|
||
concurrencyCounts := []int{1, 2, 4, 8, 16}
|
||
|
||
for _, count := range concurrencyCounts {
|
||
b.Run("Concurrent_"+Itoa(count), func(b *testing.B) {
|
||
b.ResetTimer()
|
||
|
||
// 将总迭代次数调整为b.N，确保可比较性
|
||
b.SetParallelism(count)
|
||
b.RunParallel(func(pb *testing.PB) {
|
||
counter := 0
|
||
for pb.Next() {
|
||
counter++
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(231, 231)
|
||
pn.WithDefaultCache()
|
||
|
||
_, err := pn.Generate("benchmark-concurrent-"+Itoa(counter), false).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
})
|
||
})
|
||
}
|
||
}
|
||
|
||
// BenchmarkConcurrentWithSharedInstance 测试使用共享实例进行并发生成的性能
|
||
func BenchmarkConcurrentWithSharedInstance(b *testing.B) {
|
||
concurrencyCounts := []int{1, 2, 4, 8, 16}
|
||
|
||
for _, count := range concurrencyCounts {
|
||
b.Run("SharedInstance_"+Itoa(count), func(b *testing.B) {
|
||
// 创建一个共享实例
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(231, 231)
|
||
pn.WithDefaultCache()
|
||
|
||
// 创建互斥锁保护共享实例
|
||
var mu sync.Mutex
|
||
|
||
b.ResetTimer()
|
||
b.SetParallelism(count)
|
||
b.RunParallel(func(pb *testing.PB) {
|
||
counter := 0
|
||
for pb.Next() {
|
||
counter++
|
||
// 锁定共享实例
|
||
mu.Lock()
|
||
_, err := pn.Generate("benchmark-shared-"+Itoa(counter), false).ToSVG()
|
||
mu.Unlock()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
})
|
||
})
|
||
}
|
||
}
|
||
|
||
// BenchmarkMemoryUsage 测试不同操作的内存使用情况
|
||
func BenchmarkMemoryUsage(b *testing.B) {
|
||
// 注意: 这个基准测试主要关注内存分配统计，
|
||
// Go 的基准测试框架会自动收集并报告内存统计数据
|
||
|
||
// 测试基本头像生成的内存使用
|
||
b.Run("BasicGeneration", func(b *testing.B) {
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(231, 231)
|
||
_, err := pn.Generate("memory-basic", false).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
})
|
||
|
||
// 测试添加动画的内存使用
|
||
b.Run("WithAnimations", func(b *testing.B) {
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(231, 231)
|
||
pn.WithRotateAnimation("env", 0, 360, 10, 1)
|
||
pn.WithFadeAnimation("eyes", "1", "0.3", 2, 1)
|
||
_, err := pn.Generate("memory-animations", false).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
})
|
||
|
||
// 测试缓存的内存使用
|
||
b.Run("WithCache", func(b *testing.B) {
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(231, 231)
|
||
pn.WithDefaultCache()
|
||
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
_, err := pn.Generate("memory-cache", false).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
})
|
||
|
||
// 测试大尺寸头像的内存使用
|
||
b.Run("LargeSize", func(b *testing.B) {
|
||
b.ResetTimer()
|
||
for i := 0; i < b.N; i++ {
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(1000, 1000)
|
||
_, err := pn.Generate("memory-large", false).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
})
|
||
}
|