26 lines
565 B
Go
26 lines
565 B
Go
// Package router provides router config
|
||
package router
|
||
|
||
import (
|
||
"time"
|
||
|
||
"modelRT/middleware"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
var limiter *middleware.Limiter
|
||
|
||
func init() {
|
||
limiter = middleware.NewLimiter(10, 1*time.Minute) // 设置限流器，允许每分钟最多请求10次
|
||
}
|
||
|
||
// RegisterRoutes define func of register all routes
|
||
func RegisterRoutes(engine *gin.Engine) {
|
||
// use global middlewares
|
||
engine.Use(middleware.StartTrace(), limiter.Middleware)
|
||
routeGroup := engine.Group("")
|
||
registerDiagramRoutes(routeGroup)
|
||
registerAttrRoutes(routeGroup)
|
||
}
|