mirror of
https://gitee.com/shikong-sk/gofiber-study
synced 2025-02-23 07:12:16 +08:00
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/swagger"
|
|
"gofiber.study.skcks.cn/common/errorx"
|
|
"gofiber.study.skcks.cn/controller/auth"
|
|
"gofiber.study.skcks.cn/controller/casbin"
|
|
"gofiber.study.skcks.cn/controller/test"
|
|
"gofiber.study.skcks.cn/controller/topical"
|
|
"gofiber.study.skcks.cn/controller/types"
|
|
"gofiber.study.skcks.cn/controller/user"
|
|
"reflect"
|
|
)
|
|
|
|
func RegisterController(app *fiber.App) {
|
|
controllers := []types.ControllerInterface{
|
|
auth.NewController(app),
|
|
user.NewController(app),
|
|
casbin.NewController(app),
|
|
test.NewController(app),
|
|
topical.NewController(app),
|
|
}
|
|
|
|
args := make([]reflect.Value, 0)
|
|
for _, controller := range controllers {
|
|
reflectRegisterRoute(controller, args)
|
|
}
|
|
}
|
|
|
|
func reflectRegisterRoute(controller interface{}, args []reflect.Value) {
|
|
t := reflect.ValueOf(controller)
|
|
for i := 0; i < t.NumMethod(); i++ {
|
|
method := t.Method(i)
|
|
method.Call(args)
|
|
}
|
|
}
|
|
|
|
// ErrorHandler
|
|
//
|
|
// @Summary 错误处理
|
|
// @Description 错误信息示例
|
|
// @Tags Error
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Failure default {object} errorx.CodeErrorResponse
|
|
// @Router /error [get]
|
|
// @Router /error [post]
|
|
// @Router /error [put]
|
|
// @Router /error [delete]
|
|
// @Router /error [patch]
|
|
func ErrorHandler(app *fiber.App) {
|
|
app.All("/error", func(ctx *fiber.Ctx) error {
|
|
return ctx.JSON(errorx.NewDefaultError("错误信息示例"))
|
|
})
|
|
}
|
|
|
|
func SwaggerHandler(app *fiber.App) {
|
|
app.Get("/swagger/*", swagger.New(swagger.Config{
|
|
DeepLinking: false,
|
|
// Expand ("list") or Collapse ("none") tag groups by default
|
|
DocExpansion: "list",
|
|
}))
|
|
}
|