update: postgresql兼容ssl链接

go-home-admin · Nov 20, 2024 · 3c54b65 · 3c54b65
1 parent e3cb2ab
commit 3c54b65
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/bootstrap/providers/postgresql/postgresql_provider.go b/bootstrap/providers/postgresql/postgresql_provider.go
@@ -9,6 +9,7 @@ import (
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
 	"gorm.io/gorm/logger"
+	"strings"
 )
 
 // PostgresqlProvider @Bean("postgresql")
@@ -51,11 +52,16 @@ func (p *PostgresqlProvider) Init() {
 			gConf.Logger.LogMode(logger.LogLevel(logrus.DebugLevel))
 		}
 
-		dsn := postgres.Open(fmt.Sprintf("host=%v user=%v password=%v dbname=%v port=%v sslmode=disable TimeZone=%v", host, username, password, dbname, port, timezone))
+		dsnStr := fmt.Sprintf("host=%v user=%v password=%v dbname=%v port=%v sslmode=require TimeZone=%v", host, username, password, dbname, port, timezone)
+		dsn := postgres.Open(dsnStr)
 		db, err := gorm.Open(dsn, gConf)
 		if err != nil {
-			logrus.Error("postgresql 链接错误", err)
-			panic(err)
+			dsn = postgres.Open(strings.Replace(dsnStr, "require", "disable", 1))
+			db, err = gorm.Open(dsn, gConf)
+			if err != nil {
+				logrus.Error("postgresql 链接错误", err)
+				panic(err)
+			}
 		}
 		p.dbs[name.(string)] = db
 	}