306 lines
7.3 KiB
Go
306 lines
7.3 KiB
Go
|
// repo_bench_test.go — repository 层性能基准测试
|
|||
|
|
// 覆盖：批量写入、并发只读查询、分页列表、更新状态、软删除
|
|||
|
|
package repository
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"sync"
|
|||
|
|
"sync/atomic"
|
|||
|
|
"testing"
|
|||
|
|
|
|||
|
|
_ "modernc.org/sqlite"
|
|||
|
|
gormsqlite "gorm.io/driver/sqlite"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
"gorm.io/gorm/logger"
|
|||
|
|
|
|||
|
|
"github.com/user-management-system/internal/domain"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var repoBenchCounter int64
|
|||
|
|
|
|||
|
|
// openBenchDB 为 Benchmark 打开独立内存 DB（不依赖 *testing.T）
|
|||
|
|
func openBenchDB(b *testing.B) *gorm.DB {
|
|||
|
|
b.Helper()
|
|||
|
|
id := atomic.AddInt64(&repoBenchCounter, 1)
|
|||
|
|
dsn := fmt.Sprintf("file:repobenchdb%d?mode=memory&cache=private", id)
|
|||
|
|
db, err := gorm.Open(gormsqlite.New(gormsqlite.Config{
|
|||
|
|
DriverName: "sqlite",
|
|||
|
|
DSN: dsn,
|
|||
|
|
}), &gorm.Config{
|
|||
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
b.Fatalf("openBenchDB: %v", err)
|
|||
|
|
}
|
|||
|
|
if err := db.AutoMigrate(
|
|||
|
|
&domain.User{},
|
|||
|
|
&domain.Role{},
|
|||
|
|
&domain.Permission{},
|
|||
|
|
&domain.UserRole{},
|
|||
|
|
&domain.RolePermission{},
|
|||
|
|
); err != nil {
|
|||
|
|
b.Fatalf("AutoMigrate: %v", err)
|
|||
|
|
}
|
|||
|
|
return db
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// seedUsers 往 DB 插入 n 条用户
|
|||
|
|
func seedUsers(b *testing.B, repo *UserRepository, n int) {
|
|||
|
|
b.Helper()
|
|||
|
|
ctx := context.Background()
|
|||
|
|
for i := 0; i < n; i++ {
|
|||
|
|
if err := repo.Create(ctx, &domain.User{
|
|||
|
|
Username: fmt.Sprintf("benchuser%06d", i),
|
|||
|
|
Email: domain.StrPtr(fmt.Sprintf("bench%06d@example.com", i)),
|
|||
|
|
Phone: domain.StrPtr(fmt.Sprintf("1380000%04d", i%10000)),
|
|||
|
|
Password: "hashed_placeholder",
|
|||
|
|
Status: domain.UserStatusActive,
|
|||
|
|
}); err != nil {
|
|||
|
|
b.Fatalf("seedUsers i=%d: %v", i, err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_Create — 单条写入吞吐 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_Create(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
repo.Create(ctx, &domain.User{ //nolint:errcheck
|
|||
|
|
Username: fmt.Sprintf("cr_%d_%d", b.N, i),
|
|||
|
|
Email: domain.StrPtr(fmt.Sprintf("cr_%d_%d@bench.com", b.N, i)),
|
|||
|
|
Password: "hash",
|
|||
|
|
Status: domain.UserStatusActive,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_BulkCreate — 批量写入（串行） ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_BulkCreate(b *testing.B) {
|
|||
|
|
sizes := []int{10, 100, 500}
|
|||
|
|
for _, size := range sizes {
|
|||
|
|
size := size
|
|||
|
|
b.Run(fmt.Sprintf("batch=%d", size), func(b *testing.B) {
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
b.StopTimer()
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
users := make([]*domain.User, size)
|
|||
|
|
for j := 0; j < size; j++ {
|
|||
|
|
users[j] = &domain.User{
|
|||
|
|
Username: fmt.Sprintf("bulk_%d_%d_%d", i, j, size),
|
|||
|
|
Password: "hash",
|
|||
|
|
Status: domain.UserStatusActive,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
b.StartTimer()
|
|||
|
|
for _, u := range users {
|
|||
|
|
repo.Create(ctx, u) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_GetByID — 主键查询 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_GetByID(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 1000)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
b.RunParallel(func(pb *testing.PB) {
|
|||
|
|
id := int64(1)
|
|||
|
|
for pb.Next() {
|
|||
|
|
repo.GetByID(ctx, id) //nolint:errcheck
|
|||
|
|
id++
|
|||
|
|
if id > 1000 {
|
|||
|
|
id = 1
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_GetByUsername — 索引查询 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_GetByUsername(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 500)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
repo.GetByUsername(ctx, fmt.Sprintf("benchuser%06d", i%500)) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_GetByEmail — 索引查询 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_GetByEmail(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 500)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
repo.GetByEmail(ctx, fmt.Sprintf("bench%06d@example.com", i%500)) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_List — 分页列表 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_List(b *testing.B) {
|
|||
|
|
pageSizes := []int{10, 50, 200}
|
|||
|
|
for _, ps := range pageSizes {
|
|||
|
|
ps := ps
|
|||
|
|
b.Run(fmt.Sprintf("pageSize=%d", ps), func(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 1000)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
repo.List(ctx, 0, ps) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_ListByStatus ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_ListByStatus(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 1000)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
repo.ListByStatus(ctx, domain.UserStatusActive, 0, 20) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_UpdateStatus ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_UpdateStatus(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 200)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
id := int64(i%200) + 1
|
|||
|
|
repo.UpdateStatus(ctx, id, domain.UserStatusActive) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_Update — 全字段更新 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_Update(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 100)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
id := int64(i%100) + 1
|
|||
|
|
u, err := repo.GetByID(ctx, id)
|
|||
|
|
if err != nil {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
u.Nickname = fmt.Sprintf("nick%d", i)
|
|||
|
|
repo.Update(ctx, u) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_Delete — 软删除 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_Delete(b *testing.B) {
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
b.StopTimer()
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
repo.Create(ctx, &domain.User{Username: "victim", Password: "hash", Status: domain.UserStatusActive}) //nolint:errcheck
|
|||
|
|
b.StartTimer()
|
|||
|
|
repo.Delete(ctx, 1) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_ExistsByUsername ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_ExistsByUsername(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 500)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
b.RunParallel(func(pb *testing.PB) {
|
|||
|
|
i := 0
|
|||
|
|
for pb.Next() {
|
|||
|
|
repo.ExistsByUsername(ctx, fmt.Sprintf("benchuser%06d", i%500)) //nolint:errcheck
|
|||
|
|
i++
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_ConcurrentReadWrite — 高并发读写混合 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_ConcurrentReadWrite(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 200)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
|
|||
|
|
var mu sync.Mutex // SQLite 不支持多写并发，需要序列化写入
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
b.RunParallel(func(pb *testing.PB) {
|
|||
|
|
i := int64(1)
|
|||
|
|
for pb.Next() {
|
|||
|
|
if i%10 == 0 {
|
|||
|
|
// 10% 写操作
|
|||
|
|
mu.Lock()
|
|||
|
|
repo.UpdateLastLogin(ctx, i%200+1, "10.0.0.1") //nolint:errcheck
|
|||
|
|
mu.Unlock()
|
|||
|
|
} else {
|
|||
|
|
// 90% 读操作
|
|||
|
|
repo.GetByID(ctx, i%200+1) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
i++
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ---------- BenchmarkRepo_Search — 模糊搜索 ----------
|
|||
|
|
|
|||
|
|
func BenchmarkRepo_Search(b *testing.B) {
|
|||
|
|
db := openBenchDB(b)
|
|||
|
|
repo := NewUserRepository(db)
|
|||
|
|
seedUsers(b, repo, 2000)
|
|||
|
|
ctx := context.Background()
|
|||
|
|
b.ResetTimer()
|
|||
|
|
|
|||
|
|
keywords := []string{"benchuser000", "bench0001", "benchuser05"}
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
repo.Search(ctx, keywords[i%len(keywords)], 0, 20) //nolint:errcheck
|
|||
|
|
}
|
|||
|
|
}
|