MiFaZhan
122 lines
3.9 KiB
Go
122 lines
3.9 KiB
Go
// Package gen
|
|
// @Link https://github.com/xiujiecn/xiu-admin
|
|
// @Copyright Copyright (c) 2026 LiXiujie
|
|
// @Author Lxj <xxx@example.com>
|
|
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
|
// @AutoGenerate Version
|
|
package iot
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/logic/iot/simcardbatch/warningconfig"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
|
|
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/utility/iot_excelize"
|
|
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
"github.com/xuri/excelize/v2"
|
|
)
|
|
|
|
type sIotTSimWarningConfig struct{}
|
|
|
|
func NewIotTSimWarningConfig() *sIotTSimWarningConfig {
|
|
return &sIotTSimWarningConfig{}
|
|
}
|
|
|
|
func init() {
|
|
service.RegisterIotTSimWarningConfig(NewIotTSimWarningConfig())
|
|
}
|
|
|
|
func (s *sIotTSimWarningConfig) DownloadExcelTemplate(ctx context.Context, itemType int) error {
|
|
f := excelize.NewFile()
|
|
defer func() { _ = f.Close() }()
|
|
|
|
processor := warningconfig.NewProcessor(itemType)
|
|
headers := processor.Config().TemplateHeaders
|
|
if err := f.SetSheetRow("Sheet1", "A1", &headers); err != nil {
|
|
return err
|
|
}
|
|
if err := iot_excelize.ApplyHeaderColumnsTextStyle(f); err != nil {
|
|
return err
|
|
}
|
|
fileName := "sim_warning_config_template"
|
|
if len(headers) > 0 && headers[0] == "流量池编号" {
|
|
fileName = "sim_flow_pool_warning_config_template"
|
|
}
|
|
return iot_excelize.WriteHTTP(ctx, f, fileName)
|
|
}
|
|
|
|
func (s *sIotTSimWarningConfig) SendEmailCode(ctx context.Context, in *model.TSimWarningConfigEmailCodeParam) error {
|
|
return warningconfig.SendEmailCode(ctx, in)
|
|
}
|
|
|
|
func (s *sIotTSimWarningConfig) SubmitExcel(ctx context.Context, in *model.TSimWarningConfigExcelSubmitParam) (*model.TSimWarningConfigExcelSubmitRes, error) {
|
|
if in == nil {
|
|
in = &model.TSimWarningConfigExcelSubmitParam{}
|
|
}
|
|
file, err := warningConfigUploadFile(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result, err := warningconfig.CreatePendingExcel(ctx, file, in)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return (*model.TSimWarningConfigExcelSubmitRes)(warningConfigSubmitResult(result)), nil
|
|
}
|
|
|
|
func (s *sIotTSimWarningConfig) SubmitInput(ctx context.Context, in *model.TSimWarningConfigInputSubmitParam) (*model.TSimWarningConfigInputSubmitRes, error) {
|
|
if in == nil {
|
|
in = &model.TSimWarningConfigInputSubmitParam{}
|
|
}
|
|
result, err := warningconfig.CreatePendingInput(ctx, in)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return (*model.TSimWarningConfigInputSubmitRes)(warningConfigSubmitResult(result)), nil
|
|
}
|
|
|
|
func (s *sIotTSimWarningConfig) SubmitRange(ctx context.Context, in *model.TSimWarningConfigRangeSubmitParam) (*model.TSimWarningConfigRangeSubmitRes, error) {
|
|
if in == nil {
|
|
in = &model.TSimWarningConfigRangeSubmitParam{}
|
|
}
|
|
result, err := warningconfig.CreatePendingRange(ctx, in)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return (*model.TSimWarningConfigRangeSubmitRes)(warningConfigSubmitResult(result)), nil
|
|
}
|
|
|
|
func warningConfigUploadFile(ctx context.Context) (*ghttp.UploadFile, error) {
|
|
r := g.RequestFromCtx(ctx)
|
|
if r == nil {
|
|
return nil, gerror.New("当前上下文不是HTTP请求")
|
|
}
|
|
file := r.GetUploadFile("file")
|
|
if file == nil {
|
|
return nil, gerror.New("请选择上传文件")
|
|
}
|
|
return file, nil
|
|
}
|
|
|
|
func warningConfigSubmitResult(result *warningconfig.PendingResult) *model.TSimWarningConfigSubmitRes {
|
|
if result == nil {
|
|
return &model.TSimWarningConfigSubmitRes{
|
|
BatchType: consts.BatchTypeSimWarningConfig,
|
|
BatchAsyncStatus: consts.BatchAsyncStatusPending,
|
|
}
|
|
}
|
|
return &model.TSimWarningConfigSubmitRes{
|
|
BatchNo: result.BatchNo,
|
|
BatchType: result.BatchType,
|
|
BatchAsyncStatus: result.BatchAsyncStatus,
|
|
Total: result.Total,
|
|
ValidCount: result.ValidCount,
|
|
FailCount: result.FailCount,
|
|
}
|
|
}
|