diff --git a/bootstrap/providers/postgresql/postgresql_provider.go b/bootstrap/providers/postgresql/postgresql_provider.go index 17940f3..69b8790 100644 --- a/bootstrap/providers/postgresql/postgresql_provider.go +++ b/bootstrap/providers/postgresql/postgresql_provider.go @@ -9,7 +9,6 @@ import ( "gorm.io/driver/postgres" "gorm.io/gorm" "gorm.io/gorm/logger" - "strings" ) // PostgresqlProvider @Bean("postgresql") @@ -39,6 +38,7 @@ func (p *PostgresqlProvider) Init() { username := config.GetString("username") password := config.GetString("password") dbname := config.GetString("database") + sslmode := config.GetString("sslmode", "disable") timezone := config.GetString("timezone", "Asia/Shanghai") gConf := &gorm.Config{ @@ -52,16 +52,12 @@ func (p *PostgresqlProvider) Init() { gConf.Logger.LogMode(logger.LogLevel(logrus.DebugLevel)) } - dsnStr := fmt.Sprintf("host=%v user=%v password=%v dbname=%v port=%v sslmode=require TimeZone=%v", host, username, password, dbname, port, timezone) + dsnStr := fmt.Sprintf("host=%v user=%v password=%v dbname=%v port=%v sslmode=%v TimeZone=%v", host, username, password, dbname, port, sslmode, timezone) dsn := postgres.Open(dsnStr) db, err := gorm.Open(dsn, gConf) if err != nil { - dsn = postgres.Open(strings.Replace(dsnStr, "require", "disable", 1)) - db, err = gorm.Open(dsn, gConf) - if err != nil { - logrus.Error("postgresql 链接错误", err) - panic(err) - } + logrus.Error("postgresql connect error", err) + panic(err) } p.dbs[name.(string)] = db }