Files

36 lines
644 B
Go
Raw Permalink Normal View History

package main
import (
"os"
"github.com/joho/godotenv"
)
func init() {
// 加载 .env 文件(如果存在),环境变量优先
_ = godotenv.Load()
}
type Config struct {
GotifyURL string
GotifyToken string
Port string
MySQLDSN string // MySQL 连接串,留空则不启用数据库记录
}
func LoadConfig() Config {
return Config{
GotifyURL: getEnv("GOTIFY_URL", ""),
GotifyToken: getEnv("GOTIFY_TOKEN", ""),
Port: getEnv("PORT", ":8082"),
MySQLDSN: getEnv("MYSQL_DSN", ""),
}
}
func getEnv(key, fallback string) string {
if v := os.Getenv(key); v != "" {
return v
}
return fallback
}