|
|
// Package export 导出组件的封装，一般导出的话csv性能是360插件的xlsx的几十倍
|
|
|
// 但是xlsx可以更加丰富样式，所以没有样式的情况下，用csv性能更高
|
|
|
package export
|
|
|
|
|
|
import (
|
|
|
"context"
|
|
|
"fmt"
|
|
|
"github.com/rs/xid"
|
|
|
"os"
|
|
|
)
|
|
|
|
|
|
type HeaderFieldOne struct {
|
|
|
Field string //数据对应的字段名称
|
|
|
Title string //表头名称
|
|
|
}
|
|
|
|
|
|
type HeaderField []HeaderFieldOne
|
|
|
|
|
|
type Export interface {
|
|
|
FileSuffix() string //文件后缀
|
|
|
ExportFile(data []RowData) error //导出数据
|
|
|
}
|
|
|
|
|
|
type BaseExport struct {
|
|
|
Ctx context.Context
|
|
|
hashMark string //导出任务的唯一hash
|
|
|
Header HeaderField //表头数据
|
|
|
Total int64 //需要导出的总数
|
|
|
filename string //文件名称
|
|
|
fullPath string //文件存储路径
|
|
|
customFile bool //是否自定义文件
|
|
|
}
|
|
|
|
|
|
// Init 实例化导出
|
|
|
// @params
|
|
|
//
|
|
|
// header HeaderField excel表头
|
|
|
// total int64 导出数据的总量条数
|
|
|
// hashMark string 自定义的唯一标识
|
|
|
func (e *BaseExport) Init(ctx context.Context, header HeaderField, total int64, opts ...Option) {
|
|
|
e.Ctx = ctx
|
|
|
e.Total = total
|
|
|
e.Header = header
|
|
|
e.hashMark = BuildHashMark()
|
|
|
|
|
|
//设置参数
|
|
|
for _, opt := range opts {
|
|
|
opt(e)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// GetHashMark 获取导出的唯一标识
|
|
|
func (e *BaseExport) GetHashMark() string {
|
|
|
return e.hashMark
|
|
|
}
|
|
|
|
|
|
// GetBasePath 获取基础路径
|
|
|
func (e *BaseExport) GetBasePath() string {
|
|
|
path, _ := os.Getwd()
|
|
|
return path + "/tmp/export/"
|
|
|
}
|
|
|
|
|
|
// SetFilename 设置文件名称
|
|
|
func (e *BaseExport) SetFilename(ept Export) {
|
|
|
if e.filename != "" { //已经设置好了就不设置了
|
|
|
return
|
|
|
}
|
|
|
e.filename = e.GetHashMark() + "." + ept.FileSuffix()
|
|
|
e.fullPath = e.GetBasePath() + e.filename
|
|
|
}
|
|
|
|
|
|
// GetFilename 获取文件名称
|
|
|
func (e *BaseExport) GetFilename() string {
|
|
|
return e.filename
|
|
|
}
|
|
|
|
|
|
// GetFullPath 获取文件全路径
|
|
|
func (e *BaseExport) GetFullPath() string {
|
|
|
return e.fullPath
|
|
|
}
|
|
|
|
|
|
// IsCustomerFile 文件是否已经存在
|
|
|
func (e *BaseExport) IsCustomerFile() bool {
|
|
|
return e.customFile
|
|
|
}
|
|
|
|
|
|
// BuildHashMark 初始化当前导出任务的唯一标识
|
|
|
func BuildHashMark() string {
|
|
|
return fmt.Sprintf("ept_%s", xid.New().String())
|
|
|
}
|