34 lines
735 B
Go
34 lines
735 B
Go
package initialize
|
|
|
|
import (
|
|
"kra/internal/plugin/announcement/router"
|
|
"kra/internal/server/middleware"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// RouterConfig 路由配置
|
|
type RouterConfig struct {
|
|
RouterPrefix string
|
|
}
|
|
|
|
func Router(engine *gin.Engine, db *gorm.DB) {
|
|
RouterWithConfig(engine, db, nil)
|
|
}
|
|
|
|
func RouterWithConfig(engine *gin.Engine, db *gorm.DB, cfg *RouterConfig) {
|
|
// 获取配置中的路由前缀
|
|
routerPrefix := ""
|
|
if cfg != nil {
|
|
routerPrefix = cfg.RouterPrefix
|
|
}
|
|
|
|
public := engine.Group(routerPrefix)
|
|
private := engine.Group(routerPrefix)
|
|
private.Use(middleware.JWTAuth()).Use(middleware.CasbinHandler())
|
|
|
|
routerInstance := router.NewRouter(db)
|
|
routerInstance.Info.Init(public, private)
|
|
}
|