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,142 @@
+package trafficcutoffconfig
+
+import (
+	"context"
+	"strings"
+	"time"
+
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/os/gtime"
+
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/consts"
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model/entity"
+)
+
+const (
+	noDetailsReason            = "达量断网配置批次没有可处理明细"
+	validationFailureReason    = "部分达量断网配置校验失败"
+	emptyICCIDReason           = "ICCID不能为空"
+	duplicateICCIDReason       = "ICCID在批次内重复"
+	unauthorizedReason         = "ICCID不在当前操作员授权范围内"
+	cardMissingReason          = "ICCID不存在或不属于当前租户"
+	msisdnMismatchReason       = "MSISDN与卡档案不一致"
+	imsiMismatchReason         = "IMSI与卡档案不一致"
+	flowPoolCardReason         = "流量池卡不支持达量断网配置"
+	packageTotalExceededReason = "达量断网阈值不能大于套餐总流量"
+)
+
+type executor struct {
+	repository *Repository
+}
+
+const trafficCutoffConfigProcessingRetryAfter = 2 * time.Minute
+
+func NewExecutor() *executor {
+	return &executor{repository: NewRepository()}
+}
+
+func (e *executor) CanExecuteProcessing(_ context.Context, main *entity.BatchMain) (bool, error) {
+	if main == nil || main.BatchType != consts.BatchTypeTrafficCutoffConfig || main.BatchAsyncStatus != consts.BatchAsyncStatusProcessing {
+		return false, nil
+	}
+	if main.StartTime == nil || main.StartTime.IsZero() {
+		return true, nil
+	}
+	return gtime.Now().Sub(main.StartTime) >= trafficCutoffConfigProcessingRetryAfter, nil
+}
+
+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 config batch failed")
+	}
+	validateTask(task)
+	return gerror.Wrap(e.repository.Commit(ctx, task), "commit traffic cutoff config batch failed")
+}
+
+func validateTask(task *Task) {
+	if task == nil || len(task.Details) == 0 {
+		if task != nil {
+			task.FailureReason = noDetailsReason
+		}
+		return
+	}
+
+	iccidCounts := make(map[string]int, len(task.Details))
+	for _, detail := range task.Details {
+		if detail == nil {
+			continue
+		}
+		detail.ICCID = strings.TrimSpace(detail.ICCID)
+		detail.MSISDN = strings.TrimSpace(detail.MSISDN)
+		detail.IMSI = strings.TrimSpace(detail.IMSI)
+		if detail.ICCID != "" {
+			iccidCounts[detail.ICCID]++
+		}
+	}
+
+	for _, detail := range task.Details {
+		validateDetail(task, detail)
+		if detail != nil && detail.Reason == "" && iccidCounts[detail.ICCID] > 1 {
+			detail.Reason = duplicateICCIDReason
+		}
+	}
+	_, failCount := countResult(task)
+	if failCount > 0 {
+		task.FailureReason = validationFailureReason
+	} else {
+		task.FailureReason = ""
+	}
+}
+
+func validateDetail(task *Task, detail *Detail) {
+	if detail == nil || detail.Reason != "" {
+		return
+	}
+	if detail.ICCID == "" {
+		detail.Reason = emptyICCIDReason
+		return
+	}
+	if !detail.Authorized {
+		detail.Reason = unauthorizedReason
+		return
+	}
+	if detail.Card == nil {
+		detail.Reason = cardMissingReason
+		return
+	}
+	detail.Card.ICCID = strings.TrimSpace(detail.Card.ICCID)
+	detail.Card.MSISDN = strings.TrimSpace(detail.Card.MSISDN)
+	detail.Card.IMSI = strings.TrimSpace(detail.Card.IMSI)
+	if detail.MSISDN != "" && detail.MSISDN != detail.Card.MSISDN {
+		detail.Reason = msisdnMismatchReason
+		return
+	}
+	if detail.IMSI != "" && detail.IMSI != detail.Card.IMSI {
+		detail.Reason = imsiMismatchReason
+		return
+	}
+	if detail.Card.IsFlowPool != 0 {
+		detail.Reason = flowPoolCardReason
+		return
+	}
+	if task.Config.TrafficCutoffStatus == 1 &&
+		detail.Card.PackageTotalFlowMb > 0 &&
+		task.Config.TrafficCutoffThreshold > detail.Card.PackageTotalFlowMb {
+		detail.Reason = packageTotalExceededReason
+	}
+}
+
+func countResult(task *Task) (successCount, failCount int) {
+	if task == nil {
+		return 0, 0
+	}
+	for _, detail := range task.Details {
+		if detail != nil && detail.Success() {
+			successCount++
+		} else {
+			failCount++
+		}
+	}
+	return successCount, failCount
+}