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()) + } +}