chore: 并入 iotlcm-worker-svr 历史到 worker-svr/（原提交 96f0f4e）

2026-11-02 13:28:03 +08:00
parent 6de2e503cc 96f0f4eacd
commit 2923c78755
422 changed files with 80993 additions and 0 deletions
@@ -0,0 +1,172 @@
+package trafficcutoff
+
+import (
+	"context"
+	"strings"
+
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
+
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/service/batchresult"
+)
+
+const (
+	cascadeFailureReason     = "同一批次存在无效流量断网明细，已跳过更新"
+	validationSkipReason     = "批次校验未通过，已跳过流量断网更新"
+	emptyICCIDReason         = "ICCID不能为空"
+	duplicateICCIDReason     = "ICCID在批次内重复"
+	unauthorizedReason       = "ICCID不在当前操作员授权范围内"
+	cardMissingReason        = "ICCID不存在或不属于当前租户"
+	statusCanceledReason     = "ICCID已销户"
+	statusInconsistentReason = "同一批次内卡状态不一致"
+)
+
+type executor struct {
+	repository Repository
+}
+
+func NewExecutor(repository Repository) Executor {
+	return &executor{repository: repository}
+}
+
+func (e *executor) Execute(ctx context.Context, batchNo string) error {
+	task, err := e.repository.Load(ctx, strings.TrimSpace(batchNo))
+	if err != nil {
+		return gerror.Wrap(err, "load traffic cutoff batch")
+	}
+	if !validateTask(task) {
+		if err = e.repository.SaveValidationFailure(ctx, task); err != nil {
+			return gerror.Wrap(err, "save traffic cutoff validation failure")
+		}
+		e.saveResultFile(ctx, task)
+		return nil
+	}
+	if err = e.repository.CommitBusiness(ctx, task); err != nil {
+		return gerror.Wrap(err, "commit traffic cutoff business")
+	}
+	e.saveResultFile(ctx, task)
+	return nil
+}
+
+func (e *executor) saveResultFile(ctx context.Context, task *Task) {
+	request := batchresult.Request{
+		BatchNo:   task.Batch.BatchNo,
+		BatchType: task.Batch.BatchType,
+		TenantID:  task.Batch.TenantID,
+		Rows:      task.ResultRows(),
+	}
+	if err := e.repository.SaveResultFile(ctx, task, request); err != nil {
+		g.Log().Errorf(ctx, "save traffic cutoff result file status failed, batchNo:%s, err:%+v", task.Batch.BatchNo, err)
+	}
+}
+
+func validateTask(task *Task) bool {
+	if task == nil || len(task.Details) == 0 {
+		if task != nil {
+			task.FailureReason = "流量断网批次无有效明细"
+		}
+		return false
+	}
+	if task.TrafficCutoffStatus != 1 {
+		task.FailureReason = "流量断网批次必须开启断网"
+		markAllFailed(task, task.FailureReason)
+		return false
+	}
+	if task.TrafficCutoffThreshold <= 0 {
+		task.FailureReason = "流量断网阈值必须大于0"
+		markAllFailed(task, task.FailureReason)
+		return false
+	}
+
+	hasFailure := false
+	iccids := make(map[string]int, len(task.Details))
+	statusSet := make(map[int]struct{})
+	for _, detail := range task.Details {
+		if detail == nil {
+			hasFailure = true
+			continue
+		}
+		validateDetail(detail)
+		if detail.Reason != "" {
+			hasFailure = true
+			continue
+		}
+		iccids[detail.ICCID]++
+		statusSet[detail.BossCardStatusBefore] = struct{}{}
+	}
+	for _, detail := range task.Details {
+		if detail == nil || detail.Reason != "" {
+			continue
+		}
+		if iccids[detail.ICCID] > 1 {
+			detail.Reason = duplicateICCIDReason
+			hasFailure = true
+		}
+	}
+	if len(statusSet) > 1 {
+		for _, detail := range task.Details {
+			if detail != nil && detail.Reason == "" {
+				detail.Reason = statusInconsistentReason
+			}
+		}
+		hasFailure = true
+	}
+	if !hasFailure {
+		task.ValidationOK = true
+		return true
+	}
+	for _, detail := range task.Details {
+		if detail != nil && detail.Reason == "" {
+			detail.Reason = cascadeFailureReason
+		}
+	}
+	task.FailureReason = validationSkipReason
+	return false
+}
+
+func validateDetail(detail *Detail) {
+	if detail.Reason != "" {
+		return
+	}
+	detail.ICCID = strings.TrimSpace(detail.ICCID)
+	detail.InputMSISDN = strings.TrimSpace(detail.InputMSISDN)
+	detail.InputIMSI = strings.TrimSpace(detail.InputIMSI)
+	if detail.ICCID == "" {
+		detail.Reason = emptyICCIDReason
+		return
+	}
+	if !detail.Authorized {
+		detail.Reason = unauthorizedReason
+		return
+	}
+	if detail.Card == nil {
+		detail.Reason = cardMissingReason
+		return
+	}
+	detail.MSISDN = strings.TrimSpace(detail.Card.MSISDN)
+	detail.IMSI = strings.TrimSpace(detail.Card.IMSI)
+	detail.BossCardStatusBefore = detail.Card.BossCardStatus
+	detail.TrafficCutoffStatusBefore = detail.Card.TrafficCutoffStatus
+	detail.TrafficCutoffThresholdBefore = detail.Card.TrafficCutoffThreshold
+
+	if detail.InputMSISDN != "" && detail.InputMSISDN != detail.MSISDN {
+		detail.Reason = "MSISDN与卡档案不一致"
+		return
+	}
+	if detail.InputIMSI != "" && detail.InputIMSI != detail.IMSI {
+		detail.Reason = "IMSI与卡档案不一致"
+		return
+	}
+	if detail.BossCardStatusBefore == BossCardStatusCanceled {
+		detail.Reason = statusCanceledReason
+		return
+	}
+}
+
+func markAllFailed(task *Task, reason string) {
+	for _, detail := range task.Details {
+		if detail != nil {
+			detail.Reason = reason
+		}
+	}
+}