huang
984ccccc63
在项目宪章中新增第九条原则"数据库设计原则"，明确禁止使用数据库外键约束和ORM关联标签。 主要变更： - 新增原则IX：数据库设计原则（Database Design Principles） - 强制要求：数据库表不得使用外键约束 - 强制要求：GORM模型不得使用ORM关联标签（foreignKey、hasMany等） - 强制要求：表关系必须通过ID字段手动维护 - 强制要求：关联数据查询必须显式编写，避免ORM魔法 - 强制要求：时间字段由GORM处理，不使用数据库触发器 设计理念： - 提升业务逻辑灵活性（无数据库约束限制） - 优化高并发性能（无外键检查开销） - 增强代码可读性（显式查询，无隐式预加载） - 简化数据库架构和迁移流程 - 支持分布式和微服务场景 版本升级：2.3.0 → 2.4.0（MINOR）
87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
package queue
|
|
|
|
import (
|
|
"github.com/break/junhong_cmp_fiber/pkg/config"
|
|
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
|
"github.com/hibiken/asynq"
|
|
"github.com/redis/go-redis/v9"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// Server Asynq Worker 服务器
|
|
type Server struct {
|
|
server *asynq.Server
|
|
logger *zap.Logger
|
|
}
|
|
|
|
// NewServer 创建新的 Asynq 服务器
|
|
func NewServer(redisClient *redis.Client, queueCfg *config.QueueConfig, logger *zap.Logger) *Server {
|
|
// 从 Redis 客户端获取配置
|
|
opts := redisClient.Options()
|
|
|
|
// 解析队列优先级配置
|
|
queues := ParseQueueConfig(queueCfg)
|
|
|
|
// 设置并发数
|
|
concurrency := queueCfg.Concurrency
|
|
if concurrency <= 0 {
|
|
concurrency = constants.DefaultConcurrency
|
|
}
|
|
|
|
// 创建 Asynq 服务器配置
|
|
asynqServer := asynq.NewServer(
|
|
asynq.RedisClientOpt{
|
|
Addr: opts.Addr,
|
|
Password: opts.Password,
|
|
DB: opts.DB,
|
|
},
|
|
asynq.Config{
|
|
// 并发数
|
|
Concurrency: concurrency,
|
|
// 队列优先级配置
|
|
Queues: queues,
|
|
// 重试延迟函数（指数退避）
|
|
RetryDelayFunc: asynq.DefaultRetryDelayFunc,
|
|
// 是否记录详细日志
|
|
LogLevel: asynq.WarnLevel,
|
|
},
|
|
)
|
|
|
|
return &Server{
|
|
server: asynqServer,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// Start 启动 Worker 服务器
|
|
func (s *Server) Start(mux *asynq.ServeMux) error {
|
|
s.logger.Info("Worker 服务器启动中...")
|
|
|
|
if err := s.server.Start(mux); err != nil {
|
|
s.logger.Error("Worker 服务器启动失败", zap.Error(err))
|
|
return err
|
|
}
|
|
|
|
s.logger.Info("Worker 服务器启动成功")
|
|
return nil
|
|
}
|
|
|
|
// Shutdown 优雅关闭服务器
|
|
func (s *Server) Shutdown() {
|
|
s.logger.Info("Worker 服务器关闭中...")
|
|
s.server.Shutdown()
|
|
s.logger.Info("Worker 服务器已关闭")
|
|
}
|
|
|
|
// Run 启动并阻塞运行（用于主函数）
|
|
func (s *Server) Run(mux *asynq.ServeMux) error {
|
|
s.logger.Info("Worker 服务器启动中...")
|
|
|
|
if err := s.server.Run(mux); err != nil {
|
|
s.logger.Error("Worker 服务器运行失败", zap.Error(err))
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|