MiFaZhan
57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package cardsource
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchimport"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchinput/numrange"
|
|
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
)
|
|
|
|
type RangeConfig struct {
|
|
StartIccid string
|
|
EndIccid string
|
|
StartMsisdn string
|
|
EndMsisdn string
|
|
StartImsi string
|
|
EndImsi string
|
|
MaxRows int
|
|
EmptyMessage string
|
|
OverLimitMessage string
|
|
CountMismatchMessage string
|
|
}
|
|
|
|
func BuildRangeRows[T batchimport.Row](ctx context.Context, cfg RangeConfig, factory func(rowNo int, iccid, msisdn, imsi string) T) ([]batchimport.Row, error) {
|
|
iccids, err := numrange.Build(ctx, cfg.StartIccid, cfg.EndIccid, "ICCID", cfg.MaxRows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
msisdns, err := numrange.Build(ctx, cfg.StartMsisdn, cfg.EndMsisdn, "MSISDN", cfg.MaxRows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
imsis, err := numrange.Build(ctx, cfg.StartImsi, cfg.EndImsi, "IMSI", cfg.MaxRows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(iccids) != len(msisdns) || len(iccids) != len(imsis) {
|
|
return nil, gerror.Newf(cfg.CountMismatchMessage, len(iccids), len(msisdns), len(imsis))
|
|
}
|
|
if len(iccids) == 0 {
|
|
return nil, gerror.New(cfg.EmptyMessage)
|
|
}
|
|
if cfg.MaxRows > 0 && len(iccids) > cfg.MaxRows {
|
|
return nil, gerror.Newf(cfg.OverLimitMessage, cfg.MaxRows)
|
|
}
|
|
|
|
rows := make([]batchimport.Row, 0, len(iccids))
|
|
for i := range iccids {
|
|
if err = ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
rows = append(rows, factory(i+1, iccids[i], msisdns[i], imsis[i]))
|
|
}
|
|
return rows, nil
|
|
}
|