86 lines
2.7 KiB
Go
86 lines
2.7 KiB
Go
package controller
|
|
|
|
import (
|
|
"electricity_bill_calc/logger"
|
|
"electricity_bill_calc/model"
|
|
"electricity_bill_calc/response"
|
|
"electricity_bill_calc/security"
|
|
"electricity_bill_calc/service"
|
|
"github.com/gofiber/fiber/v2"
|
|
"go.uber.org/zap"
|
|
"net/http"
|
|
)
|
|
|
|
var StatisticsWithdrawLog = logger.Named("Handler", "StatisticsWithdraw")
|
|
|
|
func InitializeStatisticsController(router *fiber.App) {
|
|
router.Get("/audits", security.OPSAuthorize, currentAuditAmount)
|
|
router.Get("/stat/reports", security.OPSAuthorize, statReports)
|
|
|
|
}
|
|
|
|
//获取当前系统中待审核的内容数量
|
|
func currentAuditAmount(c *fiber.Ctx) error {
|
|
StatisticsWithdrawLog.Info("开始获取当前系统中待审核的内容数量")
|
|
result := response.NewResult(c)
|
|
amount, err := service.WithdrawService.AuditWaits()
|
|
if err != nil {
|
|
StatisticsWithdrawLog.Error("获取当前系统中待审核的内容数量出错", zap.Error(err))
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
|
|
return result.Success("已经获取到指定的统计信息",
|
|
fiber.Map{"withdraw": amount})
|
|
}
|
|
|
|
func statReports(c *fiber.Ctx) error {
|
|
result := response.NewResult(c)
|
|
session, err := _retreiveSession(c)
|
|
if err != nil {
|
|
return result.Unauthorized(err.Error())
|
|
}
|
|
var (
|
|
enterprises int64 = 0
|
|
parks int64 = 0
|
|
reports []model.ParkPeriodStatistics
|
|
)
|
|
if session.Type != 0 {
|
|
enterprises, err = service.StatisticsService.EnabledEnterprises()
|
|
if err != nil {
|
|
StatisticsWithdrawLog.Error(err.Error())
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
parks, err = service.StatisticsService.EnabledParks()
|
|
if err != nil {
|
|
StatisticsWithdrawLog.Error(err.Error())
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
//TODO: 2023.07.26 报表数据库结构改变，此处逻辑复杂放在最后处理
|
|
reports, err = service.StatisticsService.ParkNewestState()
|
|
if err != nil {
|
|
StatisticsWithdrawLog.Error(err.Error())
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
} else {
|
|
parks, err = service.StatisticsService.EnabledParks(session.Uid)
|
|
if err != nil {
|
|
StatisticsWithdrawLog.Error(err.Error())
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
//TODO: 2023.07.26 报表数据库结构改变，此处逻辑复杂放在最后处理
|
|
reports, err = service.StatisticsService.ParkNewestState(session.Uid)
|
|
if err != nil {
|
|
StatisticsWithdrawLog.Error(err.Error())
|
|
return result.Error(http.StatusInternalServerError, err.Error())
|
|
}
|
|
}
|
|
|
|
return result.Success("已经完成园区报告的统计。", fiber.Map{
|
|
"statistics": fiber.Map{
|
|
"enterprises": enterprises,
|
|
"parks": parks,
|
|
"reports": reports,
|
|
},
|
|
})
|
|
}
|