Skip to content

Commit 3c54b65

Browse files
author
zodial
committed
update: postgresql兼容ssl链接
1 parent e3cb2ab commit 3c54b65

1 file changed

Lines changed: 9 additions & 3 deletions

File tree

bootstrap/providers/postgresql/postgresql_provider.go

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ import (
99
"gorm.io/driver/postgres"
1010
"gorm.io/gorm"
1111
"gorm.io/gorm/logger"
12+
"strings"
1213
)
1314

1415
// PostgresqlProvider @Bean("postgresql")
@@ -51,11 +52,16 @@ func (p *PostgresqlProvider) Init() {
5152
gConf.Logger.LogMode(logger.LogLevel(logrus.DebugLevel))
5253
}
5354

54-
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))
55+
dsnStr := fmt.Sprintf("host=%v user=%v password=%v dbname=%v port=%v sslmode=require TimeZone=%v", host, username, password, dbname, port, timezone)
56+
dsn := postgres.Open(dsnStr)
5557
db, err := gorm.Open(dsn, gConf)
5658
if err != nil {
57-
logrus.Error("postgresql 链接错误", err)
58-
panic(err)
59+
dsn = postgres.Open(strings.Replace(dsnStr, "require", "disable", 1))
60+
db, err = gorm.Open(dsn, gConf)
61+
if err != nil {
62+
logrus.Error("postgresql 链接错误", err)
63+
panic(err)
64+
}
5965
}
6066
p.dbs[name.(string)] = db
6167
}

0 commit comments

Comments
 (0)