95 lines
3.4 KiB
Go
95 lines
3.4 KiB
Go
package queue
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
"ycc-server/app/main/model"
|
|
"ycc-server/pkg/lzkit/lzUtils"
|
|
|
|
"ycc-server/app/main/api/internal/svc"
|
|
"ycc-server/app/main/api/internal/types"
|
|
|
|
"github.com/hibiken/asynq"
|
|
pkgerrors "github.com/pkg/errors"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
)
|
|
|
|
type UnfreezeCommissionHandler struct {
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewUnfreezeCommissionHandler(svcCtx *svc.ServiceContext) *UnfreezeCommissionHandler {
|
|
return &UnfreezeCommissionHandler{
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *UnfreezeCommissionHandler) ProcessTask(ctx context.Context, t *asynq.Task) error {
|
|
var payload types.MsgUnfreezeCommissionPayload
|
|
if err := json.Unmarshal(t.Payload(), &payload); err != nil {
|
|
return fmt.Errorf("解析解冻任务负载失败: %w", err)
|
|
}
|
|
|
|
// 1. 查询冻结任务
|
|
freezeTask, err := l.svcCtx.AgentFreezeTaskModel.FindOne(ctx, payload.FreezeTaskId)
|
|
if err != nil {
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
logx.Errorf("解冻任务失败，冻结任务不存在: freezeTaskId=%d", payload.FreezeTaskId)
|
|
return asynq.SkipRetry // 任务不存在，跳过重试
|
|
}
|
|
return fmt.Errorf("查询冻结任务失败: freezeTaskId=%d, err=%w", payload.FreezeTaskId, err)
|
|
}
|
|
|
|
// 2. 检查任务状态
|
|
if freezeTask.Status != 1 {
|
|
logx.Infof("解冻任务跳过，任务已处理: freezeTaskId=%d, status=%d", payload.FreezeTaskId, freezeTask.Status)
|
|
return nil // 任务已处理，不重试
|
|
}
|
|
|
|
// 3. 检查解冻时间是否已到
|
|
if time.Now().Before(freezeTask.UnfreezeTime) {
|
|
logx.Infof("解冻任务跳过，未到解冻时间: freezeTaskId=%d, unfreezeTime=%v", payload.FreezeTaskId, freezeTask.UnfreezeTime)
|
|
// 重新发送延迟任务
|
|
if err := l.svcCtx.AsynqService.SendUnfreezeTask(payload.FreezeTaskId, freezeTask.UnfreezeTime); err != nil {
|
|
logx.Errorf("重新发送解冻任务失败: freezeTaskId=%d, err=%v", payload.FreezeTaskId, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 4. 使用事务处理解冻
|
|
err = l.svcCtx.AgentFreezeTaskModel.Trans(ctx, func(transCtx context.Context, session sqlx.Session) error {
|
|
// 4.1 更新冻结任务状态
|
|
freezeTask.Status = 2 // 已解冻
|
|
freezeTask.ActualUnfreezeTime = lzUtils.TimeToNullTime(time.Now())
|
|
if updateErr := l.svcCtx.AgentFreezeTaskModel.UpdateWithVersion(transCtx, session, freezeTask); updateErr != nil {
|
|
return pkgerrors.Wrapf(updateErr, "更新冻结任务状态失败")
|
|
}
|
|
|
|
// 4.2 更新钱包（解冻余额）
|
|
wallet, walletErr := l.svcCtx.AgentWalletModel.FindOneByAgentId(transCtx, freezeTask.AgentId)
|
|
if walletErr != nil {
|
|
return pkgerrors.Wrapf(walletErr, "查询钱包失败, agentId: %d", freezeTask.AgentId)
|
|
}
|
|
|
|
wallet.FrozenBalance -= freezeTask.FreezeAmount
|
|
wallet.Balance += freezeTask.FreezeAmount
|
|
if updateWalletErr := l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet); updateWalletErr != nil {
|
|
return pkgerrors.Wrapf(updateWalletErr, "更新钱包失败")
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
logx.Errorf("解冻任务处理失败: freezeTaskId=%d, err=%v", payload.FreezeTaskId, err)
|
|
return fmt.Errorf("解冻任务处理失败: freezeTaskId=%d, err=%w", payload.FreezeTaskId, err)
|
|
}
|
|
|
|
logx.Infof("解冻任务处理成功: freezeTaskId=%d, agentId=%d, amount=%.2f", payload.FreezeTaskId, freezeTask.AgentId, freezeTask.FreezeAmount)
|
|
return nil
|
|
}
|