50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
|
package queue
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"bdrp-server/app/main/api/internal/service"
|
|||
|
|
"bdrp-server/app/main/api/internal/svc"
|
|||
|
|
"bdrp-server/app/main/api/internal/types"
|
|||
|
|
|
|||
|
|
"github.com/hibiken/asynq"
|
|||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type AgentMembershipExpireHandleHandler struct {
|
|||
|
|
svcCtx *svc.ServiceContext
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewAgentMembershipExpireHandleHandler(svcCtx *svc.ServiceContext) *AgentMembershipExpireHandleHandler {
|
|||
|
|
return &AgentMembershipExpireHandleHandler{
|
|||
|
|
svcCtx: svcCtx,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (l *AgentMembershipExpireHandleHandler) ProcessTask(ctx context.Context, t *asynq.Task) error {
|
|||
|
|
var payload types.MsgAgentMembershipExpireHandlePayload
|
|||
|
|
if err := json.Unmarshal(t.Payload(), &payload); err != nil {
|
|||
|
|
logx.Errorf("解析会员到期处理任务payload失败: %v", err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
if payload.AgentID <= 0 {
|
|||
|
|
return fmt.Errorf("无效代理ID: %d", payload.AgentID)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
now := time.Now()
|
|||
|
|
downgraded, err := service.DowngradeExpiredMembership(ctx, l.svcCtx.AgentModel, payload.AgentID, now)
|
|||
|
|
if err != nil {
|
|||
|
|
logx.Errorf("会员到期处理失败，代理ID: %d, 错误: %v", payload.AgentID, err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
if downgraded {
|
|||
|
|
logx.Infof("会员到期处理成功，代理ID: %d 已降级为普通代理", payload.AgentID)
|
|||
|
|
} else {
|
|||
|
|
logx.Infof("会员到期处理跳过，代理ID: %d 当前无需降级", payload.AgentID)
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|