From 2f9a4b5f6c87176ae8a79c74dea51ece35c2fb80 Mon Sep 17 00:00:00 2001
From: coward <coward@mrx.ltd>
Date: Tue, 4 Mar 2025 08:48:06 +0800
Subject: [PATCH] =?UTF-8?q?:arrow=5Fup:=E6=B7=BB=E5=8A=A0dontenv=E5=8C=85?=
 =?UTF-8?q?=EF=BC=8C=E5=BD=93=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F=E4=B8=8D?=
 =?UTF-8?q?=E5=AD=98=E5=9C=A8=E6=97=B6=E8=AF=BB=E5=8F=96env=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 go.mod             |  1 +
 go.sum             |  2 ++
 initialize/init.go | 10 ++++++++++
 3 files changed, 13 insertions(+)

diff --git a/go.mod b/go.mod
index 8322e27..1a9c063 100644
--- a/go.mod
+++ b/go.mod
@@ -78,6 +78,7 @@ require (
 	github.com/jackc/puddle/v2 v2.2.1 // indirect
 	github.com/jinzhu/inflection v1.0.0 // indirect
 	github.com/jinzhu/now v1.1.5 // indirect
+	github.com/joho/godotenv v1.5.1 // indirect
 	github.com/jonboulle/clockwork v0.4.0 // indirect
 	github.com/josharian/native v1.1.0 // indirect
 	github.com/jpillora/backoff v1.0.0 // indirect
diff --git a/go.sum b/go.sum
index a625889..c0c05f6 100644
--- a/go.sum
+++ b/go.sum
@@ -776,6 +776,8 @@ github.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHW
 github.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U=
 github.com/joefitzgerald/rainbow-reporter v0.1.0/go.mod h1:481CNgqmVHQZzdIbN52CupLJyoVwB10FQ/IQlF1pdL8=
 github.com/joho/godotenv v1.3.0/go.mod h1:7hK45KPybAkOC6peb+G5yklZfMxEjkZhHbwpqxOKXbg=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 github.com/jonboulle/clockwork v0.1.0/go.mod h1:Ii8DK3G1RaLaWxj9trq07+26W01tbo22gdxWY5EU2bo=
 github.com/jonboulle/clockwork v0.2.2/go.mod h1:Pkfl5aHPm1nk2H9h0bjmnJD/BcgbGXUBGnn1kMkgxc8=
 github.com/jonboulle/clockwork v0.4.0 h1:p4Cf1aMWXnXAUh8lVfewRBx1zaTSYKrKMF2g3ST4RZ4=
diff --git a/initialize/init.go b/initialize/init.go
index 3dddb64..625480b 100644
--- a/initialize/init.go
+++ b/initialize/init.go
@@ -9,6 +9,7 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/glebarez/sqlite"
 	"github.com/go-resty/resty/v2"
+	"github.com/joho/godotenv"
 	"github.com/redis/go-redis/v9"
 	"github.com/spf13/viper"
 	"golang.zx2c4.com/wireguard/wgctrl"
@@ -27,6 +28,7 @@ import (
 func Init() {
 	initLogger()     // 初始化日志
 	initConfig()     // 读取配置文件
+	initEnv()        // 加载环境变量文件
 	initWireguard()  // 初始化wireguard客户端
 	initDatabase()   // 初始化数据库
 	initRedis()      // 初始化redis
@@ -153,3 +155,11 @@ func initLogger() {
 		FileEnable: true,
 	})
 }
+
+// initEnv
+// @description: 初始化环境变量
+func initEnv() {
+	if err := godotenv.Load("./.env"); err != nil {
+		log.Errorf("加载.env文件失败: %v", err.Error())
+	}
+}