mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-01 22:46:38 +08:00
ryan
2f60329886
代码质量全量清理，零行为变化：golangci 扩展集 13 类 linter（gosec/modernize/perfsprint/canonicalheader/usestdlibvars/wastedassign/intrange/errorlint/forcetypeassert/recvcheck/exhaustive/unparam）全量修复，测试代码质量（testifylint/thelper/usetesting）25→0，frpc 进程生命周期真 bug（进程组击杀）、全仓 go test -race 6 类数据竞争（含 1 个生产竞争）、SPDX license 头补齐 131 文件、前端测试套件 next-intl 迁移后 44 失败→全绿、过期 swagger 文档重新生成、pnpm-workspace 构建审批。 Experiments: #2-#17, #18, #20, #21, #23 Metric: total_issues 108 → 8 (-92.6%)
115 lines
2.9 KiB
Go
115 lines
2.9 KiB
Go
// Copyright 2025 linux.do
|
||
// Copyright 2026 Arctel.net
|
||
// SPDX-License-Identifier: Apache-2.0
|
||
|
||
package task
|
||
|
||
import (
|
||
"github.com/Rain-kl/Wavelet/internal/infra/config"
|
||
"github.com/hibiken/asynq"
|
||
"github.com/redis/go-redis/v9"
|
||
"github.com/redis/go-redis/v9/maintnotifications"
|
||
)
|
||
|
||
type redisClientConnOpt struct {
|
||
options redis.Options
|
||
}
|
||
|
||
func (opt redisClientConnOpt) MakeRedisClient() any {
|
||
return redis.NewClient(&opt.options)
|
||
}
|
||
|
||
type redisClusterConnOpt struct {
|
||
options redis.ClusterOptions
|
||
}
|
||
|
||
func (opt redisClusterConnOpt) MakeRedisClient() any {
|
||
return redis.NewClusterClient(&opt.options)
|
||
}
|
||
|
||
type redisFailoverConnOpt struct {
|
||
options redis.FailoverOptions
|
||
maintNotificationsEnabled bool
|
||
}
|
||
|
||
func (opt redisFailoverConnOpt) MakeRedisClient() any {
|
||
client := redis.NewFailoverClient(&opt.options)
|
||
// go-redis v9.16 does not expose maintenance notification settings on
|
||
// FailoverOptions, so apply the configured mode before the client is used.
|
||
client.Options().MaintNotificationsConfig = maintNotificationsConfig(opt.maintNotificationsEnabled)
|
||
return client
|
||
}
|
||
|
||
func maintNotificationsConfig(enabled bool) *maintnotifications.Config {
|
||
mode := maintnotifications.ModeDisabled
|
||
if enabled {
|
||
mode = maintnotifications.ModeAuto
|
||
}
|
||
return &maintnotifications.Config{Mode: mode}
|
||
}
|
||
|
||
// RedisOpt asynq Redis 连接配置（兼容 Standalone/Sentinel/Cluster）
|
||
var RedisOpt asynq.RedisConnOpt
|
||
|
||
// AsynqClient asynq 客户端，用于任务入队
|
||
var AsynqClient *asynq.Client
|
||
|
||
func init() {
|
||
RedisOpt = NewRedisConnOpt()
|
||
AsynqClient = asynq.NewClient(RedisOpt)
|
||
}
|
||
|
||
// NewRedisConnOpt 根据配置返回对应的 asynq Redis 连接选项
|
||
func NewRedisConnOpt() asynq.RedisConnOpt {
|
||
cfg := config.Config.Redis
|
||
addrs := cfg.Addrs
|
||
|
||
if cfg.ClusterMode {
|
||
return redisClusterConnOpt{
|
||
options: redis.ClusterOptions{
|
||
Addrs: addrs,
|
||
Username: cfg.Username,
|
||
Password: cfg.Password,
|
||
MaintNotificationsConfig: maintNotificationsConfig(cfg.MaintNotifications),
|
||
},
|
||
}
|
||
}
|
||
|
||
if cfg.MasterName != "" {
|
||
return redisFailoverConnOpt{
|
||
maintNotificationsEnabled: cfg.MaintNotifications,
|
||
options: redis.FailoverOptions{
|
||
MasterName: cfg.MasterName,
|
||
SentinelAddrs: addrs,
|
||
Username: cfg.Username,
|
||
Password: cfg.Password,
|
||
DB: cfg.DB,
|
||
},
|
||
}
|
||
}
|
||
|
||
addr := "localhost:6379"
|
||
if len(addrs) > 0 {
|
||
addr = addrs[0]
|
||
}
|
||
return redisClientConnOpt{
|
||
options: redis.Options{
|
||
Addr: addr,
|
||
Username: cfg.Username,
|
||
Password: cfg.Password,
|
||
DB: cfg.DB,
|
||
PoolSize: cfg.PoolSize,
|
||
MaintNotificationsConfig: maintNotificationsConfig(cfg.MaintNotifications),
|
||
},
|
||
}
|
||
}
|
||
|
||
// PrefixedQueue 返回带前缀的队列名，用于 Cluster 模式隔离
|
||
func PrefixedQueue(queue string) string {
|
||
prefix := config.Config.Redis.KeyPrefix
|
||
if prefix == "" {
|
||
return queue
|
||
}
|
||
return prefix + queue
|
||
}
|