iotlcm/worker-svr/internal/service/warningconfig/executor.go

package warningconfig

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 warningConfigProcessingRetryAfter = 2 * time.Minute

type executor struct {
	repository *Repository
}

func NewExecutor() *executor {
	return &executor{repository: NewRepository()}
}

func (e *executor) CanExecuteProcessing(_ context.Context, main *entity.BatchMain) (bool, error) {
	if main == nil || main.BatchType != consts.BatchTypeSimWarningConfig || main.BatchAsyncStatus != consts.BatchAsyncStatusProcessing {
		return false, nil
	}
	if main.StartTime == nil || main.StartTime.IsZero() {
		return true, nil
	}
	return gtime.Now().Sub(main.StartTime) >= warningConfigProcessingRetryAfter, 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 warning config batch failed")
	}
	validateTask(task)
	return gerror.Wrap(e.repository.Commit(ctx, task), "commit warning config batch failed")
}

func validateTask(task *Task) {
	if task == nil || len(task.Details) == 0 {
		if task != nil {
			task.FailureReason = "预警配置批次没有可处理明细"
		}
		return
	}
	for _, detail := range task.Details {
		validateDetail(task, detail)
	}
	_, failCount := countResult(task)
	if failCount > 0 {
		markAtomicFailure(task, "预警配置批次存在失败明细，整批未执行")
	}
}

func markAtomicFailure(task *Task, reason string) {
	if task == nil {
		return
	}
	reason = strings.TrimSpace(reason)
	if reason == "" {
		reason = "本批次存在失败明细，整批未执行"
	}
	task.FailureReason = reason
	for _, detail := range task.Details {
		if detail != nil && strings.TrimSpace(detail.Reason) == "" {
			detail.Reason = "本批次存在失败明细，整批未执行"
		}
	}
}

func validateDetail(task *Task, detail *Detail) {
	if detail == nil || detail.Reason != "" {
		return
	}
	config := detail.Config
	if config.ItemType == warnItemTypeFlowPool {
		validateFlowPoolDetail(config, detail)
		return
	}
	validateCardDetail(config, detail)
}

func validateCardDetail(config Config, detail *Detail) {
	detail.ICCID = strings.TrimSpace(detail.ICCID)
	detail.MSISDN = strings.TrimSpace(detail.MSISDN)
	detail.IMSI = strings.TrimSpace(detail.IMSI)
	if detail.ICCID == "" {
		detail.Reason = "ICCID不能为空"
		return
	}
	if !detail.Authorized {
		detail.Reason = "ICCID不在当前操作员授权范围内"
		return
	}
	if detail.Card == nil {
		detail.Reason = "ICCID不存在或不属于当前租户"
		return
	}
	if detail.MSISDN != "" && detail.MSISDN != detail.Card.MSISDN {
		detail.Reason = "MSISDN与卡档案不一致"
		return
	}
	if detail.IMSI != "" && detail.IMSI != detail.Card.IMSI {
		detail.Reason = "IMSI与卡档案不一致"
		return
	}
	if detail.Card.BossCardStatus == consts.SimBossCardStatusStopped {
		detail.Reason = "卡已停机，不能配置预警"
		return
	}
	if detail.Card.BossCardStatus != consts.SimBossCardStatusActivated {
		detail.Reason = "当前卡状态为" + bossCardStatusText(detail.Card.BossCardStatus) + "，只有已激活卡可以配置预警"
		return
	}
	if detail.Card.PackageStatus != packageStatusEffective {
		detail.Reason = "套餐未生效，不能配置预警"
		return
	}
	if detail.Card.IsFlowPool != 0 {
		detail.Reason = "流量池卡暂不支持单卡预警配置"
		return
	}
	if strings.TrimSpace(detail.Card.CurrentPackageCode) == "" {
		detail.Reason = "当前套餐编号为空，不能配置预警"
		return
	}
	switch config.WarnType {
	case warnTypeCardTraffic:
		validateTrafficDetail(config, detail)
	case warnTypeExpireDay, warnTypeExpireMonth:
		if detail.Card.FlowPkgEndAt == nil || detail.Card.FlowPkgEndAt.IsZero() {
			detail.Reason = "套餐到期时间为空，不能配置到期预警"
		}
	default:
		detail.Reason = "预警类型无效"
	}
}

func validateFlowPoolDetail(config Config, detail *Detail) {
	detail.FlowPoolCode = strings.TrimSpace(detail.FlowPoolCode)
	if detail.FlowPoolCode == "" {
		detail.Reason = "流量池编号不能为空"
		return
	}
	if !detail.Authorized {
		detail.Reason = "流量池不在当前操作员授权范围内"
		return
	}
	if detail.Pool == nil {
		detail.Reason = "流量池不存在或不属于当前租户"
		return
	}
	if detail.Pool.PoolStatus != flowPoolStatusNormal {
		detail.Reason = "流量池状态不是正常，不能配置预警"
		return
	}
	if detail.Pool.TotalFlowMb <= 0 {
		detail.Reason = "流量池总流量为空，不能配置预警"
		return
	}
	if config.WarnType != warnTypeFlowPoolPercent || config.WarnUnit != warnUnitPercent {
		detail.Reason = "流量池预警仅支持百分比配置"
		return
	}
	if config.AutoAction != 0 {
		detail.Reason = "流量池预警暂不支持第三次关停"
	}
}

func validateTrafficDetail(config Config, detail *Detail) {
	if detail.Card.PackageTotalFlowMb <= 0 {
		detail.Reason = "套餐总流量为空，不能配置流量预警"
		return
	}
	if config.WarnUnit != warnUnitMb {
		return
	}
	total := detail.Card.PackageTotalFlowMb
	if config.FirstWarn > total || config.SecondWarn > total || config.ThirdWarn > total {
		detail.Reason = "流量阈值不能大于套餐总流量"
	}
}

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
}

func bossCardStatusText(status int) string {
	switch status {
	case consts.SimBossCardStatusActivated:
		return "已激活"
	case consts.SimBossCardStatusStopped:
		return "已停机"
	case 1:
		return "待激活"
	default:
		return "未知状态"
	}
}