chore: 并入 iotlcm-admin-svr 历史到 admin-svr/（原提交 49818c6）
This commit is contained in:
MiFaZhan
@@ -0,0 +1,121 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/dao"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchimport"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/contexts"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/rabbitmq"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/xgorm/handler"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
)
|
||||
|
||||
type AuthorizedRow interface {
|
||||
GetIccid() string
|
||||
SetAuthorized(bool)
|
||||
}
|
||||
|
||||
type SubmitOptions struct {
|
||||
Remark string
|
||||
}
|
||||
|
||||
type BatchExecuteMessage struct {
|
||||
BatchNo string `json:"batchNo"`
|
||||
BatchType int `json:"batchType"`
|
||||
}
|
||||
|
||||
type SubmitResult struct {
|
||||
BatchNo string
|
||||
BatchType int
|
||||
BatchAsyncStatus int
|
||||
Total int
|
||||
}
|
||||
|
||||
func PrepareAuthorizedRows[T AuthorizedRow](
|
||||
ctx context.Context,
|
||||
rows []batchimport.Row,
|
||||
rowFactory func(batchimport.Row) (T, error),
|
||||
typeErrorMsg string,
|
||||
) error {
|
||||
cardRows := make([]T, 0, len(rows))
|
||||
iccids := make([]string, 0, len(rows))
|
||||
for _, item := range rows {
|
||||
row, err := rowFactory(item)
|
||||
if err != nil {
|
||||
return gerror.New(typeErrorMsg)
|
||||
}
|
||||
cardRows = append(cardRows, row)
|
||||
if id := row.GetIccid(); id != "" {
|
||||
iccids = append(iccids, id)
|
||||
}
|
||||
}
|
||||
if len(iccids) == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
var visible []struct {
|
||||
Iccid string `orm:"iccid"`
|
||||
}
|
||||
columns := dao.TSimCardInfo.Columns()
|
||||
err := handler.Model(dao.TSimCardInfo.Ctx(ctx), &handler.Option{FilterTenant: true, FilterAuth: true}).
|
||||
Fields(columns.Iccid, columns.TenantId, columns.DeptId, columns.CreatedDept, columns.CreatedBy).
|
||||
WhereIn(columns.Iccid, iccids).
|
||||
Scan(&visible)
|
||||
if err != nil {
|
||||
return gerror.Wrap(err, "计算SIM卡数据权限失败")
|
||||
}
|
||||
|
||||
authorized := make(map[string]struct{}, len(visible))
|
||||
for _, card := range visible {
|
||||
authorized[strings.TrimSpace(card.Iccid)] = struct{}{}
|
||||
}
|
||||
for _, row := range cardRows {
|
||||
_, ok := authorized[row.GetIccid()]
|
||||
row.SetAuthorized(ok)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func PublishBatchMessage(ctx context.Context, result *batchimport.Result, queueName, logPrefix string) {
|
||||
_ = PublishBatchMessageStrict(ctx, result, queueName, logPrefix)
|
||||
}
|
||||
|
||||
func PublishBatchMessageStrict(ctx context.Context, result *batchimport.Result, queueName, logPrefix string) error {
|
||||
if result == nil || strings.TrimSpace(result.BatchNo) == "" {
|
||||
return nil
|
||||
}
|
||||
if err := rabbitmq.Publish(ctx, queueName, BatchExecuteMessage{
|
||||
BatchNo: result.BatchNo,
|
||||
BatchType: result.BatchType,
|
||||
}); err != nil {
|
||||
g.Log().Errorf(ctx, "%s batch async publish failed, batchNo=%s, batchType=%d, tenantId=%s, error=%+v",
|
||||
logPrefix, result.BatchNo, result.BatchType, contexts.GetTenantId(ctx), err)
|
||||
return gerror.Wrap(err, "publish batch async message")
|
||||
}
|
||||
g.Log().Infof(ctx, "%s batch async published, queue=%s, batchNo=%s, batchType=%d, tenantId=%s",
|
||||
logPrefix, queueName, result.BatchNo, result.BatchType, contexts.GetTenantId(ctx))
|
||||
return nil
|
||||
}
|
||||
|
||||
func CurrentOperator(ctx context.Context) batchimport.OperatorContext {
|
||||
return batchimport.OperatorContext{
|
||||
TenantId: contexts.GetTenantId(ctx),
|
||||
DeptId: contexts.GetDeptId(ctx),
|
||||
UserId: contexts.GetUserId(ctx),
|
||||
}
|
||||
}
|
||||
|
||||
func PendingSubmitResult(result *batchimport.Result, status int) *SubmitResult {
|
||||
res := &SubmitResult{BatchAsyncStatus: status}
|
||||
if result == nil {
|
||||
return res
|
||||
}
|
||||
res.BatchNo = result.BatchNo
|
||||
res.BatchType = result.BatchType
|
||||
res.Total = result.TotalRows
|
||||
return res
|
||||
}
|
||||
@@ -0,0 +1,102 @@
|
||||
package cardsource
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchimport"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchinput/text"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
)
|
||||
|
||||
type IccidLine struct {
|
||||
RowNo int
|
||||
Iccid string
|
||||
}
|
||||
|
||||
type CardIdentityLine struct {
|
||||
RowNo int
|
||||
Iccid string
|
||||
Msisdn string
|
||||
Imsi string
|
||||
Imei string
|
||||
Reason string
|
||||
}
|
||||
|
||||
func ParseIccidLines(ctx context.Context, input string) ([]IccidLine, error) {
|
||||
lines, err := text.ParseNonEmptyLines(ctx, input)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
result := make([]IccidLine, 0, len(lines))
|
||||
for _, line := range lines {
|
||||
result = append(result, IccidLine{
|
||||
RowNo: line.RowNo,
|
||||
Iccid: line.Text,
|
||||
})
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func ParseCardIdentityLines(ctx context.Context, input string) ([]CardIdentityLine, error) {
|
||||
lines, err := text.ParseWhitespaceFields(ctx, input)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
result := make([]CardIdentityLine, 0, len(lines))
|
||||
for _, line := range lines {
|
||||
item := CardIdentityLine{RowNo: line.RowNo}
|
||||
switch len(line.Fields) {
|
||||
case 3:
|
||||
item.Iccid = line.Fields[0]
|
||||
item.Msisdn = line.Fields[1]
|
||||
item.Imsi = line.Fields[2]
|
||||
case 4:
|
||||
item.Iccid = line.Fields[0]
|
||||
item.Msisdn = line.Fields[1]
|
||||
item.Imsi = line.Fields[2]
|
||||
item.Imei = line.Fields[3]
|
||||
default:
|
||||
item.Reason = "格式错误，需为 ICCID MSISDN IMSI [IMEI]"
|
||||
}
|
||||
result = append(result, item)
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func BuildIccidRows[T batchimport.Row](ctx context.Context, input, emptyMessage string, factory func(IccidLine) T) ([]batchimport.Row, error) {
|
||||
lines, err := ParseIccidLines(ctx, input)
|
||||
if err != nil {
|
||||
if errors.Is(err, text.ErrEmptyInput) {
|
||||
return nil, gerror.New(emptyMessage)
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
rows := make([]batchimport.Row, 0, len(lines))
|
||||
for _, line := range lines {
|
||||
rows = append(rows, factory(line))
|
||||
}
|
||||
return rows, nil
|
||||
}
|
||||
|
||||
func BuildCardIdentityRows[T batchimport.Row](ctx context.Context, input, emptyMessage string, factory func(CardIdentityLine) T) ([]batchimport.Row, error) {
|
||||
lines, err := ParseCardIdentityLines(ctx, input)
|
||||
if err != nil {
|
||||
if errors.Is(err, text.ErrEmptyInput) {
|
||||
return nil, gerror.New(emptyMessage)
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
rows := make([]batchimport.Row, 0, len(lines))
|
||||
for _, line := range lines {
|
||||
row := factory(line)
|
||||
row.AddErrorReason(line.Reason)
|
||||
rows = append(rows, row)
|
||||
}
|
||||
return rows, nil
|
||||
}
|
||||
@@ -0,0 +1,56 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user