增加定时任务 处理异常

router/router.go

@@ -2,18 +2,36 @@ package router
 
 import (
 	v1 "energy-management-system/api/v1"
+	"energy-management-system/global"
 	"energy-management-system/middleware"
 	"energy-management-system/router/routes"
+	"energy-management-system/utils/code"
 	"energy-management-system/utils/exception"
+	"energy-management-system/utils/recovery"
 	"github.com/gin-gonic/gin"
+	"net/http"
 )
 
 func InitRouter() *gin.Engine {
-	gin.SetMode(gin.DebugMode)
+
+	if global.AppConf.IsDebug {
+		gin.SetMode(gin.DebugMode)
+	} else {
+		gin.SetMode("release")
+	}
+
 	r := gin.Default()
-	r.NoRoute(exception.NotFoundR)
-	r.NoMethod(exception.NotFoundM)
-	r.Use(middleware.Cors(), middleware.Recovery())
+	//r.NoRoute(exception.NotFoundR)
+	//r.NoMethod(exception.NotFoundM)
+
+	r.NoRoute(func(c *gin.Context) {
+		c.JSON(http.StatusForbidden, exception.E{Code: code.NOT_FOUND_ROUTE, Msg: "Not Found Route", Err: nil})
+	})
+	r.NoMethod(func(c *gin.Context) {
+		c.JSON(http.StatusForbidden, exception.E{Code: code.NOT_FOUND_METH, Msg: "Not Found Method"})
+	})
+
+	r.Use(middleware.Cors(), middleware.Log, recovery.GinRecovery)
 	api := r.Group("api")
 	{
 		controllersV1 := new(v1.Controller)