添加 README.md

README.md

@@ -0,0 +1,197 @@
+## 安装
+
+```bash
+$ go get  gitea.cdlsxd.cn/self-tools/l_excel_export
+```
+
+
+## 使用
+```go
+	func (s *GoodsService) ExportGoodsV2(ctx http.Context) error {
+var (
+pbRequest     pb.GetGoodsReqs
+goodsIds      []int32
+goodsMediaMap = make(map[int32][]string)
+data          []map[string]interface{}
+)
+//fmt.Printf("begin----%s", time.Now().Format(time.TimeOnly))
+request := ctx.Request().Body
+body, err := io.ReadAll(request)
+if err != nil {
+return errors.New("参数解析失败")
+}
+err = sonic.Unmarshal(body, &pbRequest)
+if err != nil {
+return errors.New("参数解析失败")
+}
+reply, err := s.GoodsBiz.Range(ctx, &pbRequest)
+if err != nil {
+return err
+}
+if len(reply.Data) == 0 {
+return errors.New("未获取到商品数据")
+}
+for _, v := range reply.Data {
+goodsIds = append(goodsIds, v.Id)
+}
+
+bestGoodsSupplierMap, err := goods.NewGoodsSupplierRelationBiz(types.ToTmplConf(s.c)).GetGoodsBestSupplierInfo(ctx, goodsIds, 0, false)
+
+mediaList, err := goods.NewGoodsMediaBiz(types.ToTmplConf(s.c)).Range(ctx, &pb.GetGoodsMediaListReqs{
+Search: &pb.GoodsMediaSearch{GoodsIds: goodsIds},
+})
+if err != nil {
+return err
+}
+for _, media := range mediaList.Data {
+
+goodsMediaMap[media.GoodsId] = append(goodsMediaMap[media.GoodsId], media.Url)
+}
+//获取分类信息
+
+cateMap, err := goods.NewGoodsCateGoryRelationBiz(types.ToTmplConf(s.c)).GoodsRelationMap(ctx, goodsIds)
+if err != nil {
+return err
+}
+cateNameMap := make(map[int32][]string, len(cateMap))
+for k, v := range cateMap {
+for _, cate := range v {
+cateNameMap[k] = append(cateNameMap[k], cate.Name)
+}
+}
+
+// 循环生成记录
+for _, v := range reply.Data {
+var (
+cate      string
+media     string
+supplier  string
+wareHouse string
+costPrice float64
+)
+if _, exist := cateNameMap[v.Id]; exist {
+cate = strings.Join(cateNameMap[v.Id], ",")
+}
+if _, exist := goodsMediaMap[v.Id]; exist {
+media = strings.Join(goodsMediaMap[v.Id], ",")
+}
+if _, exist := bestGoodsSupplierMap[v.Id]; exist {
+supplier = bestGoodsSupplierMap[v.Id].SupplierName
+wareHouse = bestGoodsSupplierMap[v.Id].WarehouseName
+costPrice = bestGoodsSupplierMap[v.Id].CostPrice
+}
+data = append(data, map[string]interface{}{
+"0":  v.GoodsBarCode,
+"1":  cate,
+"2":  v.Title,
+"3":  v.GoodsNum,
+"4":  v.GoodsCode,
+"5":  v.Brand,
+"6":  v.Unit,
+"7":  v.GoodsAttributes,
+"8":  v.Introduction,
+"9":  fmt.Sprintf("%d", v.SellByDate),
+"10": v.SellByDateUnit,
+"11": v.ExternalUrl,
+"12": media,
+"13": fmt.Sprintf("%.2f", v.ExternalPrice),
+"14": fmt.Sprintf("%.2f", v.SalesPrice),
+"15": fmt.Sprintf("%.2f", costPrice),
+"16": fmt.Sprintf("%.2f", v.TaxRate),
+"17": supplier,
+"18": wareHouse,
+"19": v.Remark,
+"20": fmt.Sprintf("%.2f", v.Length),
+"21": fmt.Sprintf("%.2f", v.Width),
+"22": fmt.Sprintf("%.2f", v.Height),
+"23": fmt.Sprintf("%.2f", v.Weight),
+})
+}
+taskId, err := excel_export.NewExport(excel_export.WithJobName("goods_export")).Run(&excel_export.Config{FileName: "商品导出",
+Data: data,
+Ext:  excel_export.Xlsx,
+Head: []*excel_export.FiledMapping{
+{FieldName: "0", ColName: "条码"},
+{FieldName: "1", ColName: "分类名称"},
+{FieldName: "2", ColName: "货品名称"},
+{FieldName: "3", ColName: "货品编号 "},
+{FieldName: "4", ColName: "商品货号"},
+{FieldName: "5", ColName: "品牌"},
+{FieldName: "6", ColName: "单位 "},
+{FieldName: "7", ColName: "规格参数"},
+{FieldName: "8", ColName: "货品说明"},
+{FieldName: "9", ColName: "保质期 "},
+{FieldName: "10", ColName: "保质期单位"},
+{FieldName: "11", ColName: "链接"},
+{FieldName: "12", ColName: "货品图片 "},
+{FieldName: "13", ColName: "电商销售价格"},
+{FieldName: "14", ColName: "销售价"},
+{FieldName: "15", ColName: "供应商报价 "},
+{FieldName: "16", ColName: "税率"},
+{FieldName: "17", ColName: "默认供应商"},
+{FieldName: "18", ColName: "默认存放仓库 "},
+{FieldName: "19", ColName: "备注"},
+{FieldName: "20", ColName: "长"},
+{FieldName: "21", ColName: "宽 "},
+{FieldName: "22", ColName: "高"},
+{FieldName: "23", ColName: "重量"},
+},
+SavePath: "./log/export/excel",
+})
+
+if err != nil {
+return err
+}
+//fmt.Println(time.Now().Format(time.TimeOnly))
+return ctx.Result(200, map[string]interface{}{
+"task_id": taskId,
+})
+}
+
+func (s *GoodsService) ExportTaskInfo(ctx http.Context) error {
+job_name := ctx.Query().Get("job_name")
+taskId := ctx.Query().Get("task_id")
+if taskId == "" {
+return errors.New("任务id不能为空")
+}
+if job_name == "" {
+job_name = "goods_export"
+}
+info, err := excel_export.NewExport(excel_export.WithJobName(job_name)).TaskInfo(taskId)
+if err != nil {
+return err
+}
+return ctx.Result(200, info)
+}
+
+func (s *GoodsService) ExportTaskHIs(ctx http.Context) error {
+job_name := ctx.Query().Get("job_name")
+page := ctx.Query().Get("page")
+limit := ctx.Query().Get("limit")
+pageInt, err := strconv.ParseInt(page, 10, 64)
+numInt, err := strconv.ParseInt(limit, 10, 64)
+if err != nil {
+return err
+}
+if job_name == "" {
+job_name = "goods_export"
+}
+res, err := excel_export.NewExport(excel_export.WithJobName(job_name)).TaskHis(int(pageInt), int(numInt))
+if err != nil {
+return err
+}
+return ctx.Result(200, res)
+}
+
+func (s *GoodsService) ExportDownloadExcel(ctx http.Context) error {
+job_name := ctx.Query().Get("job_name")
+taskId := ctx.Query().Get("task_id")
+if taskId == "" {
+return errors.New("任务id不能为空")
+}
+if job_name == "" {
+job_name = "goods_export"
+}
+return excel_export.NewExport(excel_export.WithJobName(job_name)).DownloadExcel(ctx.Response(), taskId)
+}
+```

export.go

@@ -3,13 +3,17 @@ package excel_export
 import (
 	"encoding/json"
 	"fmt"
+	"gitea.cdlsxd.cn/self-tools/l_excel_export/export_err"
 	"gitea.cdlsxd.cn/self-tools/l_excel_export/pkg"
 	"gitea.cdlsxd.cn/self-tools/l_excel_export/types"
-	"time"
-
-	"gitea.cdlsxd.cn/self-tools/l_excel_export/export_err"
 	"github.com/xuri/excelize/v2"
+
+	"io"
+	"math"
+	"net/http"
 	"os"
+	"sort"
+	"time"
 )
 
 type ExportExcel struct {
@@ -25,19 +29,17 @@ type ExportExcel struct {
 	header     []interface{}
 }
 
-func NewExport(config *Config, opts ...Option) (*ExportExcel, error) {
+func NewExport(opts ...Option) *ExportExcel {
-	export := &ExportExcel{
+	e := &ExportExcel{}
-		config: config,
-	}
 	for _, opt := range opts {
-		opt(export) // 应用选项
+		opt(e) // 应用选项
 	}
-	err := export.check()
+	e.initJobName()
-	return export, err
+	return e
 }
 
-func (e *ExportExcel) Run() (taskId string, err error) {
+func (e *ExportExcel) Run(config *Config) (taskId string, err error) {
-	err = e.init()
+	err = e.init(config)
 	if err != nil {
 		return
 	}
@@ -66,6 +68,81 @@ func (e *ExportExcel) Run() (taskId string, err error) {
 	return e.task.TaskId, nil
 }
 
+func (e *ExportExcel) TaskInfo(task_id string) (task *types.Task, err error) {
+
+	e.logPath, err = pkg.DefaultLogPath(e.jobName)
+
+	if err != nil {
+		return nil, err
+	}
+	e.task = &types.Task{
+		TaskId: task_id,
+	}
+
+	taskInfo, err := os.ReadFile(e.logFile())
+	if err != nil {
+		return nil, err
+	}
+	_ = json.Unmarshal(taskInfo, e.task)
+	return e.task, nil
+}
+
+func (e *ExportExcel) DownloadExcel(w http.ResponseWriter, task_id string) (err error) {
+	task, err := e.TaskInfo(task_id)
+	if err != nil {
+		return err
+	}
+	_, exist := os.Stat(task.Url)
+	if exist != nil {
+		return fmt.Errorf("文件不存在")
+	}
+
+	file, err := os.Open(task.Url)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+	payload, err := io.ReadAll(file)
+	if err != nil {
+		return err
+	}
+	// 设置HTTP响应头
+	// 打开为预览
+	//ctx.Response().Header().Set("Content-Type", "image/png")
+	// 打开为下载
+	w.Header().Set("Content-Type", "application/octet-stream")
+	w.Header().Set("Content-Disposition", "attachment; filename="+task.Url)
+	// 将结果写入
+	_, err = w.Write(payload)
+	return
+}
+
+func (e *ExportExcel) TaskHis(page int, num int) (res *types.ResPage, err error) {
+	var data []*types.Task
+	e.logPath, err = pkg.DefaultLogPath(e.jobName)
+	if err != nil {
+		return res, err
+	}
+
+	entries := e.SortLogFileWithStatus()
+	count := len(entries)
+
+	begin := (page - 1) * num
+	entEnd := begin + num
+	if count < entEnd {
+		entEnd = count
+	}
+	data = entries[begin:entEnd]
+
+	return &types.ResPage{
+		Page:     page,
+		Limit:    num,
+		Total:    count,
+		Data:     data,
+		LastPage: int(math.Ceil(float64(count) / float64(num))),
+	}, nil
+}
+
 func (e *ExportExcel) run(f *excelize.File) (err error) {
 
 	index, err := f.NewStreamWriter(e.getSheetName())
@@ -106,7 +183,13 @@ func (e *ExportExcel) getUrl() string {
 	return fmt.Sprintf("%s/%s_%s%s", e.config.SavePath, e.config.FileName, e.task.TaskId, e.config.Ext)
 }
 
-func (e *ExportExcel) init() error {
+func (e *ExportExcel) init(config *Config) error {
+	e.config = config
+
+	err := e.check()
+	if err != nil {
+		return err
+	}
 	e.task = &types.Task{
 		TaskId:  pkg.CreateTaskId(),
 		Process: 0,
@@ -114,9 +197,8 @@ func (e *ExportExcel) init() error {
 		Ctime:   time.Now().Format(time.DateTime),
 		Ftime:   "",
 		Status:  Init,
-		FileAddr: "",
 	}
-	err := e.updateTask()
+	err = e.updateTask()
 	if err != nil {
 		return err
 	}
@@ -195,9 +277,7 @@ func (e *ExportExcel) path() string {
 }
 
 func (e *ExportExcel) check() (err error) {
-	if len(e.jobName) == 0 {
+
-		e.jobName = time.Now().Format("default")
-	}
 	if e.config.SavePath == "" && e.saveFunc == nil {
 		return export_err.ErrNotSetSaveWay
 	}
@@ -209,9 +289,40 @@ func (e *ExportExcel) check() (err error) {
 	return
 }
 
+func (e *ExportExcel) initJobName() {
+	if len(e.jobName) == 0 {
+		e.jobName = "default"
+	}
+}
+
 func (e *ExportExcel) getSheetName() string {
 	if len(e.sheetName) == 0 {
 		e.sheetName = "Sheet1"
 	}
 	return e.sheetName
 }
+
+func (e *ExportExcel) SortLogFileWithStatus() (fileInfoList []*types.Task) {
+
+	// 获取目录中的文件信息
+	d, _ := os.Open(e.logPath)
+	defer d.Close()
+	files, _ := d.ReadDir(0)
+
+	// 填充切片
+	for _, file := range files {
+		fileName := file.Name()
+		bytes, _ := os.ReadFile(e.logPath + "/" + fileName)
+		var info types.Task
+		_ = json.Unmarshal(bytes, &info)
+		fileInfoList = append(fileInfoList, &info)
+	}
+
+	// 根据修改时间对切片进行排序
+
+	sort.Slice(fileInfoList, func(i, j int) bool {
+
+		return fileInfoList[i].Ctime >= fileInfoList[j].Ctime
+	})
+	return fileInfoList
+}

export_test.go

@@ -40,7 +40,9 @@ func TestData(t *testing.T) {
 		Order{OrderNum: "asdasd", OrderID: 15, CusNum: "asdzczxfaxc"},
 	}
 
-	out, err := NewExport(&Config{
+	out := NewExport(WithJobName("a"))
+
+	task_id, err := out.Run(&Config{
 		FileName: "a",
 		Data:     c,
 		Ext:      Xlsx,
@@ -54,10 +56,16 @@ func TestData(t *testing.T) {
 	if err != nil {
 		panic(err)
 	}
-	task_id, err := out.Run()
-	if err != nil {
-		panic(err)
-	}
 	fmt.Println(task_id)
 	select {}
 }
+
+func TestTaskInfo(t *testing.T) {
+	out, err := NewExport(WithJobName("a")).TaskInfo("1740733651965395783")
+	fmt.Println(out, err)
+}
+
+func TestTaskHis(t *testing.T) {
+	out, err := NewExport(WithJobName("a")).TaskHis(1, 2)
+	fmt.Println(out, err)
+}

pkg/pkg.go

@@ -9,6 +9,8 @@ import (
 	"time"
 )
 
+type ()
+
 func DefaultLogPath(jobName string) (string, error) {
 	path, err := os.Getwd()
 	path = fmt.Sprintf("%s/%s/%s", path, "log/export", jobName)

types/types.go

@@ -1,5 +1,10 @@
 package types
 
+import (
+	"io/fs"
+	"time"
+)
+
 type (
 	Task struct {
 		TaskId  string `json:"task_id"`
@@ -8,6 +13,19 @@ type (
 		Ctime   string `json:"ctime"`
 		Ftime   string `json:"ftime"`
 		Status  int8   `json:"status"`
-		FileAddr string `json:"file_addr"`
+	}
+
+	ResPage struct {
+		Page     int     `json:"current_page"`
+		Limit    int     `json:"per_page"`
+		Total    int     `json:"total"`
+		LastPage int     `json:"last_page"`
+		Data     []*Task `json:"data"`
+	}
+
+	FileInfoStatus struct {
+		fs.FileInfo
+		Status int
+		Time   time.Time
 	}
 )