通过反射注册路由

app/app.go

@@ -12,8 +12,6 @@ import (
 	"gofiber.study.skcks.cn/global"
 )
 
-var App *fiber.App
-
 func Run() {
 	err := viper.Unmarshal(global.Config)
 	if err != nil {
@@ -28,20 +26,20 @@ func Run() {
 	reloadDataSources(global.Config)
 	reloadJwt(global.Config)
 
-	App = reloadApp(global.Config)
+	global.App = reloadApp(global.Config)
 
-	if err := App.Listen(global.Config.Server.Addr); err != nil {
+	if err := global.App.Listen(global.Config.Server.Addr); err != nil {
 		logger.Log.Fatalf("[x] [Fiber] 致命错误: %s", err)
 	}
 
 }
 
 func reloadApp(c *config.BasicConfig) *fiber.App {
-	if App != nil {
-		_ = App.Shutdown()
+	if global.App != nil {
+		_ = global.App.Shutdown()
 	}
 
-	app := fiber.New(fiber.Config{
+	global.App = fiber.New(fiber.Config{
 		JSONEncoder:   json.Marshal,
 		JSONDecoder:   json.Unmarshal,
 		Prefork:       c.Server.PreFork,
@@ -65,20 +63,19 @@ func reloadApp(c *config.BasicConfig) *fiber.App {
 	})
 
 	if c.Server.EnableRoutesMsg {
-		routes(app)
+		routes(global.App)
 	}
 
-	controller.SwaggerHandler(app)
-	controller.RegisterController(app)
-	controller.HelloWorld(app)
-	controller.ErrorHandler(app)
+	controller.SwaggerHandler(global.App)
+	controller.RegisterController(global.App)
+	controller.ErrorHandler(global.App)
 
-	app.Hooks().OnListen(func() error {
+	global.App.Hooks().OnListen(func() error {
 		utils.MainAppExec(func() {
 			logger.Log.Infoln("[√] [服务启动完成]")
 		})
 		return nil
 	})
 
-	return app
+	return global.App
 }

controller/auth/auth.go

@@ -0,0 +1 @@
+package auth

controller/controller.go

@@ -4,34 +4,31 @@ import (
 	"github.com/gofiber/fiber/v2"
 	"github.com/gofiber/swagger"
 	"gofiber.study.skcks.cn/common/errorx"
-	"gofiber.study.skcks.cn/common/logger"
-	"gofiber.study.skcks.cn/common/response"
 	"gofiber.study.skcks.cn/controller/casbin"
 	"gofiber.study.skcks.cn/controller/test"
 	"gofiber.study.skcks.cn/controller/user"
+	"reflect"
 )
 
 func RegisterController(app *fiber.App) {
-	user.RegisterController(app)
+	controllers := make([]interface{}, 0)
+	controllers = append(controllers, user.NewUserController(app))
+
+	for _, controller := range controllers {
+		reflectRegisterRoute(controller)
+	}
+
 	casbin.RegisterController(app)
 	test.RegisterController(app)
 }
 
-// HelloWorld
-//
-// @Summary      HelloWorld
-// @Description  简单的 HelloWorld 示例
-// @Tags         HelloWorld
-// @Accept       json
-// @Produce      json
-// @Success      200  {object}  response.Response{data=string}
-// @Failure	     default {object}  errorx.CodeErrorResponse
-// @Router       / [get]
-func HelloWorld(app *fiber.App) {
-	app.Get("/", func(c *fiber.Ctx) error {
-		logger.Log.Infof("\n%s", c.Request().String())
-		return c.JSON(response.NewResponse("Hello, World 👋!"))
-	})
+func reflectRegisterRoute(controller interface{}) {
+	t := reflect.ValueOf(controller)
+	for i := 0; i < t.NumMethod(); i++ {
+		method := t.Method(i)
+		args := make([]reflect.Value, 0)
+		method.Call(args)
+	}
 }
 
 // ErrorHandler

controller/types/type.go

@@ -0,0 +1,15 @@
+package types
+
+import (
+	"github.com/gofiber/fiber/v2"
+)
+
+type Controller struct {
+	Group fiber.Router
+}
+
+func NewController(app *fiber.App, group string) *Controller {
+	return &Controller{
+		Group: app.Group(group),
+	}
+}

controller/user/user.go

@@ -4,12 +4,46 @@ import (
 	"github.com/gofiber/fiber/v2"
 	"gofiber.study.skcks.cn/common/errorx"
 	"gofiber.study.skcks.cn/common/response"
+	"gofiber.study.skcks.cn/controller/types"
 	"gofiber.study.skcks.cn/services/user"
 )
 
+type Controller struct {
+	*types.Controller
+}
+
+func NewUserController(app *fiber.App) *Controller {
+	controller := Controller{
+		Controller: types.NewController(app, "/user"),
+	}
+
+	//t := reflect.TypeOf(&controller)
+	//for i := 0; i < t.NumMethod(); i++ {
+	//	method := t.Method(i)
+	//	args := make([]reflect.Value, 0)
+	//	args = append(args, reflect.ValueOf(&controller))
+	//	method.Func.Call(args)
+	//}
+	return &controller
+}
+
+func (c *Controller) GetByAccount() {
+	c.Group.Get("/account", func(ctx *fiber.Ctx) error {
+		account := ctx.Query("account")
+
+		u, err := user.GetUserByAccount(account)
+		if err != nil {
+			return ctx.JSON(errorx.NewDefaultError(err.Error()))
+		}
+
+		return ctx.JSON(response.NewResponse(u))
+	})
+}
+
 func RegisterController(app *fiber.App) {
-	group := app.Group("/user")
-	group.Add(fiber.MethodGet, "/account", getByAccount)
+	NewUserController(app)
+	//group := app.Group("/user")
+	//group.Add(fiber.MethodGet, "/account", getByAccount)
 }
 
 // getByAccount 根据 账号 获取用户信息

global/app.go

@@ -0,0 +1,5 @@
+package global
+
+import "github.com/gofiber/fiber/v2"
+
+var App *fiber.App