代码清理

server/pkg/orm/orm.go

@@ -3,7 +3,7 @@ package orm
 import (
 	"fmt"
 	"log/slog"
-	"os"
+	"proxy-server/server/pkg/env"
 
 	"github.com/pkg/errors"
 	"gorm.io/driver/postgres"
@@ -14,16 +14,9 @@ import (
 var DB *gorm.DB
 
 func Init() {
-	Host := os.Getenv("DB_HOST")
-	Port := os.Getenv("DB_PORT")
-	Database := os.Getenv("DB_DATABASE")
-	Username := os.Getenv("DB_USERNAME")
-	Password := os.Getenv("DB_PASSWORD")
-	Timezone := os.Getenv("DB_TIMEZONE")
-
 	dsn := fmt.Sprintf(
-		"host=%s port=%s user=%s password=%s dbname=%s sslmode=disable TimeZone=%s",
-		Host, Port, Username, Password, Database, Timezone,
+		"host=%s port=%d user=%s password=%s dbname=%s sslmode=disable TimeZone=%s",
+		env.DbHost, env.DbPort, env.DbUsername, env.DbPassword, env.DbDatabase, env.DbTimezone,
 	)
 
 	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{