143 lines
2.6 KiB
Go
143 lines
2.6 KiB
Go
|
package lib
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"SciencesServer/utils"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/tealeg/xlsx"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Excel struct {
|
|||
|
|
File *xlsx.File
|
|||
|
|
|
|||
|
|
name string
|
|||
|
|
sheet string
|
|||
|
|
title []string
|
|||
|
|
content *ExcelContent
|
|||
|
|
//Style
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type ExcelContent struct {
|
|||
|
|
Kind ExcelContentKind
|
|||
|
|
Content []interface{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type ExcelContentKind int
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
ExcelContentKindForStruct ExcelContentKind = iota + 1
|
|||
|
|
ExcelContentKindForSlice
|
|||
|
|
ExcelContentKindForText
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type OptionExcel func(excel *Excel)
|
|||
|
|
|
|||
|
|
func WithExcelName(name string) OptionExcel {
|
|||
|
|
return func(excel *Excel) {
|
|||
|
|
excel.name = name
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func WithExcelSheet(sheet string) OptionExcel {
|
|||
|
|
return func(excel *Excel) {
|
|||
|
|
excel.sheet = sheet
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func WithExcelTitle(title []string) OptionExcel {
|
|||
|
|
return func(excel *Excel) {
|
|||
|
|
excel.title = title
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func WithExcelContent(content *ExcelContent) OptionExcel {
|
|||
|
|
return func(excel *Excel) {
|
|||
|
|
excel.content = content
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func WithExcelStyle() OptionExcel {
|
|||
|
|
return func(excel *Excel) {
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Export 导出
|
|||
|
|
func (this *Excel) Export() error {
|
|||
|
|
this.File = xlsx.NewFile()
|
|||
|
|
sheet, _ := this.File.AddSheet(this.sheet)
|
|||
|
|
row := sheet.AddRow()
|
|||
|
|
cell := new(xlsx.Cell)
|
|||
|
|
|
|||
|
|
style := &xlsx.Style{
|
|||
|
|
//Fill: *xlsx.NewFill("solid", "EFEFDE", "EFEFDE"),
|
|||
|
|
Border: xlsx.Border{RightColor: "FF"},
|
|||
|
|
}
|
|||
|
|
if this.title != nil && len(this.title) > 0 {
|
|||
|
|
for _, v := range this.title {
|
|||
|
|
cell = row.AddCell()
|
|||
|
|
cell.Value = v
|
|||
|
|
cell.SetStyle(style)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if this.content != nil {
|
|||
|
|
for _, v := range this.content.Content {
|
|||
|
|
row = sheet.AddRow()
|
|||
|
|
|
|||
|
|
if this.content.Kind == ExcelContentKindForStruct {
|
|||
|
|
row.WriteStruct(v, -1)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if this.content.Kind == ExcelContentKindForSlice {
|
|||
|
|
row.WriteSlice(v, -1)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
for i := 0; i < len(this.title); i++ {
|
|||
|
|
cell = row.AddCell()
|
|||
|
|
// TODO：存在不同类型，请自行加判断
|
|||
|
|
switch v.(type) {
|
|||
|
|
case []string:
|
|||
|
|
cell.Value = v.([]string)[i]
|
|||
|
|
break
|
|||
|
|
case []int:
|
|||
|
|
cell.Value = fmt.Sprintf("%v", v.([]int)[i])
|
|||
|
|
break
|
|||
|
|
case []interface{}:
|
|||
|
|
cell.Value = fmt.Sprintf("%v", v.([]interface{})[i])
|
|||
|
|
break
|
|||
|
|
default:
|
|||
|
|
return errors.New("未知数据类型")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Save 保存
|
|||
|
|
func (this *Excel) Save(path string) error {
|
|||
|
|
return this.File.Save(this.name + ".xlsx")
|
|||
|
|
isExist, err := utils.PathExists(path)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
if !isExist {
|
|||
|
|
if err = utils.MkdirAll(path); err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
fmt.Printf("%s/%s.xlsx", path, this.name)
|
|||
|
|
return nil
|
|||
|
|
//return this.Save(fmt.Sprintf("%s\%s.xlsx", path, this.name))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewExcel(options ...OptionExcel) *Excel {
|
|||
|
|
out := new(Excel)
|
|||
|
|
|
|||
|
|
for _, opt := range options {
|
|||
|
|
opt(out)
|
|||
|
|
}
|
|||
|
|
return out
|
|||
|
|
}
|