modelRT/router/router.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, clientToken string) {
	// use global middlewares
	engine.Use(middleware.StartTrace(), limiter.Middleware)
	routeGroup := engine.Group("")
	registerDiagramRoutes(routeGroup)
	registerAttrRoutes(routeGroup)
	registerMeasurementRoutes(routeGroup, clientToken)
	registerDataRoutes(routeGroup)
	registerMonitorRoutes(routeGroup)
}