iotlcm/server/internal/logic/iot/simcardbatch/cancel/submitter.go

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
}