diff --git a/controller/god_mode.go b/controller/god_mode.go index 62b1c45..e431348 100644 --- a/controller/god_mode.go +++ b/controller/god_mode.go @@ -7,192 +7,170 @@ import ( "electricity_bill_calc/service" "net/http" - "github.com/gin-gonic/gin" + "github.com/gofiber/fiber/v2" ) -func InitializeGodModeController(router *gin.Engine) { +func InitializeGodModeController(router *fiber.App) { 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.POST("/report/:rid/meters", security.SingularityAuthorize, gmResynchronizeReportEndUserRecord) - 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) + 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.Post("/report/:rid/meters", security.SingularityAuthorize, gmResynchronizeReportEndUserRecord) + 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) { +func gmResetReportSummary(c *fiber.Ctx) error { result := response.NewResult(c) - requestReportId := c.Param("rid") + requestReportId := c.Params("rid") done, err := service.GodModeService.ClearReportSummary(requestReportId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功重置指定报表的园区总览部分。") - return + return result.Error(http.StatusInternalServerError, "未能成功重置指定报表的园区总览部分。") } - result.Success("指定报表的园区总览已经重置。") + return result.Success("指定报表的园区总览已经重置。") } -func gmResetReportMaintenance(c *gin.Context) { +func gmResetReportMaintenance(c *fiber.Ctx) error { result := response.NewResult(c) - requestReportId := c.Param("rid") + requestReportId := c.Params("rid") done, err := service.GodModeService.ClearReportMaintenances(requestReportId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功重置指定报表的配电维护费部分。") - return + return result.Error(http.StatusInternalServerError, "未能成功重置指定报表的配电维护费部分。") } - result.Success("指定报表的配电维护费已经重置。") + return result.Success("指定报表的配电维护费已经重置。") } -func gmResynchronizeReportEndUserRecord(c *gin.Context) { +func gmResynchronizeReportEndUserRecord(c *fiber.Ctx) error { result := response.NewResult(c) - requestReportId := c.Param("rid") + requestReportId := c.Params("rid") done, err := service.GodModeService.ResynchronizeEndUser(requestReportId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功重置指定报表的抄表记录基本档案。") - return + return result.Error(http.StatusInternalServerError, "未能成功重置指定报表的抄表记录基本档案。") } - result.Success("指定报表的抄表记录基本档案已经重新同步。") + return result.Success("指定报表的抄表记录基本档案已经重新同步。") } -func gmResetReportEndUserRecord(c *gin.Context) { +func gmResetReportEndUserRecord(c *fiber.Ctx) error { result := response.NewResult(c) - requestReportId := c.Param("rid") + requestReportId := c.Params("rid") done, err := service.GodModeService.ResetEndUserRegisterRecords(requestReportId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功重置指定报表的抄表记录部分。") - return + return result.Error(http.StatusInternalServerError, "未能成功重置指定报表的抄表记录部分。") } - result.Success("指定报表的抄表记录已经重置。") + return result.Success("指定报表的抄表记录已经重置。") } -func gmResetReport(c *gin.Context) { +func gmResetReport(c *fiber.Ctx) error { result := response.NewResult(c) - requestReportId := c.Param("rid") + requestReportId := c.Params("rid") done, err := service.GodModeService.ResetReport(requestReportId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功重置指定报表。") - return + return result.Error(http.StatusInternalServerError, "未能成功重置指定报表。") } - result.Success("指定报表已经重置。") + return result.Success("指定报表已经重置。") } -func gmDeleteReport(c *gin.Context) { +func gmDeleteReport(c *fiber.Ctx) error { result := response.NewResult(c) - requestReportId := c.Param("rid") + requestReportId := c.Params("rid") done, err := service.GodModeService.DeleteReport(requestReportId) if err != nil { if ipErr, ok := err.(exceptions.ImproperOperateError); ok { - result.NotAccept(ipErr.Message) + return result.NotAccept(ipErr.Message) } else { - result.Error(http.StatusInternalServerError, err.Error()) + return result.Error(http.StatusInternalServerError, err.Error()) } - return } if !done { - result.Error(http.StatusInternalServerError, "未能成功删除指定报表。") - return + return result.Error(http.StatusInternalServerError, "未能成功删除指定报表。") } - result.Success("指定报表已经删除。") + return result.Success("指定报表已经删除。") } -func gmDeleteSpecificMaintenance(c *gin.Context) { +func gmDeleteSpecificMaintenance(c *fiber.Ctx) error { result := response.NewResult(c) - requestParkId := c.Param("pid") - requestMaintenanceId := c.Param("mid") + requestParkId := c.Params("pid") + requestMaintenanceId := c.Params("mid") done, err := service.GodModeService.RemoveSpecificMaintenance(requestParkId, requestMaintenanceId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功删除指定的维护费用记录。") - return + return result.Error(http.StatusInternalServerError, "未能成功删除指定的维护费用记录。") } - result.Success("指定维护费用记录已经删除。") + return result.Success("指定维护费用记录已经删除。") } -func gmDeleteAllMaintenance(c *gin.Context) { +func gmDeleteAllMaintenance(c *fiber.Ctx) error { result := response.NewResult(c) - requestParkId := c.Param("pid") + requestParkId := c.Params("pid") done, err := service.GodModeService.RemoveAllMaintenance(requestParkId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功删除全部维护费用记录。") - return + return result.Error(http.StatusInternalServerError, "未能成功删除全部维护费用记录。") } - result.Success("全部维护费用记录已经删除。") + return result.Success("全部维护费用记录已经删除。") } -func gmDeleteAllMeters(c *gin.Context) { +func gmDeleteAllMeters(c *fiber.Ctx) error { result := response.NewResult(c) - requestParkId := c.Param("pid") + requestParkId := c.Params("pid") done, err := service.GodModeService.RemoveAllMeters(requestParkId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功删除全部终端表计档案记录。") - return + return result.Error(http.StatusInternalServerError, "未能成功删除全部终端表计档案记录。") } - result.Success("全部终端表计档案记录已经删除。") + return result.Success("全部终端表计档案记录已经删除。") } -func gmDeletePark(c *gin.Context) { +func gmDeletePark(c *fiber.Ctx) error { result := response.NewResult(c) - requestParkId := c.Param("pid") + requestParkId := c.Params("pid") done, err := service.GodModeService.RemovePark(requestParkId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功删除指定的园区。") - return + return result.Error(http.StatusInternalServerError, "未能成功删除指定的园区。") } - result.Success("指定的园区已经删除。") + return result.Success("指定的园区已经删除。") } -func gmDeleteUser(c *gin.Context) { +func gmDeleteUser(c *fiber.Ctx) error { result := response.NewResult(c) - requestUserId := c.Param("uid") + requestUserId := c.Params("uid") done, err := service.GodModeService.DeleteUser(requestUserId) if err != nil { - result.Error(http.StatusInternalServerError, err.Error()) - return + return result.Error(http.StatusInternalServerError, err.Error()) } if !done { - result.Error(http.StatusInternalServerError, "未能成功删除指定的用户。") - return + return result.Error(http.StatusInternalServerError, "未能成功删除指定的用户。") } - result.Success("指定的用户及其关联信息已经删除。") + return result.Success("指定的用户及其关联信息已经删除。") }