统计当前系统中的报表

controller/statistics.go

@@ -2,6 +2,7 @@ package controller
 
 import (
 	"electricity_bill_calc/logger"
+	"electricity_bill_calc/model"
 	"electricity_bill_calc/response"
 	"electricity_bill_calc/security"
 	"electricity_bill_calc/service"
@@ -14,6 +15,7 @@ var StatisticsWithdrawLog = logger.Named("Handler", "StatisticsWithdraw")
 
 func InitializeStatisticsController(router *fiber.App) {
 	router.Get("/audits", security.OPSAuthorize, currentAuditAmount)
+	router.Get("/stat/reports", statReports)
 
 }
 
@@ -31,4 +33,53 @@ func currentAuditAmount(c *fiber.Ctx) error {
 		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,
+		},
+	})
+}