🚀 feat(config): update server port handling and entrypoint to accept dynamic port configuration

🛠️ fix(main): change server address to use dynamic port from command line argument

🔧 chore(config): remove deprecated server_port configuration from struct and set default values

Author: w7years

Diff for: cmd/proxy/main.go

@@ -23,7 +23,9 @@ import (
 )
 
 func main() {
-	configPath := flag.String("config", "config.yaml", "配置文件路径")
+	configPath := flag.String("config", "config.yaml", "config file path")
+	serverPort := flag.Int("port", 3002, "server port")
+
 	flag.Parse()
 
 	// Load config
@@ -54,7 +56,7 @@ func main() {
 	router := gin.New()
 	// Add health check route
 	router.GET("/generate_204", func(c *gin.Context) {
-		logger.Logger.Debugln("generate_204 success.")
+		logger.Logger.Infoln("generate_204 success.")
 		c.Status(http.StatusNoContent)
 	})
 
@@ -84,7 +86,7 @@ func main() {
 
 	// Set up server
 	srv := &http.Server{
-		Addr:         ":" + cfg.ServerPort,
+		Addr:         fmt.Sprintf(":%d", *serverPort),
 		Handler:      router,
 		ReadTimeout:  90 * time.Second,
 		WriteTimeout: 90 * time.Second,
@@ -103,7 +105,7 @@ func main() {
 			logger.Logger.Fatalf("Failed to start server: %v", err)
 			errChan <- err
 		} else {
-			logger.Logger.Infof("Server started successfully: %s", cfg.ServerPort)
+			logger.Logger.Infof("Server started successfully: %d", *serverPort)
 		}
 	}()
 

Diff for: config.yaml

@@ -1,4 +1,3 @@
-server_port: "3002"
 rate_limit: "100-M"
 fixed_request_ip: ""
 max_request_body_size_mb: 100

Diff for: entrypoint.sh

@@ -1,2 +1,2 @@
 #!/bin/sh
-exec ./proxy -config "$CONFIG_PATH"
+exec ./proxy -config "${CONFIG_PATH}" -port "${PORT}"

Diff for: internal/config/config.go

@@ -8,7 +8,6 @@ import (
 
 // Config config struct
 type Config struct {
-	ServerPort           string            `mapstructure:"server_port"`
 	RateLimit            string            `mapstructure:"rate_limit"`
 	MaxRequestBodySizeMB int               `mapstructure:"max_request_body_size_mb"`
 	FixedRequestIP       string            `mapstructure:"fixed_request_ip"`
@@ -24,7 +23,6 @@ func LoadConfig(configPath string) (*Config, error) {
 	viper.SetConfigFile(configPath)
 
 	// Set default values
-	viper.SetDefault("server_port", "3001")
 	viper.SetDefault("rate_limit", "100-M")
 	viper.SetDefault("max_request_body_size_mb", "100")
 	viper.SetDefault("log_dir", "logs")

Diff for: internal/middleware/middleware.go

@@ -16,8 +16,8 @@ func APIKeyAuthMiddleware() gin.HandlerFunc {
 			apiKey = c.GetHeader("x-api-key")
 		}
 		if apiKey == "" {
-			logger.Logger.Warn("未授权的访问请求")
-			c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "未授权"})
+			logger.Logger.Error("Unauthorized access request")
+			c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "Unauthorized"})
 			return
 		}
 		c.Next()
@@ -48,8 +48,8 @@ func LimitRequestBody(maxBytes int64) gin.HandlerFunc {
 	return func(c *gin.Context) {
 		c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, maxBytes)
 		if err := c.Request.ParseForm(); err != nil {
-			logger.Logger.Warn("请求体过大")
-			c.AbortWithStatusJSON(http.StatusRequestEntityTooLarge, gin.H{"error": "请求体过大"})
+			logger.Logger.Error("Request body too large")
+			c.AbortWithStatusJSON(http.StatusRequestEntityTooLarge, gin.H{"error": "Request body too large"})
 			return
 		}
 		c.Next()