go fiber: 抛出自定义异常

一，代码：

1,自定义错误类:

package configimport ("fmt"
)
//定义错误代码和错误信息
type MyError struct {Code    intMsg     string
}
//需要定义通用的Error()方法
func (e MyError) Error() string {return fmt.Sprintf("Code: %d,Msg: %s",e.Code, e.Msg)
}

2,代码中抛出异常

package businessimport ("industry/config")//得到多个用户，按分页返回
func GetUserList(page int ,pageSize int) (string,error) {err := config.MyError{Code: 100, Msg: "遇到错误抛出自定义异常1234"}panic(&err)return "1,2,3",nil
}

3, middleware中捕获异常，按类型进行处理

package middlewareimport ("fmt""runtime/debug""time""github.com/gofiber/fiber/v2""industry/config"
)func NewRecover(c *fiber.Ctx) (err error) { //nolint:nonamedreturns // Uses recover() to overwrite the error// Catch panicsdefer func() {if r := recover(); r != nil {//如果是自定义异常时，则不需写日志if err1, ok1 := r.(*config.MyError); ok1 {fmt.Println("----------找到了自定义错误:")fmt.Println(err1)fmt.Println("错误代码:"+fmt.Sprintf("%d", err1.Code))fmt.Println("错误信息:"+err1.Msg)//接收到用户自定义错误，不再写到日志中err = err1} else {timeStr:=time.Now().Format("2006-01-02 15:04:05")//得到时间,ip地址,请求的地址，参数，user-agent,fullURL := c.OriginalURL()clientIP := c.IP()method := c.Method() // 获取请求方法userAgent := c.Get("User-Agent")// 获取所有查询参数queryParams := fmt.Sprintf("%v",c.Queries())fmt.Println("Query Params:", queryParams)//写到错误日志中filename:="/data/logs/gologs/exception-" + time.Now().Format("2006-01-02") + ".log"content:="["+timeStr+"] "+r.(error).Error()+"\n"+clientIP+" "+method+" "+fullURL+" "+userAgent+"\n"+queryParams+"\n"+string(debug.Stack())+"\n"config.GlobalWriteFile(filename,content)var ok boolif err, ok = r.(error); !ok {// Set error that will call the global error handlererr = fmt.Errorf("%v", r)}}}}()// Return err if exist, else move to next handlerreturn c.Next()
}

4, 返回错误信息时也按类型判断后处理

routes.go

package routesimport ("github.com/gofiber/fiber/v2""industry/controller""industry/middleware""industry/config""fmt"
)func SetupRoutes() *fiber.App {app := fiber.New(fiber.Config{ErrorHandler: func(c *fiber.Ctx, err error) error {// 发送自定义错误页面// 如果是自定义错误，不再统一按500返回if err1,ok1:=err.(*config.MyError);ok1 {fmt.Println(err1)fmt.Println("route:错误是自定义异常")return c.SendString("内部错误代码:"+fmt.Sprintf("%d", err1.Code)+",错误信息:"+err1.Msg)} else {fmt.Println("route:错误不是自定义异常")return c.SendString("内部错误:500,"+err.Error())}},})app.Use(middleware.NewRecover)app.Use(middleware.ApiSign)//用户模块userController := controller.NewUserController()user := app.Group("/user")user.Get("/info", userController.GetUser)user.Post("/", userController.CreateUser)user.Get("/list", userController.ListUser)//文章模块articleController := controller.NewArticleController()article := app.Group("/article")article.Get("/info", articleController.GetArticle)article.Post("/add", articleController.CreateArticle)article.Get("/list", articleController.ListArticle)return app
}

二，测试效果: