enhance(meter):部署抄表模板下载功能。

excel/meter_reading.go

@@ -1,10 +1,14 @@
 package excel
 
 import (
+	"electricity_bill_calc/logger"
 	"electricity_bill_calc/model"
+	"electricity_bill_calc/tools"
+	"fmt"
 	"io"
 
 	"github.com/xuri/excelize/v2"
+	"go.uber.org/zap"
 )
 
 var meterReadingsRecognizers = []*ColumnRecognizer{
@@ -22,10 +26,117 @@ func NewMeterReadingsExcelAnalyzer(file io.Reader) (*ExcelAnalyzer[model.Reading
 
 type MeterReadingsExcelTemplateGenerator struct {
 	file *excelize.File
+	log  *zap.Logger
 }
 
 func NewMeterReadingsExcelTemplateGenerator() *MeterReadingsExcelTemplateGenerator {
 	return &MeterReadingsExcelTemplateGenerator{
 		file: excelize.NewFile(),
+		log:  logger.Named("Excel", "MeterReadings"),
 	}
 }
+
+func (MeterReadingsExcelTemplateGenerator) titles() *[]interface{} {
+	return &[]interface{}{
+		"抄表序号",
+		"抄表时间",
+		"表计编号",
+		"表计名称",
+		"商户名称",
+		"倍率",
+		"有功（总）",
+		"有功（尖）",
+		"有功（峰）",
+		"有功（谷）",
+	}
+}
+
+func (g MeterReadingsExcelTemplateGenerator) Close() {
+	g.file.Close()
+}
+
+func (g MeterReadingsExcelTemplateGenerator) WriteTo(w io.Writer) (int64, error) {
+	return g.file.WriteTo(w)
+}
+
+func (g MeterReadingsExcelTemplateGenerator) WriteTemplateData(meters []*model.SimpleMeterDocument) error {
+	var err error
+	defaultSheet := g.file.GetSheetName(0)
+	g.log.Debug("选定默认输出表格", zap.String("sheet", defaultSheet))
+	err = g.file.SetColWidth(defaultSheet, "A", "E", 30)
+	if err != nil {
+		g.log.Error("未能设定长型单元格的宽度。", zap.Error(err))
+		return fmt.Errorf("未能设定长型单元格的宽度，%w", err)
+	}
+	err = g.file.SetColWidth(defaultSheet, "F", "F", 10)
+	if err != nil {
+		g.log.Error("未能设定倍率单元格的宽度。", zap.Error(err))
+		return fmt.Errorf("未能设定倍率单元格的宽度，%w", err)
+	}
+	err = g.file.SetColWidth(defaultSheet, "G", "J", 20)
+	if err != nil {
+		g.log.Error("未能设定短型单元格的宽度。", zap.Error(err))
+		return fmt.Errorf("未能设定短型单元格的宽度，%w", err)
+	}
+	err = g.file.SetSheetRow(defaultSheet, "A1", g.titles())
+	if err != nil {
+		g.log.Error("未能输出模板标题。", zap.Error(err))
+		return fmt.Errorf("未能输出模板标题，%w", err)
+	}
+	err = g.file.SetRowHeight(defaultSheet, 1, 30)
+	if err != nil {
+		g.log.Error("未能设定标题行的高度。", zap.Error(err))
+		return fmt.Errorf("未能设定标题行的高度，%w", err)
+	}
+
+	dateTimeExp := "yyyy-mm-dd hh:mm;@"
+	dateTimeColStyle, err := g.file.NewStyle(&excelize.Style{
+		CustomNumFmt: &dateTimeExp,
+	})
+	if err != nil {
+		g.log.Error("未能创建日期时间格式。", zap.Error(err))
+		return fmt.Errorf("未能创建日期时间格式，%w", err)
+	}
+	g.file.SetCellStyle(defaultSheet, "B2", "B1048576", dateTimeColStyle)
+
+	numExp := "0.0000;@"
+	numColStyle, err := g.file.NewStyle(&excelize.Style{
+		CustomNumFmt: &numExp,
+	})
+	if err != nil {
+		g.log.Error("未能创建抄表数字格式。", zap.Error(err))
+		return fmt.Errorf("未能创建抄表数字格式，%w", err)
+	}
+	g.file.SetCellStyle(defaultSheet, "F2", "J1048576", numColStyle)
+
+	stream, err := g.file.NewStreamWriter(defaultSheet)
+	if err != nil {
+		g.log.Error("未能创建流式写入器。", zap.Error(err))
+		return fmt.Errorf("未能创建流式写入器，%w", err)
+	}
+
+	for i, meter := range meters {
+		startCell, err := excelize.CoordinatesToCellName(1, i+2)
+		if err != nil {
+			g.log.Error("未能定位输出数据的起始单元格。", zap.Error(err))
+			return fmt.Errorf("未能定位输出数据的起始单元格，%w", err)
+		}
+		if err := stream.SetRow(startCell, []interface{}{
+			meter.Seq,
+			"",
+			meter.Code,
+			tools.DefaultTo(meter.Address, ""),
+			tools.DefaultTo(meter.TenementName, ""),
+			meter.Ratio,
+		}); err != nil {
+			g.log.Error("向模板写入数据出现错误。", zap.Error(err))
+			return fmt.Errorf("向模板写入数据出现错误，%w", err)
+		}
+	}
+	if err = stream.Flush(); err != nil {
+		g.log.Error("未能刷新流式写入器。", zap.Error(err))
+		return fmt.Errorf("未能刷新流式写入器，%w", err)
+	}
+
+	return err
+}