optimize dockerfile and config generate of docker deploy

deploy/dockerfile/modelrt.Dockerfile

@@ -9,7 +9,11 @@ RUN CGO_ENABLED=0 go build -ldflags="-s -w" -o modelrt main.go
 
 FROM alpine:latest
 WORKDIR /app
+ARG USER_ID=1000
+RUN adduser -D -u ${USER_ID} modelrt
 COPY --from=builder /app/modelrt ./modelrt
 COPY configs/config.example.yaml ./configs/config.example.yaml
+RUN chown -R modelrt:modelrt /app
 RUN chmod +x /app/modelrt
-CMD ["/app/modelrt"]
+USER modelrt
+CMD ["/app/modelrt", "-modelRT_config_dir=/app/configs"]

main.go

@@ -12,7 +12,6 @@ import (
 	"os/signal"
 	"path/filepath"
 	"syscall"
-	"time"
 
 	"modelRT/alert"
 	"modelRT/config"
@@ -22,7 +21,6 @@ import (
 	_ "modelRT/docs"
 	"modelRT/handler"
 	"modelRT/logger"
-	"modelRT/middleware"
 	"modelRT/model"
 	"modelRT/pool"
 	realtimedata "modelRT/real-time-data"
@@ -36,12 +34,6 @@ import (
 	"gorm.io/gorm"
 )
 
-var limiter *middleware.Limiter
-
-func init() {
-	limiter = middleware.NewLimiter(10, 1*time.Minute) // 设置限流器，允许每分钟最多请求10次
-}
-
 var (
 	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")
@@ -80,14 +72,16 @@ func main() {
 		log.Println("configuration file not found,checking for example file")
 
 		exampleConfigPath := filepath.Join(*modelRTConfigDir, *modelRTConfigName+".example."+*modelRTConfigType)
+		configDir := filepath.Dir(configPath)
+		if err := os.MkdirAll(configDir, 0o755); err != nil {
+			panic(fmt.Errorf("failed to create config directory %s:%w", configDir, err))
+		}
 		if _, err := os.Stat(exampleConfigPath); err == nil {
 			if err := util.CopyFile(exampleConfigPath, configPath); err != nil {
-				panicErr := fmt.Errorf("failed to copy example config file:%w", err)
-				panic(panicErr)
+				panic(fmt.Errorf("failed to copy example config file:%w", err))
 			}
 		} else {
-			panicErr := errors.New("no config file and no config example file found")
-			panic(panicErr)
+			panic(errors.New("no config file and no config example file found"))
 		}
 	}
 
@@ -191,7 +185,7 @@ func main() {
 	engine.GET("/rt/datas/query", handler.QueryRealTimeDataHandler)
 
 	// dashborad api
-	dashboard := engine.Group("/dashboard", limiter.Middleware)
+	dashboard := engine.Group("/dashboard")
 	{
 		dashboard.GET("/load", nil)
 		dashboard.GET("/query", nil)

router/measurement.go

@@ -3,15 +3,14 @@ package router
 
 import (
 	"modelRT/handler"
-	"modelRT/middleware"
 
 	"github.com/gin-gonic/gin"
 )
 
 // 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.Use(middleware.SetTokenMiddleware(clientToken))
+	g.Use(middlewares...)
 	g.GET("load", handler.MeasurementGetHandler)
 	g.GET("recommend", handler.MeasurementRecommendHandler)
 }

router/router.go

@@ -9,20 +9,21 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
-var limiter *middleware.Limiter
+var measurementLimiter *middleware.Limiter
 
 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
 func RegisterRoutes(engine *gin.Engine, clientToken string) {
 	// use global middlewares
-	engine.Use(middleware.StartTrace(), limiter.Middleware)
+	engine.Use(middleware.StartTrace())
 	routeGroup := engine.Group("")
 	registerDiagramRoutes(routeGroup)
 	registerAttrRoutes(routeGroup)
-	registerMeasurementRoutes(routeGroup, clientToken)
+	registerMeasurementRoutes(routeGroup, middleware.SetTokenMiddleware(clientToken), measurementLimiter.Middleware)
 	registerDataRoutes(routeGroup)
 	registerMonitorRoutes(routeGroup)
 }