This commit is contained in:
2025-12-30 17:55:00 +08:00
parent 1150c43219
commit 12659c1d14
15 changed files with 704 additions and 49 deletions

View File

@@ -2,9 +2,9 @@ package queue
import (
"context"
"fmt"
"tydata-server/app/main/api/internal/svc"
"tydata-server/app/main/api/internal/types"
"fmt"
"github.com/hibiken/asynq"
)
@@ -24,17 +24,28 @@ func NewCronJob(ctx context.Context, svcCtx *svc.ServiceContext) *CronJob {
func (l *CronJob) Register() *asynq.ServeMux {
redisClientOpt := asynq.RedisClientOpt{Addr: l.svcCtx.Config.CacheRedis[0].Host, Password: l.svcCtx.Config.CacheRedis[0].Pass}
scheduler := asynq.NewScheduler(redisClientOpt, nil)
// 注册清理查询数据任务
task := asynq.NewTask(types.MsgCleanQueryData, nil, nil)
_, err := scheduler.Register(TASKTIME, task)
if err != nil {
panic(fmt.Sprintf("定时任务注册失败:%v", err))
}
// 注释掉原来的佣金解冻定时任务,因为现在改为按需处理单个佣金记录
// unfreezeTask := asynq.NewTask(types.MsgUnfreezeCommission, nil, nil)
// _, err = scheduler.Register("0 4 * * *", unfreezeTask)
// if err != nil {
// panic(fmt.Sprintf("佣金解冻任务注册失败:%v", err))
// }
scheduler.Start()
fmt.Println("定时任务启动!!!")
mux := asynq.NewServeMux()
mux.Handle(types.MsgPaySuccessQuery, NewPaySuccessNotifyUserHandler(l.svcCtx))
mux.Handle(types.MsgCleanQueryData, NewCleanQueryDataHandler(l.svcCtx))
mux.Handle(types.MsgUnfreezeCommission, NewUnfreezeCommissionHandler(l.svcCtx))
return mux
}