网关实现自定义接口安全检查与边缘节点连接权限验证

server/env/env.go

@@ -0,0 +1,127 @@
+package env
+
+import (
+	"fmt"
+	"log/slog"
+	"os"
+	"strconv"
+
+	"github.com/joho/godotenv"
+)
+
+var (
+	AppCtrlPort uint16 = 18080
+	AppDataPort uint16 = 18081
+	AppWebPort  uint16 = 8848
+	AppLogMode         = "dev"
+
+	ClientId     string
+	ClientSecret string
+
+	RedisHost = "localhost"
+	RedisPort = "6379"
+	RedisDb   = 0
+	RedisPass = ""
+
+	EndpointOnline   string
+	EndpointOffline  string
+	EndpointAssigned string
+)
+
+func Init() {
+	var err = godotenv.Load()
+	if err != nil {
+		slog.Debug("没有本地环境变量文件")
+	}
+	var value string
+
+	value = os.Getenv("APP_CTRL_PORT")
+	if value != "" {
+		appCtrlPort, err := strconv.Atoi(value)
+		if err != nil {
+			panic(fmt.Sprintf("环境变量 APP_CTRL_PORT 格式错误: %v", err))
+		}
+		AppCtrlPort = uint16(appCtrlPort)
+	}
+
+	value = os.Getenv("APP_DATA_PORT")
+	if value != "" {
+		appDataPort, err := strconv.Atoi(value)
+		if err != nil {
+			panic(fmt.Sprintf("环境变量 APP_DATA_PORT 格式错误: %v", err))
+		}
+		AppDataPort = uint16(appDataPort)
+	}
+
+	value = os.Getenv("APP_WEB_PORT")
+	if value != "" {
+		appWebPort, err := strconv.Atoi(value)
+		if err != nil {
+			panic(fmt.Sprintf("环境变量 APP_WEB_PORT 格式错误: %v", err))
+		}
+		AppWebPort = uint16(appWebPort)
+	}
+
+	value = os.Getenv("APP_LOG_MODE")
+	if value != "" {
+		AppLogMode = value
+	}
+
+	value = os.Getenv("CLIENT_ID")
+	if value != "" {
+		ClientId = value
+	} else {
+		panic("环境变量 CLIENT_ID 未设置")
+	}
+
+	value = os.Getenv("CLIENT_SECRET")
+	if value != "" {
+		ClientSecret = value
+	} else {
+		panic("环境变量 CLIENT_SECRET 未设置")
+	}
+
+	value = os.Getenv("REDIS_HOST")
+	if value != "" {
+		RedisHost = value
+	}
+
+	value = os.Getenv("REDIS_PORT")
+	if value != "" {
+		RedisPort = value
+	}
+
+	value = os.Getenv("REDIS_DB")
+	if value != "" {
+		redisDb, err := strconv.Atoi(value)
+		if err != nil {
+			panic(fmt.Sprintf("环境变量 REDIS_DB 格式错误: %v", err))
+		}
+		RedisDb = redisDb
+	}
+
+	value = os.Getenv("REDIS_PASS")
+	if value != "" {
+		RedisPass = value
+	}
+
+	value = os.Getenv("ENDPOINT_ONLINE")
+	if value != "" {
+		EndpointOnline = value
+	} else {
+		panic("环境变量 ENDPOINT_ONLINE 未设置")
+	}
+
+	value = os.Getenv("ENDPOINT_OFFLINE")
+	if value != "" {
+		EndpointOffline = value
+	} else {
+		panic("环境变量 ENDPOINT_OFFLINE 未设置")
+	}
+	value = os.Getenv("ENDPOINT_ASSIGNED")
+	if value != "" {
+		EndpointAssigned = value
+	} else {
+		panic("环境变量 ENDPOINT_ASSIGNED 未设置")
+	}
+}