From 5f0990f22bb7c89f625470082c225944d2e0559a Mon Sep 17 00:00:00 2001 From: tbphp Date: Mon, 30 Jun 2025 16:26:10 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E6=95=B0=E6=8D=AE?= =?UTF-8?q?=E5=BA=93=E8=BF=9E=E6=8E=A5=E5=AD=97=E7=AC=A6=E4=B8=B2=E7=8E=AF?= =?UTF-8?q?=E5=A2=83=E5=8F=98=E9=87=8F=E6=94=AF=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .env.example | 3 ++- internal/db/database.go | 6 ++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/.env.example b/.env.example index 3b115e8..ed19778 100644 --- a/.env.example +++ b/.env.example @@ -99,4 +99,5 @@ REQUEST_TIMEOUT=30 RESPONSE_TIMEOUT=30 # 空闲连接超时时间（秒）- 控制连接池中空闲连接的生存时间 -IDLE_CONN_TIMEOUT=120 \ No newline at end of file +IDLE_CONN_TIMEOUT=120 +DATABASE_DSN="user:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local" \ No newline at end of file diff --git a/internal/db/database.go b/internal/db/database.go index c7bb454..584a608 100644 --- a/internal/db/database.go +++ b/internal/db/database.go @@ -15,8 +15,10 @@ import ( var DB *gorm.DB func InitDB() (*gorm.DB, error) { - // TODO: 从配置中心读取DSN - dsn := "root:1236@tcp(127.0.0.1:3306)/gpt_load?charset=utf8mb4&parseTime=True&loc=Local" + dsn := os.Getenv("DATABASE_DSN") + if dsn == "" { + return nil, fmt.Errorf("DATABASE_DSN environment variable not set") + } newLogger := logger.New( log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer