sync: 同步 v1.0.4 并脱敏配置与镜像

server/internal/logic/iot/t_sim_card_traffic_cutoff.go

@@ -2,26 +2,23 @@ package iot
 
 import (
 	"context"
-	"fmt"
 	"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/contexts"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/exportcenter"
 	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/logic/iot/simcardbatch/trafficcutoffconfig"
 	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
 	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model/do"
 	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
 	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/utility/convert"
-	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/utility/excel"
 	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/utility/iot_excelize"
 
 	"github.com/gogf/gf/v2/database/gdb"
 	"github.com/gogf/gf/v2/errors/gerror"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/ghttp"
-	"github.com/gogf/gf/v2/os/gctx"
-	"github.com/gogf/gf/v2/util/gconv"
 	"github.com/xuri/excelize/v2"
 )
 
@@ -103,25 +100,32 @@ func (s *sIotTSimCardTrafficCutoff) List(ctx context.Context, in *model.TSimCard
 	return service.IotTSimCardInfo().List(ctx, &cardParam)
 }
 
-func (s *sIotTSimCardTrafficCutoff) Export(ctx context.Context, in *model.TSimCardTrafficCutoffListParam) error {
-	if in == nil {
-		in = &model.TSimCardTrafficCutoffListParam{}
-	}
-	list, totalCount, err := s.List(ctx, in)
-	if err != nil {
-		return err
-	}
-	tags, err := convert.GetEntityDescTags(model.TSimCardTrafficCutoffExportModel{})
-	if err != nil {
-		return err
-	}
-	var exports []*model.TSimCardTrafficCutoffExportModel
-	if err = gconv.Scan(list, &exports); err != nil {
-		return err
-	}
-	fileName := "traffic-cutoff-card-info-" + gctx.CtxId(ctx)
-	sheetName := fmt.Sprintf("traffic cutoff cards total %d, exported %d", totalCount, len(exports))
-	return excel.ExportByStructs(ctx, tags, exports, fileName, sheetName)
+// Export 导出达量断网（异步：提交任务后到下载中心取文件）
+func (s *sIotTSimCardTrafficCutoff) Export(ctx context.Context, in *model.TSimCardTrafficCutoffListParam) (err error) {
+	_, err = exportcenter.Submit(ctx, &exportcenter.Request{
+		BizKey:     "tSimCardTrafficCutoff",
+		BizName:    "达量断网",
+		PermCode:   "cpm:iot:tSimCardTrafficCutoff:export",
+		QueryParam: in,
+		Tags: func() ([]string, error) {
+			return convert.GetEntityDescTags(model.TSimCardInfoListModel{})
+		},
+		Fetch: func(ctx context.Context, page, pageSize int) (rows []interface{}, total int, err error) {
+			query := *in
+			query.Page = page
+			query.PageSize = pageSize
+			list, count, err := s.List(ctx, &query)
+			if err != nil {
+				return nil, 0, err
+			}
+			rows = make([]interface{}, 0, len(list))
+			for _, item := range list {
+				rows = append(rows, item)
+			}
+			return rows, count, nil
+		},
+	})
+	return err
 }
 
 func (s *sIotTSimCardTrafficCutoff) Edit(ctx context.Context, in *model.TSimCardTrafficCutoffEditParam) error {