f

app/main/api/internal/queue/agentMembershipExpireScan.go

@@ -0,0 +1,68 @@
+package queue
+
+import (
+	"context"
+	"time"
+
+	"bdrp-server/app/main/api/internal/service"
+	"bdrp-server/app/main/api/internal/svc"
+
+	"github.com/hibiken/asynq"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type AgentMembershipExpireScanHandler struct {
+	svcCtx *svc.ServiceContext
+}
+
+func NewAgentMembershipExpireScanHandler(svcCtx *svc.ServiceContext) *AgentMembershipExpireScanHandler {
+	return &AgentMembershipExpireScanHandler{
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AgentMembershipExpireScanHandler) ProcessTask(ctx context.Context, t *asynq.Task) error {
+	now := time.Now()
+	dayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
+	dayEnd := dayStart.Add(24*time.Hour - time.Second)
+
+	// 1. 扫描当天将到期会员，并为每个会员安排准点处理任务
+	todayAgents, err := service.ListTodayWillExpireAgents(ctx, l.svcCtx.AgentModel, dayStart, dayEnd)
+	if err != nil {
+		logx.Errorf("扫描当天到期会员失败: %v", err)
+		return err
+	}
+	for _, agent := range todayAgents {
+		processAt := agent.MembershipExpiryTime.Time
+		if processAt.Before(now) {
+			processAt = now
+		}
+		if l.svcCtx.AsynqService != nil {
+			if sendErr := l.svcCtx.AsynqService.SendAgentMembershipExpireHandleTask(agent.Id, processAt); sendErr != nil {
+				logx.Errorf("安排会员到期处理任务失败，代理ID: %d, 错误: %v", agent.Id, sendErr)
+			}
+		}
+	}
+
+	// 2. 补偿处理：扫描已过期但尚未降级的会员并立即降级
+	expiredAgents, err := service.ListExpiredUnprocessedAgents(ctx, l.svcCtx.AgentModel, now)
+	if err != nil {
+		logx.Errorf("扫描已过期未处理会员失败: %v", err)
+		return err
+	}
+
+	downgradedCount := 0
+	for _, agent := range expiredAgents {
+		downgraded, degradeErr := service.DowngradeExpiredMembership(ctx, l.svcCtx.AgentModel, agent.Id, now)
+		if degradeErr != nil {
+			logx.Errorf("补偿降级失败，代理ID: %d, 错误: %v", agent.Id, degradeErr)
+			continue
+		}
+		if downgraded {
+			downgradedCount++
+		}
+	}
+
+	logx.Infof("会员到期扫描完成，当天到期会员: %d, 补偿降级数量: %d", len(todayAgents), downgradedCount)
+	return nil
+}