yangjianbo
682f546c0e
- errcheck: 修复类型断言未检查返回值的问题 - pool.go: 添加 sync.Map 类型断言安全检查 - req_client_pool.go: 添加 sync.Map 类型断言安全检查 - concurrency_cache_benchmark_test.go: 显式忽略断言返回值 - gateway_service.go: 显式忽略 WriteString 返回值 - gofmt: 修复代码格式问题 - redis.go: 注释对齐 - api_key_repo.go: 结构体字段对齐 - concurrency_cache.go: 字段对齐 - http_upstream.go: 注释对齐 - unused: 删除未使用的代码 - user_repo.go: 删除未使用的 sql 字段 - usage_service.go: 删除未使用的 calculateStats 函数 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
40 lines
1.5 KiB
Go
40 lines
1.5 KiB
Go
package infrastructure
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/config"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
// InitRedis 初始化 Redis 客户端
|
|
//
|
|
// 性能优化说明：
|
|
// 原实现使用 go-redis 默认配置，未设置连接池和超时参数：
|
|
// 1. 默认连接池大小可能不足以支撑高并发
|
|
// 2. 无超时控制可能导致慢操作阻塞
|
|
//
|
|
// 新实现支持可配置的连接池和超时参数：
|
|
// 1. PoolSize: 控制最大并发连接数（默认 128）
|
|
// 2. MinIdleConns: 保持最小空闲连接，减少冷启动延迟（默认 10）
|
|
// 3. DialTimeout/ReadTimeout/WriteTimeout: 精确控制各阶段超时
|
|
func InitRedis(cfg *config.Config) *redis.Client {
|
|
return redis.NewClient(buildRedisOptions(cfg))
|
|
}
|
|
|
|
// buildRedisOptions 构建 Redis 连接选项
|
|
// 从配置文件读取连接池和超时参数，支持生产环境调优
|
|
func buildRedisOptions(cfg *config.Config) *redis.Options {
|
|
return &redis.Options{
|
|
Addr: cfg.Redis.Address(),
|
|
Password: cfg.Redis.Password,
|
|
DB: cfg.Redis.DB,
|
|
DialTimeout: time.Duration(cfg.Redis.DialTimeoutSeconds) * time.Second, // 建连超时
|
|
ReadTimeout: time.Duration(cfg.Redis.ReadTimeoutSeconds) * time.Second, // 读取超时
|
|
WriteTimeout: time.Duration(cfg.Redis.WriteTimeoutSeconds) * time.Second, // 写入超时
|
|
PoolSize: cfg.Redis.PoolSize, // 连接池大小
|
|
MinIdleConns: cfg.Redis.MinIdleConns, // 最小空闲连接
|
|
}
|
|
}
|