MiFaZhan
93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
package iot_excelize
|
|
|
|
import (
|
|
"github.com/xuri/excelize/v2"
|
|
)
|
|
|
|
type BatchResultRow struct {
|
|
RowNo int
|
|
ItemID string
|
|
Status string
|
|
ErrorReason string
|
|
}
|
|
|
|
type BatchInfoItem struct {
|
|
Field string
|
|
Value string
|
|
}
|
|
|
|
func BuildBatchResultExcel(rows []BatchResultRow, infos []BatchInfoItem) (*excelize.File, error) {
|
|
f := excelize.NewFile()
|
|
leftStyleID, err := f.NewStyle(&excelize.Style{
|
|
Alignment: &excelize.Alignment{
|
|
Horizontal: "left",
|
|
Vertical: "center",
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resultSheet := "导入结果"
|
|
if err := f.SetSheetName("Sheet1", resultSheet); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
headers := []string{"源文件行号", "item_id", "处理状态", "异常原因"}
|
|
if err := f.SetSheetRow(resultSheet, "A1", &headers); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for i, item := range rows {
|
|
values := []interface{}{item.RowNo, item.ItemID, item.Status, item.ErrorReason}
|
|
cell, err := excelize.CoordinatesToCellName(1, i+2)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err = f.SetSheetRow(resultSheet, cell, &values); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
resultEndCell, err := excelize.CoordinatesToCellName(4, len(rows)+1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := f.SetCellStyle(resultSheet, "A1", resultEndCell, leftStyleID); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
infoSheet := "批次信息"
|
|
if _, err := f.NewSheet(infoSheet); err != nil {
|
|
return nil, err
|
|
}
|
|
infoHeaders := []string{"字段", "值"}
|
|
if err := f.SetSheetRow(infoSheet, "A1", &infoHeaders); err != nil {
|
|
return nil, err
|
|
}
|
|
for i, item := range infos {
|
|
values := []interface{}{item.Field, item.Value}
|
|
cell, err := excelize.CoordinatesToCellName(1, i+2)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err = f.SetSheetRow(infoSheet, cell, &values); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
infoEndCell, err := excelize.CoordinatesToCellName(2, len(infos)+1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := f.SetCellStyle(infoSheet, "A1", infoEndCell, leftStyleID); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_ = f.SetColWidth(resultSheet, "A", "A", 14)
|
|
_ = f.SetColWidth(resultSheet, "B", "B", 24)
|
|
_ = f.SetColWidth(resultSheet, "C", "C", 12)
|
|
_ = f.SetColWidth(resultSheet, "D", "D", 40)
|
|
_ = f.SetColWidth(infoSheet, "A", "B", 24)
|
|
f.SetActiveSheet(0)
|
|
return f, nil
|
|
}
|