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 }