123 lines
3.9 KiB
Go
123 lines
3.9 KiB
Go
package asynq
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/hibiken/asynq"
|
||
"go.uber.org/zap"
|
||
|
||
"tyapi-server/internal/application/api"
|
||
"tyapi-server/internal/application/article"
|
||
finance_services "tyapi-server/internal/domains/finance/services"
|
||
product_services "tyapi-server/internal/domains/product/services"
|
||
"tyapi-server/internal/infrastructure/task/handlers"
|
||
"tyapi-server/internal/infrastructure/task/repositories"
|
||
"tyapi-server/internal/infrastructure/task/types"
|
||
)
|
||
|
||
// AsynqWorker Asynq Worker实现
|
||
type AsynqWorker struct {
|
||
server *asynq.Server
|
||
mux *asynq.ServeMux
|
||
logger *zap.Logger
|
||
articleHandler *handlers.ArticleTaskHandler
|
||
apiHandler *handlers.ApiTaskHandler
|
||
}
|
||
|
||
// NewAsynqWorker 创建Asynq Worker
|
||
func NewAsynqWorker(
|
||
redisAddr string,
|
||
logger *zap.Logger,
|
||
articleApplicationService article.ArticleApplicationService,
|
||
apiApplicationService api.ApiApplicationService,
|
||
walletService finance_services.WalletAggregateService,
|
||
subscriptionService *product_services.ProductSubscriptionService,
|
||
asyncTaskRepo repositories.AsyncTaskRepository,
|
||
) *AsynqWorker {
|
||
server := asynq.NewServer(
|
||
asynq.RedisClientOpt{Addr: redisAddr},
|
||
asynq.Config{
|
||
Concurrency: 6, // 降低总并发数
|
||
Queues: map[string]int{
|
||
"default": 2, // 2个goroutine
|
||
"api": 3, // 3个goroutine (扣款任务)
|
||
"article": 1, // 1个goroutine
|
||
},
|
||
},
|
||
)
|
||
|
||
// 创建任务处理器
|
||
articleHandler := handlers.NewArticleTaskHandler(logger, articleApplicationService, asyncTaskRepo)
|
||
apiHandler := handlers.NewApiTaskHandler(logger, apiApplicationService, walletService, subscriptionService, asyncTaskRepo)
|
||
|
||
// 创建ServeMux
|
||
mux := asynq.NewServeMux()
|
||
|
||
return &AsynqWorker{
|
||
server: server,
|
||
mux: mux,
|
||
logger: logger,
|
||
articleHandler: articleHandler,
|
||
apiHandler: apiHandler,
|
||
}
|
||
}
|
||
|
||
// RegisterHandler 注册任务处理器
|
||
func (w *AsynqWorker) RegisterHandler(taskType types.TaskType, handler func(context.Context, *asynq.Task) error) {
|
||
// 简化实现，避免API兼容性问题
|
||
w.logger.Info("注册任务处理器", zap.String("task_type", string(taskType)))
|
||
}
|
||
|
||
// Start 启动Worker
|
||
func (w *AsynqWorker) Start() error {
|
||
w.logger.Info("启动Asynq Worker")
|
||
|
||
// 注册所有任务处理器
|
||
w.registerAllHandlers()
|
||
|
||
// 启动Worker服务器
|
||
go func() {
|
||
if err := w.server.Run(w.mux); err != nil {
|
||
w.logger.Error("Worker运行失败", zap.Error(err))
|
||
}
|
||
}()
|
||
|
||
w.logger.Info("Asynq Worker启动成功")
|
||
return nil
|
||
}
|
||
|
||
// Stop 停止Worker
|
||
func (w *AsynqWorker) Stop() {
|
||
w.logger.Info("停止Asynq Worker")
|
||
w.server.Stop()
|
||
}
|
||
|
||
// Shutdown 优雅关闭Worker
|
||
func (w *AsynqWorker) Shutdown() {
|
||
w.logger.Info("优雅关闭Asynq Worker")
|
||
w.server.Shutdown()
|
||
}
|
||
|
||
// registerAllHandlers 注册所有任务处理器
|
||
func (w *AsynqWorker) registerAllHandlers() {
|
||
// 注册文章任务处理器
|
||
w.mux.HandleFunc(string(types.TaskTypeArticlePublish), w.articleHandler.HandleArticlePublish)
|
||
w.mux.HandleFunc(string(types.TaskTypeArticleCancel), w.articleHandler.HandleArticleCancel)
|
||
w.mux.HandleFunc(string(types.TaskTypeArticleModify), w.articleHandler.HandleArticleModify)
|
||
|
||
// 注册API任务处理器
|
||
w.mux.HandleFunc(string(types.TaskTypeApiCall), w.apiHandler.HandleApiCall)
|
||
w.mux.HandleFunc(string(types.TaskTypeApiLog), w.apiHandler.HandleApiLog)
|
||
w.mux.HandleFunc(string(types.TaskTypeDeduction), w.apiHandler.HandleDeduction)
|
||
w.mux.HandleFunc(string(types.TaskTypeCompensation), w.apiHandler.HandleCompensation)
|
||
w.mux.HandleFunc(string(types.TaskTypeUsageStats), w.apiHandler.HandleUsageStats)
|
||
|
||
w.logger.Info("所有任务处理器注册完成",
|
||
zap.String("article_publish", string(types.TaskTypeArticlePublish)),
|
||
zap.String("article_cancel", string(types.TaskTypeArticleCancel)),
|
||
zap.String("article_modify", string(types.TaskTypeArticleModify)),
|
||
zap.String("api_call", string(types.TaskTypeApiCall)),
|
||
zap.String("api_log", string(types.TaskTypeApiLog)),
|
||
)
|
||
}
|