99 "gorm.io/driver/postgres"
1010 "gorm.io/gorm"
1111 "gorm.io/gorm/logger"
12- "strings"
1312)
1413
1514// PostgresqlProvider @Bean("postgresql")
@@ -39,6 +38,7 @@ func (p *PostgresqlProvider) Init() {
3938 username := config .GetString ("username" )
4039 password := config .GetString ("password" )
4140 dbname := config .GetString ("database" )
41+ sslmode := config .GetString ("sslmode" , "disable" )
4242 timezone := config .GetString ("timezone" , "Asia/Shanghai" )
4343
4444 gConf := & gorm.Config {
@@ -52,16 +52,12 @@ func (p *PostgresqlProvider) Init() {
5252 gConf .Logger .LogMode (logger .LogLevel (logrus .DebugLevel ))
5353 }
5454
55- dsnStr := fmt .Sprintf ("host=%v user=%v password=%v dbname=%v port=%v sslmode=require TimeZone=%v" , host , username , password , dbname , port , timezone )
55+ dsnStr := fmt .Sprintf ("host=%v user=%v password=%v dbname=%v port=%v sslmode=%v TimeZone=%v" , host , username , password , dbname , port , sslmode , timezone )
5656 dsn := postgres .Open (dsnStr )
5757 db , err := gorm .Open (dsn , gConf )
5858 if err != nil {
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- }
59+ logrus .Error ("postgresql connect error" , err )
60+ panic (err )
6561 }
6662 p .dbs [name .(string )] = db
6763 }
0 commit comments