chore: 并入 iotlcm-admin-svr 历史到 admin-svr/（原提交 49818c6）

2026-11-02 13:27:59 +08:00
parent 976980a3f5 49818c6dbe
commit 193f7393c0
1250 changed files with 154111 additions and 0 deletions
@@ -0,0 +1,141 @@
+package cancel
+
+import (
+	"context"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchmetadata"
+	"strings"
+
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
+	"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"
+	"github.com/gogf/gf/v2/net/ghttp"
+)
+
+type Submitter struct{}
+
+func NewSubmitter() *Submitter {
+	return &Submitter{}
+}
+
+func (s *Submitter) SubmitExcel(ctx context.Context, file *ghttp.UploadFile, remark string) (*SubmitResult, error) {
+	processor := NewProcessor()
+	req := batchimport.SubmitRequest{
+		SourceChannel: batchmetadata.Console,
+		ExecutionMode: batchmetadata.Bulk,
+		File:          file,
+		BizParams:     &SubmitOptions{Remark: strings.TrimSpace(remark)},
+		SourceType:    consts.BatchSourceTypeFile,
+		Operator:      currentOperator(ctx),
+	}
+	result, err := batchimport.CreatePendingUpload(ctx, req, processor, batchimport.PendingOptions{
+		PrepareRows: s.prepareAuthorizedRows,
+	})
+	if err != nil {
+		return nil, err
+	}
+	s.publish(ctx, result)
+	return pendingSubmitResult(result), nil
+}
+
+func (s *Submitter) SubmitInput(ctx context.Context, input, remark string) (*SubmitResult, error) {
+	rows, err := ParseInputRows(ctx, input)
+	if err != nil {
+		return nil, err
+	}
+	processor := NewProcessor()
+	req := batchimport.SubmitRequest{
+		SourceChannel: batchmetadata.Console,
+		ExecutionMode: batchmetadata.Bulk,
+		BizParams:     &SubmitOptions{Remark: strings.TrimSpace(remark)},
+		SourceType:    consts.BatchSourceTypeBatchInput,
+		Operator:      currentOperator(ctx),
+	}
+	result, err := batchimport.CreatePendingRows(ctx, req, processor, rows, batchimport.PendingOptions{
+		PrepareRows: s.prepareAuthorizedRows,
+	})
+	if err != nil {
+		return nil, err
+	}
+	s.publish(ctx, result)
+	return pendingSubmitResult(result), nil
+}
+
+func (s *Submitter) publish(ctx context.Context, result *batchimport.Result) {
+	if result == nil || strings.TrimSpace(result.BatchNo) == "" {
+		return
+	}
+	if err := publishBatchExecute(ctx, batchimport.Message{
+		BatchNo:   result.BatchNo,
+		BatchType: result.BatchType,
+	}); err != nil {
+		g.Log().Errorf(ctx, "cancel batch async publish failed, batchNo=%s, batchType=%d, tenantId=%s, error=%+v",
+			result.BatchNo, result.BatchType, contexts.GetTenantId(ctx), err)
+	}
+}
+
+var publishBatchExecute = func(ctx context.Context, msg batchimport.Message) error {
+	return rabbitmq.Publish(ctx, consts.QueueBatchExecute, msg)
+}
+
+func (s *Submitter) prepareAuthorizedRows(ctx context.Context, rows []batchimport.Row) error {
+	cardRows := make([]*Row, 0, len(rows))
+	iccids := make([]string, 0, len(rows))
+	for _, item := range rows {
+		row, ok := item.(*Row)
+		if !ok || row == nil {
+			return gerror.New("销户批次行类型错误")
+		}
+		cardRows = append(cardRows, row)
+		if row.Iccid != "" {
+			iccids = append(iccids, row.Iccid)
+		}
+	}
+	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 {
+		_, row.Authorized = authorized[row.Iccid]
+	}
+	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) *SubmitResult {
+	res := &SubmitResult{BatchAsyncStatus: consts.BatchAsyncStatusPending}
+	if result == nil {
+		return res
+	}
+	res.BatchNo = result.BatchNo
+	res.BatchType = result.BatchType
+	res.Total = result.TotalRows
+	return res
+}