Roc 1 rok temu
rodzic
commit
1decfbe7e4
2 zmienionych plików z 26 dodań i 1 usunięć
  1. 21 0
      controllers/error.go
  2. 5 1
      main.go

+ 21 - 0
controllers/error.go

@@ -0,0 +1,21 @@
+package controllers
+
+import "eta/eta_chart_lib/models"
+
+// ErrorController
+// @Description: 该控制器处理页面错误请求
+type ErrorController struct {
+	BaseCommonController
+}
+
+func (c *ErrorController) Error404() {
+	c.Data["content"] = "很抱歉您访问的地址或者方法不存在"
+	//c.TplName = "error/404.html"
+
+	br := new(models.BaseResponse).Init()
+
+	br.Msg = "您访问的资源不存在"
+	br.Ret = 404
+	c.Data["json"] = br
+	c.ServeJSON()
+}

+ 5 - 1
main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"eta/eta_chart_lib/controllers"
 	_ "eta/eta_chart_lib/routers"
 	"eta/eta_chart_lib/services/alarm_msg"
 	"eta/eta_chart_lib/utils"
@@ -18,8 +19,11 @@ func main() {
 		beego.BConfig.WebConfig.DirectoryIndex = true
 		beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
 	}
-	beego.BConfig.RecoverFunc = Recover
 
+	// 异常处理
+	beego.ErrorController(&controllers.ErrorController{})
+
+	beego.BConfig.RecoverFunc = Recover
 	beego.Run()
 }