89 lines
1.9 KiB
Go
89 lines
1.9 KiB
Go
package benchmark
|
||
|
||
import (
|
||
"strconv"
|
||
"testing"
|
||
|
||
"github.com/landaiqing/go-pixelnebula"
|
||
"github.com/landaiqing/go-pixelnebula/style"
|
||
)
|
||
|
||
// BenchmarkBasicAvatarGeneration 测试基本头像生成性能
|
||
func BenchmarkBasicAvatarGeneration(b *testing.B) {
|
||
// 重置计时器
|
||
b.ResetTimer()
|
||
|
||
for i := 0; i < b.N; i++ {
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(231, 231)
|
||
|
||
// 生成SVG
|
||
_, err := pn.Generate("benchmark-id", false).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// BenchmarkAvatarWithNoEnvironment 测试无环境头像生成性能
|
||
func BenchmarkAvatarWithNoEnvironment(b *testing.B) {
|
||
// 重置计时器
|
||
b.ResetTimer()
|
||
|
||
for i := 0; i < b.N; i++ {
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(231, 231)
|
||
|
||
// 生成无环境SVG
|
||
_, err := pn.Generate("benchmark-id", true).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// BenchmarkDifferentSizes 测试不同大小头像生成性能
|
||
func BenchmarkDifferentSizes(b *testing.B) {
|
||
sizes := []int{100, 200, 400, 800}
|
||
|
||
for _, size := range sizes {
|
||
b.Run("Size_"+Itoa(size), func(b *testing.B) {
|
||
b.ResetTimer()
|
||
|
||
for i := 0; i < b.N; i++ {
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(size, size)
|
||
|
||
_, err := pn.Generate("benchmark-size-"+Itoa(size), false).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// BenchmarkIDReuse 测试多次使用相同ID生成头像的性能（不使用缓存）
|
||
func BenchmarkIDReuse(b *testing.B) {
|
||
pn := pixelnebula.NewPixelNebula()
|
||
pn.WithStyle(style.GirlStyle)
|
||
pn.WithSize(231, 231)
|
||
|
||
b.ResetTimer()
|
||
|
||
for i := 0; i < b.N; i++ {
|
||
_, err := pn.Generate("fixed-benchmark-id", false).ToSVG()
|
||
if err != nil {
|
||
b.Fatal(err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// Itoa 简单的整数转字符串函数
|
||
func Itoa(n int) string {
|
||
return strconv.Itoa(n)
|
||
}
|