mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 00:10:25 +08:00
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package initialize
|
||
|
||
import (
|
||
"fmt"
|
||
"mayfly-go/pkg/ioc"
|
||
"mayfly-go/pkg/req"
|
||
"net/http"
|
||
"reflect"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// RouterApi
|
||
// 该接口的实现类注册到ioc中，则会自动将请求配置注册到路由中
|
||
type RouterApi interface {
|
||
// ReqConfs 获取请求配置信息
|
||
ReqConfs() *req.Confs
|
||
}
|
||
|
||
type RouterConfig struct {
|
||
ContextPath string // 请求路径上下文
|
||
}
|
||
|
||
func InitRouter(router *gin.Engine, conf RouterConfig) *gin.Engine {
|
||
// 没有路由即 404返回
|
||
router.NoRoute(func(g *gin.Context) {
|
||
g.JSON(http.StatusNotFound, gin.H{"code": 404, "msg": fmt.Sprintf("not found '%s:%s'", g.Request.Method, g.Request.URL.Path)})
|
||
})
|
||
|
||
// 设置路由组
|
||
api := router.Group(conf.ContextPath + "/api")
|
||
|
||
// 获取所有实现了RouterApi接口的实例，并注册对应路由
|
||
ras := ioc.GetBeansByType[RouterApi](reflect.TypeOf((*RouterApi)(nil)).Elem())
|
||
for _, ra := range ras {
|
||
confs := ra.ReqConfs()
|
||
if group := confs.Group; group != "" {
|
||
req.BatchSetGroup(api.Group(group), confs.Confs)
|
||
} else {
|
||
req.BatchSetGroup(api, confs.Confs)
|
||
}
|
||
}
|
||
|
||
return router
|
||
}
|