/
export.go
56 lines (47 loc) · 1.15 KB
/
export.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package xlsx
import (
"fmt"
_errors "github.com/pkg/errors"
"github.com/xuri/excelize/v2"
)
type export struct {
a1 []string
rows [][]interface{}
a1Length int
rowsLength int
sheetName string
}
func NewExport() *export {
return &export{}
}
func (e *export) SetSheetName(name string) *export {
e.sheetName = name
return e
}
func (e *export) SetA1(a1 []string) *export {
e.a1 = a1
e.a1Length = len(e.a1)
return e
}
func (e *export) SetRows(rows [][]interface{}) *export {
e.rows = rows
e.rowsLength = len(e.rows)
return e
}
// Excel 导出Excel文件
// headers 列名切片, 表头
// rows 数据切片,是一个二维数组
func (e *export) Excel() (*excelize.File, error) {
excel := excelize.NewFile()
excel.SetSheetName("Sheet1", e.sheetName)
if err := excel.SetSheetRow(e.sheetName, "A1", &e.a1); err != nil {
return excel, _errors.Wrap(err, "A1行数据 填充失败!")
}
for j := 0; j < e.rowsLength; j++ {
index := fmt.Sprintf("A%d", j+2)
if err := excel.SetSheetRow(e.sheetName, index, &e.rows[j]); err != nil {
return excel, _errors.Wrap(err, "excel.SetSheetRow() 填充数据失败!")
}
}
return excel, nil
}