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