forked from free-lancers/electricity_bill_calc_service
enhance(report):导出用户抄表模板完成,提取公共生成器类型。
This commit is contained in:
@@ -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)
|
||||
}
|
||||
|
Reference in New Issue
Block a user