refactor: 整理项目根目录结构
整理内容: - 删除 60+ 临时测试输出文件 (*.txt) - 移动二进制文件到 bin/ 目录 - 移动 Shell 脚本到 scripts/ 目录 - scripts/dev/: check_gitea.sh, check_sub2api.sh, run_tests.sh - scripts/deploy/: deploy_*.sh, simple_deploy.sh - scripts/ops/: fix_nginx.sh, fix_ssl.sh, install_docker.sh - scripts/test/: test_*.sh, test_*.bat - 移动批处理文件到 scripts/ - 移动 Python 脚本到 tools/ - 清理临时日志文件 保留根目录必要文件: - go.mod, go.sum, go.work - Makefile, docker-compose.yml - .env.example, .gitignore - README.md, AGENTS.md, DEPLOY_GUIDE.md 验证: go build ./... && go test ./... 通过
This commit is contained in:
66
internal/monitoring/collector.go
Normal file
66
internal/monitoring/collector.go
Normal file
@@ -0,0 +1,66 @@
|
||||
package monitoring
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"log"
|
||||
"runtime"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// StartSystemMetricsCollector 启动后台系统指标采集循环
|
||||
// 每 15 秒采集一次:Go runtime 指标 + 数据库连接池状态
|
||||
// CRIT-03: 这是 SLO 错误预算追踪的基础数据来源
|
||||
func StartSystemMetricsCollector(ctx context.Context, m *Metrics, slo *SLOMetrics, db *gorm.DB) {
|
||||
ticker := time.NewTicker(15 * time.Second)
|
||||
defer ticker.Stop()
|
||||
|
||||
log.Println("[monitoring] system metrics collector started (interval=15s)")
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
log.Println("[monitoring] system metrics collector stopped")
|
||||
return
|
||||
case <-ticker.C:
|
||||
collectRuntimeMetrics(m)
|
||||
collectDBMetrics(slo, db)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// collectRuntimeMetrics 采集 Go runtime 指标
|
||||
func collectRuntimeMetrics(m *Metrics) {
|
||||
var memStats runtime.MemStats
|
||||
runtime.ReadMemStats(&memStats)
|
||||
|
||||
m.SetMemoryUsage(float64(memStats.Alloc))
|
||||
m.SetGoroutines(float64(runtime.NumGoroutine()))
|
||||
}
|
||||
|
||||
// collectDBMetrics 采集数据库连接池指标并上报 SLO 指标
|
||||
func collectDBMetrics(slo *SLOMetrics, db *gorm.DB) {
|
||||
if db == nil {
|
||||
return
|
||||
}
|
||||
|
||||
sqlDB, err := db.DB()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
updateDBConnectionMetricsFromStats(slo, sqlDB.Stats())
|
||||
}
|
||||
|
||||
// updateDBConnectionMetricsFromStats 从 sql.DBStats 更新连接池指标
|
||||
func updateDBConnectionMetricsFromStats(slo *SLOMetrics, stats sql.DBStats) {
|
||||
if slo == nil {
|
||||
return
|
||||
}
|
||||
slo.SetDBConnections(
|
||||
float64(stats.InUse),
|
||||
float64(stats.MaxOpenConnections),
|
||||
)
|
||||
}
|
||||
177
internal/monitoring/slo.go
Normal file
177
internal/monitoring/slo.go
Normal file
@@ -0,0 +1,177 @@
|
||||
package monitoring
|
||||
|
||||
import (
|
||||
"sync"
|
||||
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
)
|
||||
|
||||
// SLOMetrics 服务级别目标(SLO)相关指标
|
||||
// 这些指标是 SLO 测量的基础,用于计算错误预算燃烧率
|
||||
type SLOMetrics struct {
|
||||
// 缓存命中统计(alerts.yml 引用但原来未定义)
|
||||
CacheHitsTotal *prometheus.CounterVec
|
||||
CacheOperationsTotal *prometheus.CounterVec
|
||||
|
||||
// 数据库连接池状态(alerts.yml 引用但原来未定义)
|
||||
DBConnectionsActive prometheus.Gauge
|
||||
DBConnectionsMax prometheus.Gauge
|
||||
|
||||
// Token 操作
|
||||
TokenRefreshTotal *prometheus.CounterVec
|
||||
|
||||
// 账号安全事件
|
||||
AccountLockTotal prometheus.Counter
|
||||
AnomalyDetectedTotal *prometheus.CounterVec
|
||||
|
||||
// 错误预算燃烧率(可选,用于自定义仪表盘)
|
||||
ErrorBudgetBurnRate *prometheus.GaugeVec
|
||||
|
||||
registry *prometheus.Registry
|
||||
once sync.Once
|
||||
}
|
||||
|
||||
var (
|
||||
globalSLOMetrics *SLOMetrics
|
||||
globalSLOMetricsOnce sync.Once
|
||||
)
|
||||
|
||||
// NewSLOMetrics 创建 SLO 指标实例(使用独立 registry 避免测试冲突)
|
||||
func NewSLOMetrics() *SLOMetrics {
|
||||
reg := prometheus.NewRegistry()
|
||||
m := &SLOMetrics{registry: reg}
|
||||
|
||||
m.CacheHitsTotal = prometheus.NewCounterVec(
|
||||
prometheus.CounterOpts{
|
||||
Name: "cache_hits_total",
|
||||
Help: "Total number of cache hits",
|
||||
},
|
||||
[]string{"level", "operation"}, // level: l1/l2, operation: get/set
|
||||
)
|
||||
|
||||
m.CacheOperationsTotal = prometheus.NewCounterVec(
|
||||
prometheus.CounterOpts{
|
||||
Name: "cache_operations_total",
|
||||
Help: "Total number of cache operations",
|
||||
},
|
||||
[]string{"level", "operation"},
|
||||
)
|
||||
|
||||
m.DBConnectionsActive = prometheus.NewGauge(
|
||||
prometheus.GaugeOpts{
|
||||
Name: "db_connections_active",
|
||||
Help: "Number of active database connections",
|
||||
},
|
||||
)
|
||||
|
||||
m.DBConnectionsMax = prometheus.NewGauge(
|
||||
prometheus.GaugeOpts{
|
||||
Name: "db_connections_max",
|
||||
Help: "Maximum number of database connections configured",
|
||||
},
|
||||
)
|
||||
|
||||
m.TokenRefreshTotal = prometheus.NewCounterVec(
|
||||
prometheus.CounterOpts{
|
||||
Name: "token_refresh_total",
|
||||
Help: "Total number of token refresh attempts",
|
||||
},
|
||||
[]string{"status"}, // success/failure/rate_limited
|
||||
)
|
||||
|
||||
m.AccountLockTotal = prometheus.NewCounter(
|
||||
prometheus.CounterOpts{
|
||||
Name: "account_lock_total",
|
||||
Help: "Total number of account lockout events due to failed login attempts",
|
||||
},
|
||||
)
|
||||
|
||||
m.AnomalyDetectedTotal = prometheus.NewCounterVec(
|
||||
prometheus.CounterOpts{
|
||||
Name: "anomaly_detected_total",
|
||||
Help: "Total number of anomaly login detections",
|
||||
},
|
||||
[]string{"type"}, // geo_anomaly/device_anomaly/brute_force/suspicious_ip
|
||||
)
|
||||
|
||||
m.ErrorBudgetBurnRate = prometheus.NewGaugeVec(
|
||||
prometheus.GaugeOpts{
|
||||
Name: "error_budget_burn_rate",
|
||||
Help: "Current error budget burn rate multiplier (1.0 = nominal consumption)",
|
||||
},
|
||||
[]string{"slo"}, // api-availability/api-latency/login-success-rate
|
||||
)
|
||||
|
||||
reg.MustRegister(
|
||||
m.CacheHitsTotal,
|
||||
m.CacheOperationsTotal,
|
||||
m.DBConnectionsActive,
|
||||
m.DBConnectionsMax,
|
||||
m.TokenRefreshTotal,
|
||||
m.AccountLockTotal,
|
||||
m.AnomalyDetectedTotal,
|
||||
m.ErrorBudgetBurnRate,
|
||||
)
|
||||
|
||||
return m
|
||||
}
|
||||
|
||||
// GetGlobalSLOMetrics 获取全局 SLO 指标单例(生产使用)
|
||||
func GetGlobalSLOMetrics() *SLOMetrics {
|
||||
globalSLOMetricsOnce.Do(func() {
|
||||
m := NewSLOMetrics()
|
||||
// 注册到默认 registry 以便 /metrics 端点暴露
|
||||
prometheus.DefaultRegisterer.Register(m.CacheHitsTotal) //nolint:errcheck
|
||||
prometheus.DefaultRegisterer.Register(m.CacheOperationsTotal) //nolint:errcheck
|
||||
prometheus.DefaultRegisterer.Register(m.DBConnectionsActive) //nolint:errcheck
|
||||
prometheus.DefaultRegisterer.Register(m.DBConnectionsMax) //nolint:errcheck
|
||||
prometheus.DefaultRegisterer.Register(m.TokenRefreshTotal) //nolint:errcheck
|
||||
prometheus.DefaultRegisterer.Register(m.AccountLockTotal) //nolint:errcheck
|
||||
prometheus.DefaultRegisterer.Register(m.AnomalyDetectedTotal) //nolint:errcheck
|
||||
prometheus.DefaultRegisterer.Register(m.ErrorBudgetBurnRate) //nolint:errcheck
|
||||
globalSLOMetrics = m
|
||||
})
|
||||
return globalSLOMetrics
|
||||
}
|
||||
|
||||
// GetRegistry 获取私有 registry(测试使用)
|
||||
func (m *SLOMetrics) GetRegistry() *prometheus.Registry {
|
||||
return m.registry
|
||||
}
|
||||
|
||||
// RecordCacheHit 记录缓存命中
|
||||
func (m *SLOMetrics) RecordCacheHit(level, operation string) {
|
||||
m.CacheHitsTotal.WithLabelValues(level, operation).Inc()
|
||||
m.CacheOperationsTotal.WithLabelValues(level, operation).Inc()
|
||||
}
|
||||
|
||||
// RecordCacheMiss 记录缓存未命中
|
||||
func (m *SLOMetrics) RecordCacheMiss(level, operation string) {
|
||||
m.CacheOperationsTotal.WithLabelValues(level, operation).Inc()
|
||||
}
|
||||
|
||||
// RecordTokenRefresh 记录 Token 刷新操作
|
||||
func (m *SLOMetrics) RecordTokenRefresh(status string) {
|
||||
m.TokenRefreshTotal.WithLabelValues(status).Inc()
|
||||
}
|
||||
|
||||
// RecordAccountLock 记录账号锁定事件
|
||||
func (m *SLOMetrics) RecordAccountLock() {
|
||||
m.AccountLockTotal.Inc()
|
||||
}
|
||||
|
||||
// RecordAnomaly 记录异常检测事件
|
||||
func (m *SLOMetrics) RecordAnomaly(anomalyType string) {
|
||||
m.AnomalyDetectedTotal.WithLabelValues(anomalyType).Inc()
|
||||
}
|
||||
|
||||
// SetDBConnections 更新数据库连接池状态
|
||||
func (m *SLOMetrics) SetDBConnections(active, max float64) {
|
||||
m.DBConnectionsActive.Set(active)
|
||||
m.DBConnectionsMax.Set(max)
|
||||
}
|
||||
|
||||
// SetErrorBudgetBurnRate 设置错误预算燃烧率
|
||||
func (m *SLOMetrics) SetErrorBudgetBurnRate(slo string, burnRate float64) {
|
||||
m.ErrorBudgetBurnRate.WithLabelValues(slo).Set(burnRate)
|
||||
}
|
||||
396
internal/performance/benchmark_test.go
Normal file
396
internal/performance/benchmark_test.go
Normal file
@@ -0,0 +1,396 @@
|
||||
package performance
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"encoding/hex"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/user-management-system/internal/auth"
|
||||
"github.com/user-management-system/internal/cache"
|
||||
"github.com/user-management-system/internal/domain"
|
||||
"github.com/user-management-system/internal/repository"
|
||||
"golang.org/x/crypto/argon2"
|
||||
)
|
||||
|
||||
// =============================================================================
|
||||
// Password Hashing Benchmarks (Argon2id)
|
||||
// =============================================================================
|
||||
|
||||
func BenchmarkArgon2idHashing(b *testing.B) {
|
||||
password := []byte("TestPassword123!")
|
||||
salt := make([]byte, 16)
|
||||
rand.Read(salt)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_ = argon2.IDKey(password, salt, 5, 64*1024, 4, 32)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkArgon2idHashingParallel(b *testing.B) {
|
||||
password := []byte("TestPassword123!")
|
||||
salt := make([]byte, 16)
|
||||
rand.Read(salt)
|
||||
|
||||
b.ResetTimer()
|
||||
b.RunParallel(func(pb *testing.PB) {
|
||||
localSalt := make([]byte, 16)
|
||||
rand.Read(localSalt)
|
||||
for pb.Next() {
|
||||
_ = argon2.IDKey(password, localSalt, 5, 64*1024, 4, 32)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func BenchmarkArgon2idHashingDefaultParams(b *testing.B) {
|
||||
password := []byte("TestPassword123!")
|
||||
// Default params from our config: time=5, memory=64MB, threads=4
|
||||
salt := make([]byte, 16)
|
||||
rand.Read(salt)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_ = argon2.IDKey(password, salt, 5, 64*1024, 4, 32)
|
||||
}
|
||||
b.ReportMetric(64.0, "memory_MB")
|
||||
b.ReportMetric(5.0, "time_ops")
|
||||
b.ReportMetric(4.0, "threads")
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// JWT Benchmarks
|
||||
// =============================================================================
|
||||
|
||||
func BenchmarkJWTGenerateToken(b *testing.B) {
|
||||
jwtManager := auth.NewJWT("benchmark-secret-key-32bytes!", 2*time.Hour, 7*24*time.Hour)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _, _ = jwtManager.GenerateTokenPair(int64(i), "testuser")
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkJWTValidateToken(b *testing.B) {
|
||||
jwtManager := auth.NewJWT("benchmark-secret-key-32bytes!", 2*time.Hour, 7*24*time.Hour)
|
||||
token, _, _ := jwtManager.GenerateTokenPair(1, "testuser")
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _ = jwtManager.ValidateAccessToken(token)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkJWTGenerateAndValidate(b *testing.B) {
|
||||
jwtManager := auth.NewJWT("benchmark-secret-key-32bytes!", 2*time.Hour, 7*24*time.Hour)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
token, _, _ := jwtManager.GenerateTokenPair(int64(i), "testuser")
|
||||
jwtManager.ValidateAccessToken(token)
|
||||
}
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// TOTP Benchmarks
|
||||
// =============================================================================
|
||||
|
||||
func BenchmarkTOTPGenerateSecret(b *testing.B) {
|
||||
totpManager := auth.NewTOTPManager()
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _ = totpManager.GenerateSecret("testuser")
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkTOTPGenerateCurrentCode(b *testing.B) {
|
||||
totpManager := auth.NewTOTPManager()
|
||||
secret := make([]byte, 20)
|
||||
rand.Read(secret)
|
||||
_ = secret // Use the secret
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _ = totpManager.GenerateCurrentCode(base32StdSecret())
|
||||
}
|
||||
}
|
||||
|
||||
func base32StdSecret() string {
|
||||
b := make([]byte, 20)
|
||||
rand.Read(b)
|
||||
return "JBSWY3DPEHPK3PXP" // Example base32 secret
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// Recovery Code Benchmarks
|
||||
// =============================================================================
|
||||
|
||||
func BenchmarkRecoveryCodeHashing(b *testing.B) {
|
||||
codes := generateTestCodes(10)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
for _, code := range codes {
|
||||
_, _ = auth.HashRecoveryCode(code)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkRecoveryCodeVerification(b *testing.B) {
|
||||
codes := generateTestCodes(10)
|
||||
hashedCodes := make([]string, len(codes))
|
||||
for i, code := range codes {
|
||||
h, _ := auth.HashRecoveryCode(code)
|
||||
hashedCodes[i] = h
|
||||
}
|
||||
|
||||
testCode := codes[5] // Use the 6th code
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _ = auth.VerifyRecoveryCode(testCode, hashedCodes)
|
||||
}
|
||||
}
|
||||
|
||||
func generateTestCodes(count int) []string {
|
||||
codes := make([]string, count)
|
||||
for i := 0; i < count; i++ {
|
||||
b := make([]byte, RecoveryCodeLength*2)
|
||||
rand.Read(b)
|
||||
encoded := base32Encode(b)
|
||||
codes[i] = formatRecoveryCode(encoded[:10])
|
||||
}
|
||||
return codes
|
||||
}
|
||||
|
||||
const RecoveryCodeLength = 10 // from totp.go
|
||||
|
||||
func base32Encode(b []byte) string {
|
||||
const alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567"
|
||||
result := make([]byte, (len(b)*8+4)/5)
|
||||
for i := 0; i < len(result); i++ {
|
||||
var val uint32
|
||||
var bits int
|
||||
for j := 0; j < 5 && i*5+j < len(b)*8; j++ {
|
||||
if bits < 5 {
|
||||
val = (val << bits) | uint32(b[i*5/8]>>(8-bits))&0xFF
|
||||
bits += 8
|
||||
}
|
||||
}
|
||||
result[i] = alphabet[(val>>(bits-5))&0x1F]
|
||||
}
|
||||
return string(result)
|
||||
}
|
||||
|
||||
func formatRecoveryCode(s string) string {
|
||||
if len(s) >= 10 {
|
||||
return s[:5] + "-" + s[5:10]
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// Cache Benchmarks
|
||||
// =============================================================================
|
||||
|
||||
func BenchmarkL1CacheGet(b *testing.B) {
|
||||
l1Cache := cache.NewL1Cache()
|
||||
l1Cache.Set("test-key", "test-value", 10*time.Minute)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _ = l1Cache.Get("test-key")
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkL1CacheSet(b *testing.B) {
|
||||
l1Cache := cache.NewL1Cache()
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
key := "test-key-" + hex.EncodeToString([]byte{byte(i)})
|
||||
l1Cache.Set(key, "test-value", 10*time.Minute)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkL1CacheGetMiss(b *testing.B) {
|
||||
l1Cache := cache.NewL1Cache()
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _ = l1Cache.Get("non-existent-key")
|
||||
}
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// Database Benchmarks
|
||||
// =============================================================================
|
||||
|
||||
func BenchmarkUserRepositoryCreate(b *testing.B) {
|
||||
db := setupBenchmarkDB(b)
|
||||
repo := repository.NewUserRepository(db)
|
||||
ctx := context.Background()
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
user := &domain.User{
|
||||
Username: "benchuser" + hex.EncodeToString([]byte{byte(i)}),
|
||||
Email: domain.StrPtr("bench@example.com"),
|
||||
Password: "hash",
|
||||
Status: domain.UserStatusActive,
|
||||
}
|
||||
repo.Create(ctx, user)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkUserRepositoryGetByID(b *testing.B) {
|
||||
db := setupBenchmarkDB(b)
|
||||
repo := repository.NewUserRepository(db)
|
||||
ctx := context.Background()
|
||||
|
||||
// Pre-create user
|
||||
user := &domain.User{
|
||||
Username: "benchuser",
|
||||
Email: domain.StrPtr("bench@example.com"),
|
||||
Password: "hash",
|
||||
Status: domain.UserStatusActive,
|
||||
}
|
||||
repo.Create(ctx, user)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _ = repo.GetByID(ctx, user.ID)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkUserRepositoryGetByUsername(b *testing.B) {
|
||||
db := setupBenchmarkDB(b)
|
||||
repo := repository.NewUserRepository(db)
|
||||
ctx := context.Background()
|
||||
|
||||
// Pre-create user
|
||||
user := &domain.User{
|
||||
Username: "benchuser",
|
||||
Email: domain.StrPtr("bench@example.com"),
|
||||
Password: "hash",
|
||||
Status: domain.UserStatusActive,
|
||||
}
|
||||
repo.Create(ctx, user)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _ = repo.GetByUsername(ctx, "benchuser")
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkUserRepositoryList(b *testing.B) {
|
||||
db := setupBenchmarkDB(b)
|
||||
repo := repository.NewUserRepository(db)
|
||||
ctx := context.Background()
|
||||
|
||||
// Pre-create users
|
||||
for i := 0; i < 100; i++ {
|
||||
user := &domain.User{
|
||||
Username: "benchuser" + hex.EncodeToString([]byte{byte(i)}),
|
||||
Email: domain.StrPtr("bench@example.com"),
|
||||
Password: "hash",
|
||||
Status: domain.UserStatusActive,
|
||||
}
|
||||
repo.Create(ctx, user)
|
||||
}
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_, _, _ = repo.List(ctx, 0, 100)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkUserRepositoryUpdate(b *testing.B) {
|
||||
db := setupBenchmarkDB(b)
|
||||
repo := repository.NewUserRepository(db)
|
||||
ctx := context.Background()
|
||||
|
||||
// Pre-create user
|
||||
user := &domain.User{
|
||||
Username: "benchuser",
|
||||
Email: domain.StrPtr("bench@example.com"),
|
||||
Password: "hash",
|
||||
Status: domain.UserStatusActive,
|
||||
}
|
||||
repo.Create(ctx, user)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
user.Nickname = "Updated Nickname"
|
||||
repo.Update(ctx, user)
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkRoleRepositoryCreate(b *testing.B) {
|
||||
db := setupBenchmarkDB(b)
|
||||
repo := repository.NewRoleRepository(db)
|
||||
ctx := context.Background()
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
role := &domain.Role{
|
||||
Name: "benchrole" + hex.EncodeToString([]byte{byte(i)}),
|
||||
Code: "benchrole" + hex.EncodeToString([]byte{byte(i)}),
|
||||
}
|
||||
repo.Create(ctx, role)
|
||||
}
|
||||
}
|
||||
|
||||
// HMAC benchmarks removed - ComputeHMAC is not exported from auth package
|
||||
// ConstantTimeCompare benchmarks removed - it's internal to the auth package
|
||||
|
||||
// =============================================================================
|
||||
// Concurrency Stress Tests
|
||||
// =============================================================================
|
||||
|
||||
func BenchmarkConcurrentUserCreation(b *testing.B) {
|
||||
db := setupBenchmarkDB(b)
|
||||
repo := repository.NewUserRepository(db)
|
||||
ctx := context.Background()
|
||||
|
||||
b.ResetTimer()
|
||||
b.RunParallel(func(pb *testing.PB) {
|
||||
i := 0
|
||||
for pb.Next() {
|
||||
user := &domain.User{
|
||||
Username: "benchuser" + hex.EncodeToString([]byte{byte(i % 256)}),
|
||||
Email: domain.StrPtr("bench@example.com"),
|
||||
Password: "hash",
|
||||
Status: domain.UserStatusActive,
|
||||
}
|
||||
repo.Create(ctx, user)
|
||||
i++
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func BenchmarkConcurrentCacheAccess(b *testing.B) {
|
||||
l1Cache := cache.NewL1Cache()
|
||||
|
||||
// Pre-populate cache
|
||||
for i := 0; i < 100; i++ {
|
||||
key := "test-key-" + hex.EncodeToString([]byte{byte(i)})
|
||||
l1Cache.Set(key, "test-value", 10*time.Minute)
|
||||
}
|
||||
|
||||
b.ResetTimer()
|
||||
b.RunParallel(func(pb *testing.PB) {
|
||||
i := 0
|
||||
for pb.Next() {
|
||||
key := "test-key-" + hex.EncodeToString([]byte{byte(i % 100)})
|
||||
l1Cache.Get(key)
|
||||
i++
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// Helper function - setupBenchmarkDB is defined in performance_test.go
|
||||
// =============================================================================
|
||||
Reference in New Issue
Block a user