7efa8b54c4
新增 DB/Redis 连接池配置与校验，并补充单测 网关请求体大小限制与 413 处理 HTTP/req 客户端池化并调整上游连接池默认值 并发槽位改为 ZSET+Lua 与指数退避 用量统计改 SQL 聚合并新增索引迁移 计费缓存写入改工作池并补测试/基准 测试: 在 backend/ 下运行 go test ./...
33 lines
881 B
Go
33 lines
881 B
Go
package infrastructure
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/config"
|
|
)
|
|
|
|
type dbPoolSettings struct {
|
|
MaxOpenConns int
|
|
MaxIdleConns int
|
|
ConnMaxLifetime time.Duration
|
|
ConnMaxIdleTime time.Duration
|
|
}
|
|
|
|
func buildDBPoolSettings(cfg *config.Config) dbPoolSettings {
|
|
return dbPoolSettings{
|
|
MaxOpenConns: cfg.Database.MaxOpenConns,
|
|
MaxIdleConns: cfg.Database.MaxIdleConns,
|
|
ConnMaxLifetime: time.Duration(cfg.Database.ConnMaxLifetimeMinutes) * time.Minute,
|
|
ConnMaxIdleTime: time.Duration(cfg.Database.ConnMaxIdleTimeMinutes) * time.Minute,
|
|
}
|
|
}
|
|
|
|
func applyDBPoolSettings(db *sql.DB, cfg *config.Config) {
|
|
settings := buildDBPoolSettings(cfg)
|
|
db.SetMaxOpenConns(settings.MaxOpenConns)
|
|
db.SetMaxIdleConns(settings.MaxIdleConns)
|
|
db.SetConnMaxLifetime(settings.ConnMaxLifetime)
|
|
db.SetConnMaxIdleTime(settings.ConnMaxIdleTime)
|
|
}
|