iotlcm/admin-svr/server/internal/logic/iot/simcardbatch/datacut/parser.go

package datacut

import (
	"context"
	"errors"
	"fmt"
	"strings"

	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchimport"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchinput/excel"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchinput/text"

	"github.com/gogf/gf/v2/encoding/gjson"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
)

type Row struct {
	DetailID   int64
	RowNo      int
	Iccid      string
	Msisdn     string
	Imsi       string
	Reason     string
	Authorized bool
}

func (r *Row) GetRowNo() int {
	if r == nil {
		return 0
	}
	return r.RowNo
}

func (r *Row) GetItemID() string {
	if r == nil {
		return ""
	}
	return r.Iccid
}

func (r *Row) GetErrorReasons() []string {
	if r == nil || r.Reason == "" {
		return nil
	}
	return []string{r.Reason}
}

func (r *Row) AddErrorReason(reason string) {
	if r == nil || r.Reason != "" || reason == "" {
		return
	}
	r.Reason = reason
}

func (r *Row) GetIccid() string {
	if r == nil {
		return ""
	}
	return r.Iccid
}

func (r *Row) SetAuthorized(v bool) {
	if r == nil {
		return
	}
	r.Authorized = v
}

func (r *Row) Snapshot() *gjson.Json {
	if r == nil {
		return gjson.New(g.Map{})
	}
	return gjson.New(g.Map{
		"ICCID":      r.Iccid,
		"MSISDN":     r.Msisdn,
		"IMSI":       r.Imsi,
		"authorized": r.Authorized,
	})
}

func parseExcelFileRows(ctx context.Context, filePath string) ([]batchimport.Row, error) {
	parsed, err := excel.ParseFile(ctx, filePath, excel.ParseOptions{RequiredHeaders: []string{"ICCID"}})
	if err != nil {
		return nil, err
	}
	rows := make([]batchimport.Row, 0, len(parsed))
	for _, item := range parsed {
		rows = append(rows, &Row{
			RowNo:  item.RowNo,
			Iccid:  excel.Field(item, "ICCID"),
			Msisdn: excel.Field(item, "MSISDN"),
			Imsi:   excel.Field(item, "IMSI"),
		})
	}
	return rows, nil
}

func parseInputRows(ctx context.Context, input string) ([]batchimport.Row, error) {
	lines, err := text.ParseWhitespaceFields(ctx, input)
	if err != nil {
		if errors.Is(err, text.ErrEmptyInput) {
			return nil, gerror.New("批量输入内容不能为空")
		}
		return nil, err
	}
	rows := make([]batchimport.Row, 0, len(lines))
	for _, line := range lines {
		row := &Row{RowNo: line.RowNo}
		switch len(line.Fields) {
		case 1:
			row.Iccid = line.Fields[0]
		case 3:
			row.Iccid = line.Fields[0]
			row.Msisdn = line.Fields[1]
			row.Imsi = line.Fields[2]
		default:
			row.AddErrorReason("格式错误，应为 ICCID 或 ICCID MSISDN IMSI")
		}
		rows = append(rows, row)
	}
	return rows, nil
}

func validateBasicRows(rows []batchimport.Row) error {
	firstByIccid := make(map[string]*Row, len(rows))
	issues := make([]string, 0)
	for _, item := range rows {
		row, ok := item.(*Row)
		if !ok || row == nil {
			return gerror.New("达量断网批次行类型错误")
		}
		row.Iccid = strings.TrimSpace(row.Iccid)
		row.Msisdn = strings.TrimSpace(row.Msisdn)
		row.Imsi = strings.TrimSpace(row.Imsi)
		if row.Reason != "" {
			issues = append(issues, fmt.Sprintf("第%d行：%s", row.RowNo, row.Reason))
			continue
		}
		if row.Iccid == "" {
			row.AddErrorReason("ICCID不能为空")
			issues = append(issues, fmt.Sprintf("第%d行：ICCID不能为空", row.RowNo))
			continue
		}
		if first, exists := firstByIccid[row.Iccid]; exists {
			row.AddErrorReason(fmt.Sprintf("ICCID与第%d行重复", first.RowNo))
			first.AddErrorReason("ICCID重复")
			issues = append(issues, fmt.Sprintf("第%d行：ICCID与第%d行重复", row.RowNo, first.RowNo))
			continue
		}
		firstByIccid[row.Iccid] = row
	}
	if len(issues) > 0 {
		return gerror.New("达量断网批次基础格式校验失败：" + strings.Join(issues, "；"))
	}
	return nil
}