Compare commits
2 Commits
| Author | SHA1 | Date |
|---|---|---|
 renzhiyuan
|
aeb9415fda | |
|
renzhiyuan
|
f4756089f2 |
|
|
@ -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)
|
||||
}
|
||||
```
|
||||
161
export.go
161
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) {
|
||||
export := &ExportExcel{
|
||||
config: config,
|
||||
}
|
||||
func NewExport(opts ...Option) *ExportExcel {
|
||||
e := &ExportExcel{}
|
||||
for _, opt := range opts {
|
||||
opt(export) // 应用选项
|
||||
opt(e) // 应用选项
|
||||
}
|
||||
err := export.check()
|
||||
return export, err
|
||||
e.initJobName()
|
||||
return e
|
||||
}
|
||||
|
||||
func (e *ExportExcel) Run() (taskId string, err error) {
|
||||
err = e.init()
|
||||
func (e *ExportExcel) Run(config *Config) (taskId string, err error) {
|
||||
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,17 +183,22 @@ 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 {
|
||||
e.task = &types.Task{
|
||||
TaskId: pkg.CreateTaskId(),
|
||||
Process: 0,
|
||||
Url: "",
|
||||
Ctime: time.Now().Format(time.DateTime),
|
||||
Ftime: "",
|
||||
Status: Init,
|
||||
FileAddr: "",
|
||||
func (e *ExportExcel) init(config *Config) error {
|
||||
e.config = config
|
||||
|
||||
err := e.check()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err := e.updateTask()
|
||||
e.task = &types.Task{
|
||||
TaskId: pkg.CreateTaskId(),
|
||||
Process: 0,
|
||||
Url: "",
|
||||
Ctime: time.Now().Format(time.DateTime),
|
||||
Ftime: "",
|
||||
Status: Init,
|
||||
}
|
||||
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
|
||||
}
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
}
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -1,13 +1,31 @@
|
|||
package types
|
||||
|
||||
import (
|
||||
"io/fs"
|
||||
"time"
|
||||
)
|
||||
|
||||
type (
|
||||
Task struct {
|
||||
TaskId string `json:"task_id"`
|
||||
Url string `json:"url"`
|
||||
Process int `json:"process"`
|
||||
Ctime string `json:"ctime"`
|
||||
Ftime string `json:"ftime"`
|
||||
Status int8 `json:"status"`
|
||||
FileAddr string `json:"file_addr"`
|
||||
TaskId string `json:"task_id"`
|
||||
Url string `json:"url"`
|
||||
Process int `json:"process"`
|
||||
Ctime string `json:"ctime"`
|
||||
Ftime string `json:"ftime"`
|
||||
Status int8 `json:"status"`
|
||||
}
|
||||
|
||||
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
|
||||
}
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in New Issue