From 7be45d3ffc40554a07f1c48d4daa67312b7ba736 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=90=E6=B6=9B?= Date: Tue, 6 Sep 2022 17:29:33 +0800 Subject: [PATCH] =?UTF-8?q?feat(god):=E5=AE=8C=E6=88=90=E5=A4=A9=E7=A5=9E?= =?UTF-8?q?=E6=A8=A1=E5=BC=8F=E7=9A=84=E6=8E=A7=E5=88=B6=E5=99=A8=EF=BC=8C?= =?UTF-8?q?=E6=95=B4=E4=BD=93=E5=8A=9F=E8=83=BD=E5=BE=85=E6=B5=8B=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- controller/god_mode.go | 162 +++++++++++++++++++++++++++++++++++++++++ router/router.go | 1 + 2 files changed, 163 insertions(+) diff --git a/controller/god_mode.go b/controller/god_mode.go index 7848ba2..d622b01 100644 --- a/controller/god_mode.go +++ b/controller/god_mode.go @@ -1,8 +1,11 @@ package controller import ( + "electricity_bill_calc/exceptions" "electricity_bill_calc/response" "electricity_bill_calc/security" + "electricity_bill_calc/service" + "net/http" "github.com/gin-gonic/gin" ) @@ -11,10 +14,169 @@ func InitializeGodModeController(router *gin.Engine) { gmR := router.Group("/gm") { gmR.DELETE("/report/:rid/summary", security.SingularityAuthorize, gmResetReportSummary) + gmR.DELETE("/report/:rid/maintenance", security.SingularityAuthorize, gmResetReportMaintenance) + gmR.DELETE("/report/:rid/meters", security.SingularityAuthorize, gmResetReportEndUserRecord) + gmR.DELETE("/report/:rid", security.SingularityAuthorize, gmResetReport) + gmR.DELETE("/report/:rid/force", security.SingularityAuthorize, gmDeleteReport) + gmR.DELETE("/park/:pid/maintenance/:mid", security.SingularityAuthorize, gmDeleteSpecificMaintenance) + gmR.DELETE("/park/:pid/maintenance", security.SingularityAuthorize, gmDeleteAllMaintenance) + gmR.DELETE("/park/:pid/meters", security.SingularityAuthorize, gmDeleteAllMeters) + gmR.DELETE("/park/:pid/force", security.SingularityAuthorize, gmDeletePark) + gmR.DELETE("/enterprise/:uid/force", security.SingularityAuthorize, gmDeleteUser) } } func gmResetReportSummary(c *gin.Context) { result := response.NewResult(c) + requestReportId := c.Param("rid") + done, err := service.GodModeService.ClearReportSummary(requestReportId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功重置指定报表的园区总览部分。") + return + } result.Success("指定报表的园区总览已经重置。") } + +func gmResetReportMaintenance(c *gin.Context) { + result := response.NewResult(c) + requestReportId := c.Param("rid") + done, err := service.GodModeService.ClearReportMaintenances(requestReportId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功重置指定报表的配电维护费部分。") + return + } + result.Success("指定报表的配电维护费已经重置。") +} + +func gmResetReportEndUserRecord(c *gin.Context) { + result := response.NewResult(c) + requestReportId := c.Param("rid") + done, err := service.GodModeService.ResetEndUserRegisterRecords(requestReportId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功重置指定报表的抄表记录部分。") + return + } + result.Success("指定报表的抄表记录已经重置。") +} + +func gmResetReport(c *gin.Context) { + result := response.NewResult(c) + requestReportId := c.Param("rid") + done, err := service.GodModeService.ResetReport(requestReportId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功重置指定报表。") + return + } + result.Success("指定报表已经重置。") +} + +func gmDeleteReport(c *gin.Context) { + result := response.NewResult(c) + requestReportId := c.Param("rid") + done, err := service.GodModeService.DeleteReport(requestReportId) + if err != nil { + if ipErr, ok := err.(exceptions.ImproperOperateError); ok { + result.NotAccept(ipErr.Message) + } else { + result.Error(http.StatusInternalServerError, err.Error()) + } + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功删除指定报表。") + return + } + result.Success("指定报表已经删除。") +} + +func gmDeleteSpecificMaintenance(c *gin.Context) { + result := response.NewResult(c) + requestParkId := c.Param("pid") + requestMaintenanceId := c.Param("mid") + done, err := service.GodModeService.RemoveSpecificMaintenance(requestParkId, requestMaintenanceId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功删除指定的维护费用记录。") + return + } + result.Success("指定维护费用记录已经删除。") +} + +func gmDeleteAllMaintenance(c *gin.Context) { + result := response.NewResult(c) + requestParkId := c.Param("pid") + done, err := service.GodModeService.RemoveAllMaintenance(requestParkId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功删除全部维护费用记录。") + return + } + result.Success("全部维护费用记录已经删除。") +} + +func gmDeleteAllMeters(c *gin.Context) { + result := response.NewResult(c) + requestParkId := c.Param("pid") + done, err := service.GodModeService.RemoveAllMeters(requestParkId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功删除全部终端表计档案记录。") + return + } + result.Success("全部终端表计档案记录已经删除。") +} + +func gmDeletePark(c *gin.Context) { + result := response.NewResult(c) + requestParkId := c.Param("pid") + done, err := service.GodModeService.RemovePark(requestParkId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功删除指定的园区。") + return + } + result.Success("指定的园区已经删除。") +} + +func gmDeleteUser(c *gin.Context) { + result := response.NewResult(c) + requestUserId := c.Param("uid") + done, err := service.GodModeService.DeleteUser(requestUserId) + if err != nil { + result.Error(http.StatusInternalServerError, err.Error()) + return + } + if !done { + result.Error(http.StatusInternalServerError, "未能成功删除指定的用户。") + return + } + result.Success("指定的用户及其关联信息已经删除。") +} diff --git a/router/router.go b/router/router.go index 87c8212..765a45f 100644 --- a/router/router.go +++ b/router/router.go @@ -25,6 +25,7 @@ func Router() *gin.Engine { controller.InitializeEndUserController(router) controller.InitializeWithdrawController(router) controller.InitializeStatisticsController(router) + controller.InitializeGodModeController(router) return router }