feat(通道流量阈值): AUG26-011 运营商通道流量阈值达量停机与周期复机

internal/infrastructure/carrierthreshold/task.go

@@ -0,0 +1,68 @@
+package carrierthreshold
+
+import (
+	"context"
+	"time"
+
+	"github.com/hibiken/asynq"
+
+	carrierthresholdapp "github.com/break/junhong_cmp_fiber/internal/application/carrierthreshold"
+	"github.com/break/junhong_cmp_fiber/pkg/auditcontext"
+	"github.com/break/junhong_cmp_fiber/pkg/constants"
+	"github.com/break/junhong_cmp_fiber/pkg/errors"
+)
+
+// CycleTaskHandler 执行运营商通道流量阈值周期处理任务。
+type CycleTaskHandler struct {
+	service *carrierthresholdapp.Service
+}
+
+// NewCycleTaskHandler 创建通道阈值周期处理任务 Handler。
+func NewCycleTaskHandler(service *carrierthresholdapp.Service) *CycleTaskHandler {
+	return &CycleTaskHandler{service: service}
+}
+
+// Handle 扫描已跨期的持锁，认领解锁并在条件满足时写复机事件。
+// 审计上下文固定为计划任务来源，使解锁与复机事件都可按计划任务维度追溯。
+func (h *CycleTaskHandler) Handle(ctx context.Context, task *asynq.Task) error {
+	if h == nil || h.service == nil {
+		return errors.New(errors.CodeServiceUnavailable, "通道流量阈值周期处理任务未配置")
+	}
+	taskType := constants.TaskTypeCarrierThresholdCycle
+	if task != nil && task.Type() != "" {
+		taskType = task.Type()
+	}
+	ctx = auditcontext.With(ctx, auditcontext.Context{
+		ActorKind: constants.AuditActorScheduledJob, ActorID: taskType,
+		ActorName: "运营商通道流量阈值周期处理计划任务", Source: constants.AuditSourceScheduler,
+	})
+	_, err := h.service.ProcessDueLocks(ctx, time.Now().UTC())
+	return err
+}
+
+// RecoveryTaskHandler 执行运营商通道流量阈值停复机结果恢复任务。
+type RecoveryTaskHandler struct {
+	service *carrierthresholdapp.Service
+}
+
+// NewRecoveryTaskHandler 创建通道阈值停复机结果恢复任务 Handler。
+func NewRecoveryTaskHandler(service *carrierthresholdapp.Service) *RecoveryTaskHandler {
+	return &RecoveryTaskHandler{service: service}
+}
+
+// Handle 扫描已提交的停复机子任务并只查询运营商状态回填，绝不重复发起停复机。
+func (h *RecoveryTaskHandler) Handle(ctx context.Context, task *asynq.Task) error {
+	if h == nil || h.service == nil {
+		return errors.New(errors.CodeServiceUnavailable, "通道流量阈值恢复扫描任务未配置")
+	}
+	taskType := constants.TaskTypeCarrierThresholdRecovery
+	if task != nil && task.Type() != "" {
+		taskType = task.Type()
+	}
+	ctx = auditcontext.With(ctx, auditcontext.Context{
+		ActorKind: constants.AuditActorScheduledJob, ActorID: taskType,
+		ActorName: "运营商通道流量阈值结果恢复计划任务", Source: constants.AuditSourceScheduler,
+	})
+	_, err := h.service.RecoverSubmitted(ctx, time.Now().UTC())
+	return err
+}