CL-Softwares
/
modelRT
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

optimzie rate limit of measurement group api

This commit is contained in:
douxu 2025-12-17 14:12:35 +08:00
parent 252699cb77
commit 4f5d998659
3 changed files with 8 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
main.go
View File

@ -12,7 +12,6 @@ import (
"os/signal" "os/signal"
"path/filepath" "path/filepath"
"syscall" "syscall"
"time"
"modelRT/alert" "modelRT/alert"
"modelRT/config" "modelRT/config"
@ -22,7 +21,6 @@ import (
_ "modelRT/docs" _ "modelRT/docs"
"modelRT/handler" "modelRT/handler"
"modelRT/logger" "modelRT/logger"
"modelRT/middleware"
"modelRT/model" "modelRT/model"
"modelRT/pool" "modelRT/pool"
realtimedata "modelRT/real-time-data" realtimedata "modelRT/real-time-data"
@ -36,12 +34,6 @@ import (
"gorm.io/gorm" "gorm.io/gorm"
) )
var limiter *middleware.Limiter
func init() {
limiter = middleware.NewLimiter(10, 1*time.Minute) // 设置限流器，允许每分钟最多请求10次
}
var ( var (
modelRTConfigDir = flag.String("modelRT_config_dir", "./configs", "config file dir of model runtime service") modelRTConfigDir = flag.String("modelRT_config_dir", "./configs", "config file dir of model runtime service")
modelRTConfigName = flag.String("modelRT_config_name", "config", "config file name of model runtime service") modelRTConfigName = flag.String("modelRT_config_name", "config", "config file name of model runtime service")
@ -191,7 +183,7 @@ func main() {
engine.GET("/rt/datas/query", handler.QueryRealTimeDataHandler) engine.GET("/rt/datas/query", handler.QueryRealTimeDataHandler)
// dashborad api // dashborad api
dashboard := engine.Group("/dashboard", limiter.Middleware) dashboard := engine.Group("/dashboard")
{ {
dashboard.GET("/load", nil) dashboard.GET("/load", nil)
dashboard.GET("/query", nil) dashboard.GET("/query", nil)

5
router/measurement.go
View File

@ -3,15 +3,14 @@ package router
import ( import (
"modelRT/handler" "modelRT/handler"
"modelRT/middleware"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
// registerMeasurementRoutes define func of register measurement routes // registerMeasurementRoutes define func of register measurement routes
func registerMeasurementRoutes(rg *gin.RouterGroup, clientToken string) { func registerMeasurementRoutes(rg *gin.RouterGroup, middlewares ...gin.HandlerFunc) {
g := rg.Group("/measurement/") g := rg.Group("/measurement/")
g.Use(middleware.SetTokenMiddleware(clientToken)) g.Use(middlewares...)
g.GET("load", handler.MeasurementGetHandler) g.GET("load", handler.MeasurementGetHandler)
g.GET("recommend", handler.MeasurementRecommendHandler) g.GET("recommend", handler.MeasurementRecommendHandler)
} }

9
router/router.go
View File

@ -9,20 +9,21 @@ import (
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
var limiter *middleware.Limiter var measurementLimiter *middleware.Limiter
func init() { func init() {
limiter = middleware.NewLimiter(10, 1*time.Minute) // 设置限流器，允许每分钟最多请求10次 // set a rate limiter to allow a maximum of 2 requests per second for measurement group api
measurementLimiter = middleware.NewLimiter(2, 1*time.Second)
} }
// RegisterRoutes define func of register all routes // RegisterRoutes define func of register all routes
func RegisterRoutes(engine *gin.Engine, clientToken string) { func RegisterRoutes(engine *gin.Engine, clientToken string) {
// use global middlewares // use global middlewares
engine.Use(middleware.StartTrace(), limiter.Middleware) engine.Use(middleware.StartTrace())
routeGroup := engine.Group("") routeGroup := engine.Group("")
registerDiagramRoutes(routeGroup) registerDiagramRoutes(routeGroup)
registerAttrRoutes(routeGroup) registerAttrRoutes(routeGroup)
registerMeasurementRoutes(routeGroup, clientToken) registerMeasurementRoutes(routeGroup, middleware.SetTokenMiddleware(clientToken), measurementLimiter.Middleware)
registerDataRoutes(routeGroup) registerDataRoutes(routeGroup)
registerMonitorRoutes(routeGroup) registerMonitorRoutes(routeGroup)
} }