23 lines
439 B
Go
23 lines
439 B
Go
|
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次
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func RegisterRoutes(engine *gin.Engine) {
|
|||
|
|
// use global middlewares
|
|||
|
|
engine.Use(middleware.StartTrace(), limiter.Middleware)
|
|||
|
|
routeGroup := engine.Group("")
|
|||
|
|
registerDiagramRoutes(routeGroup)
|
|||
|
|
}
|