MiFaZhan
88 lines
2.7 KiB
Go
88 lines
2.7 KiB
Go
package lifecycle
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchimport"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
|
|
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/encoding/gjson"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
type Processor struct {
|
|
config OperationConfig
|
|
}
|
|
|
|
func NewProcessor(operation Operation) (*Processor, error) {
|
|
config, err := operation.Config()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Processor{config: config}, nil
|
|
}
|
|
|
|
func (p *Processor) Config() batchimport.ProcessorConfig {
|
|
return batchimport.ProcessorConfig{
|
|
Code: "sim_card_" + p.config.Code,
|
|
BatchType: p.config.BatchType,
|
|
BatchNoPrefix: p.config.BatchPrefix,
|
|
TaskName: p.config.TaskName,
|
|
ExportScene: p.config.ExportScene,
|
|
ImportDir: p.config.ImportDir,
|
|
ResultDir: p.config.ResultDir,
|
|
MaxFileSize: consts.SimCardLifecycleMaxFileSize,
|
|
MaxRows: consts.SimCardBatchMaxRows,
|
|
TemplateHeaders: []string{"ICCID", "MSISDN", "IMSI"},
|
|
RequiredHeaders: []string{"ICCID"},
|
|
}
|
|
}
|
|
|
|
func (p *Processor) ParseRows(ctx context.Context, filePath string) ([]batchimport.Row, error) {
|
|
return ParseExcelFileRows(ctx, filePath)
|
|
}
|
|
|
|
func (p *Processor) ValidateRows(_ context.Context, rows []batchimport.Row) error {
|
|
return validateBasicRows(rows)
|
|
}
|
|
|
|
func (p *Processor) Execute(context.Context, gdb.TX, *batchimport.SubmitRequest, []batchimport.Row) error {
|
|
return gerror.New("停复机批次只能由异步执行器执行")
|
|
}
|
|
|
|
func (p *Processor) BuildBatchInfo(_ context.Context, req batchimport.SubmitRequest, result *batchimport.Result, importOss *model.SysOssSaveContentModel, resultOss *model.SysOssSaveContentModel, downloadID int64) *gjson.Json {
|
|
info := g.Map{
|
|
"operation": p.config.Code,
|
|
}
|
|
if p.config.OperType != "" {
|
|
info["operType"] = p.config.OperType
|
|
}
|
|
if p.config.Operation == OperationRestart {
|
|
info["stopOperType"] = consts.SimCardStatusOperTypeStop
|
|
info["resumeOperType"] = consts.SimCardStatusOperTypeStart
|
|
info["resumeDelaySeconds"] = 300
|
|
info["restart"] = g.Map{"phase": "created"}
|
|
}
|
|
if options, ok := req.BizParams.(*SubmitOptions); ok && options != nil {
|
|
if remark := strings.TrimSpace(options.Remark); remark != "" {
|
|
info["remark"] = remark
|
|
}
|
|
}
|
|
if result != nil && result.ImportFileOssId > 0 {
|
|
info["importFileOssId"] = result.ImportFileOssId
|
|
} else if importOss != nil && importOss.OssId > 0 {
|
|
info["importFileOssId"] = importOss.OssId
|
|
}
|
|
if resultOss != nil {
|
|
info["resultFileOssId"] = resultOss.OssId
|
|
}
|
|
if downloadID > 0 {
|
|
info["resultDownloadId"] = downloadID
|
|
}
|
|
return gjson.New(info)
|
|
}
|