enhance(report):导出用户抄表模板完成,提取公共生成器类型。

This commit is contained in:
徐涛
2022-08-22 22:52:17 +08:00
parent ef214c135b
commit 67cfb60515
4 changed files with 134 additions and 3 deletions

View File

@@ -9,6 +9,7 @@ import (
"strconv"
"github.com/gin-gonic/gin"
"github.com/samber/lo"
)
func InitializeEndUserController(router *gin.Engine) {
@@ -52,12 +53,28 @@ func downloadEndUserRegisterTemplate(c *gin.Context) {
result.NotFound(err.Error())
return
}
gen := excel.NewMeterPVExcelTemplateGenerator()
defer gen.Close()
gen.WriteMeterData(users)
meterType, err := service.ReportService.RetreiveParkEndUserMeterType(requestReportId)
if err != nil {
result.Error(http.StatusInternalServerError, err.Error())
return
}
if meterType == -1 {
result.NotFound("未能确定用户表计类型。")
return
}
c.Status(http.StatusOK)
c.Header("Content-Type", "application/octet-stream")
c.Header("Content-Transfer-Encoding", "binary")
c.Header("Content-Disposition", "attachment; filename=抄表记录.xslx")
gen := lo.Ternary[excel.ExcelTemplateGenerator](
meterType == 0,
excel.NewMeterNonPVExcelTemplateGenerator(),
excel.NewMeterPVExcelTemplateGenerator(),
)
defer gen.Close()
gen.WriteMeterData(users)
gen.WriteTo(c.Writer)
}