45 lines
1.5 KiB
Go
45 lines
1.5 KiB
Go
package task
|
||
|
||
import (
|
||
"tyapi-server/internal/infrastructure/task/implementations/asynq"
|
||
"tyapi-server/internal/infrastructure/task/interfaces"
|
||
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
// TaskFactory 任务工厂
|
||
type TaskFactory struct{}
|
||
|
||
// NewTaskFactory 创建任务工厂
|
||
func NewTaskFactory() *TaskFactory {
|
||
return &TaskFactory{}
|
||
}
|
||
|
||
// CreateApiTaskQueue 创建API任务队列
|
||
func (f *TaskFactory) CreateApiTaskQueue(redisAddr string, logger interface{}) interfaces.ApiTaskQueue {
|
||
// 这里可以根据配置选择不同的实现
|
||
// 目前使用Asynq实现
|
||
return asynq.NewAsynqApiTaskQueue(redisAddr, logger.(*zap.Logger))
|
||
}
|
||
|
||
// CreateArticleTaskQueue 创建文章任务队列
|
||
func (f *TaskFactory) CreateArticleTaskQueue(redisAddr string, logger interface{}) interfaces.ArticleTaskQueue {
|
||
// 这里可以根据配置选择不同的实现
|
||
// 目前使用Asynq实现
|
||
return asynq.NewAsynqArticleTaskQueue(redisAddr, logger.(*zap.Logger))
|
||
}
|
||
|
||
// NewApiTaskQueue 创建API任务队列（包级别函数）
|
||
func NewApiTaskQueue(redisAddr string, logger *zap.Logger) interfaces.ApiTaskQueue {
|
||
return asynq.NewAsynqApiTaskQueue(redisAddr, logger)
|
||
}
|
||
|
||
// NewAsynqClient 创建Asynq客户端（包级别函数）
|
||
func NewAsynqClient(redisAddr string, scheduledTaskRepo interface{}, logger *zap.Logger) *asynq.AsynqClient {
|
||
return asynq.NewAsynqClient(redisAddr, logger)
|
||
}
|
||
|
||
// NewArticleTaskQueue 创建文章任务队列（包级别函数）
|
||
func NewArticleTaskQueue(redisAddr string, logger *zap.Logger) interfaces.ArticleTaskQueue {
|
||
return asynq.NewAsynqArticleTaskQueue(redisAddr, logger)
|
||
} |