124 lines
3.6 KiB
Go
124 lines
3.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"electricity_bill_calc/excel"
|
|
"electricity_bill_calc/response"
|
|
"electricity_bill_calc/security"
|
|
"electricity_bill_calc/service"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
func InitializeEndUserController(router *gin.Engine) {
|
|
router.GET("/report/:rid/submeter", security.EnterpriseAuthorize, fetchEndUserInReport)
|
|
router.GET("/report/:rid/meter/template", security.EnterpriseAuthorize, downloadEndUserRegisterTemplate)
|
|
router.POST("/report/:rid/meter/batch", security.EnterpriseAuthorize, uploadEndUserRegisterTemplate)
|
|
}
|
|
|
|
func fetchEndUserInReport(c *gin.Context) {
|
|
result := response.NewResult(c)
|
|
requestReportId := c.Param("rid")
|
|
if !ensureReportBelongs(c, result, requestReportId) {
|
|
return
|
|
}
|
|
keyword := c.DefaultQuery("keyword", "")
|
|
requestPage, err := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
if err != nil {
|
|
result.NotAccept("查询参数[page]格式不正确。")
|
|
return
|
|
}
|
|
endUsers, totalItem, err := service.EndUserService.SearchEndUserRecord(requestReportId, keyword, requestPage)
|
|
if err != nil {
|
|
result.NotFound(err.Error())
|
|
return
|
|
}
|
|
result.Json(
|
|
http.StatusOK,
|
|
"已获取到符合条件的终端用户集合",
|
|
response.NewPagedResponse(requestPage, totalItem).ToMap(),
|
|
gin.H{"meters": endUsers},
|
|
)
|
|
}
|
|
|
|
func downloadEndUserRegisterTemplate(c *gin.Context) {
|
|
result := response.NewResult(c)
|
|
requestReportId := c.Param("rid")
|
|
if !ensureReportBelongs(c, result, requestReportId) {
|
|
return
|
|
}
|
|
users, err := service.EndUserService.AllEndUserRecord(requestReportId)
|
|
if err != nil {
|
|
result.NotFound(err.Error())
|
|
return
|
|
}
|
|
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=抄表记录.xlsx")
|
|
|
|
gen := lo.Ternary[excel.ExcelTemplateGenerator](
|
|
meterType == 0,
|
|
excel.NewMeterNonPVExcelTemplateGenerator(),
|
|
excel.NewMeterPVExcelTemplateGenerator(),
|
|
)
|
|
defer gen.Close()
|
|
gen.WriteMeterData(users)
|
|
gen.WriteTo(c.Writer)
|
|
}
|
|
|
|
func uploadEndUserRegisterTemplate(c *gin.Context) {
|
|
result := response.NewResult(c)
|
|
requestReportId := c.Param("rid")
|
|
if !ensureReportBelongs(c, result, requestReportId) {
|
|
return
|
|
}
|
|
meterType, err := service.ReportService.RetreiveParkEndUserMeterType(requestReportId)
|
|
if err != nil {
|
|
result.Error(http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
if meterType == -1 {
|
|
result.NotFound("未能确定用户表计类型。")
|
|
return
|
|
}
|
|
|
|
uploadedFile, err := c.FormFile("data")
|
|
if err != nil {
|
|
result.NotAccept("没有接收到上传的档案文件。")
|
|
return
|
|
}
|
|
archiveFile, err := uploadedFile.Open()
|
|
if err != nil {
|
|
result.Error(http.StatusInternalServerError, err.Error())
|
|
return
|
|
}
|
|
if meterType == 0 {
|
|
errs := service.EndUserService.BatchImportNonPVRegister(requestReportId, archiveFile)
|
|
if errs.Len() > 0 {
|
|
result.Json(http.StatusInternalServerError, "上传抄表文件存在解析错误", gin.H{"errors": errs.Errs})
|
|
return
|
|
}
|
|
} else {
|
|
errs := service.EndUserService.BatchImportPVRegister(requestReportId, archiveFile)
|
|
if errs.Len() > 0 {
|
|
result.Json(http.StatusInternalServerError, "上传抄表文件存在解析错误", gin.H{"errors": errs.Errs})
|
|
return
|
|
}
|
|
}
|
|
result.Json(http.StatusOK, "已经成功完成抄表记录的导入。", gin.H{"errors": make([]error, 0)})
|
|
}
|