platform/pkg/env/env.go

package env

import (
	"log/slog"
	"os"
	"strconv"

	"github.com/gofiber/fiber/v2/log"
	"github.com/joho/godotenv"
)

// region app

var (
	AppName = "platform"
	AppPort = "8080"
)

func loadApp() {
	_AppName := os.Getenv("APP_NAME")
	if _AppName != "" {
		AppName = _AppName
	}

	_AppPort := os.Getenv("APP_PORT")
	if _AppPort != "" {
		AppPort = _AppPort
	}
}

// endregion

// region db

var (
	DbHost     = "localhost"
	DbPort     = "5432"
	DbName     string
	DbUserName string
	DbPassword string
)

func loadDb() {
	_DbHost := os.Getenv("DB_HOST")
	if _DbHost != "" {
		DbHost = _DbHost
	}

	_DbPort := os.Getenv("DB_PORT")
	if _DbPort != "" {
		DbPort = _DbPort
	}

	_DbName := os.Getenv("DB_NAME")
	if _DbName != "" {
		DbName = _DbName
	} else {
		panic("环境变量 DB_NAME 的值不能为空")
	}

	_DbUserName := os.Getenv("DB_USERNAME")
	if _DbUserName != "" {
		DbUserName = _DbUserName
	} else {
		panic("环境变量 DB_USERNAME 的值不能为空")
	}

	_DbPassword := os.Getenv("DB_PASSWORD")
	if _DbPassword != "" {
		DbPassword = _DbPassword
	} else {
		panic("环境变量 DB_PASSWORD 的值不能为空")
	}
}

// endregion

// region redis

var (
	RedisHost = "localhost"
	RedisPort = "6379"
	RedisDb   = 0
	RedisPass = ""
)

func loadRedis() {
	_RedisHost := os.Getenv("REDIS_HOST")
	if _RedisHost != "" {
		RedisHost = _RedisHost
	}

	_RedisPort := os.Getenv("REDIS_PORT")
	if _RedisPort != "" {
		RedisPort = _RedisPort
	}

	_RedisDb := os.Getenv("REDIS_DB")
	if _RedisDb != "" {
		atoi, err := strconv.Atoi(_RedisDb)
		if err != nil {
			panic("环境变量 REDIS_DB 的值不是数字")
		}
		RedisDb = atoi
	}

	_RedisPass := os.Getenv("REDIS_PASS")
	if _RedisPass != "" {
		RedisPass = _RedisPass
	}
}

// endregion

// region log

var (
	LogLevel = slog.LevelDebug
)

func loadLog() {
	_LogLevel := os.Getenv("LOG_LEVEL")
	switch _LogLevel {
	case "debug":
		LogLevel = slog.LevelDebug
	case "info":
		LogLevel = slog.LevelInfo
	case "warn":
		LogLevel = slog.LevelWarn
	case "error":
		LogLevel = slog.LevelError
	}
}

// endregion

// region remote

var (
	RemoteAddr  = "http://103.139.212.110:9989"
	RemoteToken string
)

func loadRemote() {
	_RemoteAddr := os.Getenv("REMOTE_ADDR")
	if _RemoteAddr != "" {
		RemoteAddr = _RemoteAddr
	}

	_RemoteToken := os.Getenv("REMOTE_TOKEN")
	if _RemoteToken == "" {
		panic("环境变量 REMOTE_TOKEN 的值不能为空")
	}
	RemoteToken = _RemoteToken
}

// endregion

// region debug

var (
	// DebugHttpDump 是否打印请求和响应的原始数据
	DebugHttpDump = false
	// DebugExternalChange 是否实际执行非幂等外部接口的调用。
	// 例如外部数据修改接口，在内部接口调试时可以关闭，避免对外部数据产生影响
	DebugExternalChange = true
)

func loadDebug() {
	debugHttpDump := os.Getenv("DEBUG_HTTP_DUMP")
	if debugHttpDump != "" {
		value, err := strconv.ParseBool(debugHttpDump)
		if err != nil {
			panic("环境变量 DEBUG_HTTP_DUMP 的值不是布尔值")
		}
		DebugHttpDump = value
	}

	debugExternalChange := os.Getenv("DEBUG_EXTERNAL_CHANGE")
	if debugExternalChange != "" {
		value, err := strconv.ParseBool(debugExternalChange)
		if err != nil {
			panic("环境变量 DEBUG_EXTERNAL_CHANGE 的值不是布尔值")
		}
		DebugExternalChange = value
	}
}

// endregion

func Init() {
	err := godotenv.Load()
	if err != nil {
		log.Debug("❓ 没有本地环境变量")
	} else {
		log.Debug("✔ 加载本地环境变量")
	}

	loadApp()
	loadDb()
	loadRedis()
	loadLog()
	loadDebug()
	loadRemote()
}