From 68f1ccfed40fa9f5fb1a5efe2e1123ac38763cf9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=AE=E5=87=89?= Date: Wed, 19 Jan 2022 09:14:31 +0800 Subject: [PATCH] :sparkles: add sslmode for postgres --- bootstrap/model.go | 5 ++--- conf/config.go | 2 ++ 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/bootstrap/model.go b/bootstrap/model.go index 8c4510b0..8c8caf17 100644 --- a/bootstrap/model.go +++ b/bootstrap/model.go @@ -60,14 +60,13 @@ func InitModel() { } case "postgres": { - dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai", - databaseConfig.Host, databaseConfig.User, databaseConfig.Password, databaseConfig.Name, databaseConfig.Port) + dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=Asia/Shanghai", + databaseConfig.Host, databaseConfig.User, databaseConfig.Password, databaseConfig.Name, databaseConfig.Port, databaseConfig.SslMode) db, err := gorm.Open(postgres.Open(dsn), gormConfig) if err != nil { log.Errorf("failed to connect database:%s", err.Error()) } conf.DB = db - } default: log.Fatalf("not supported database type: %s", databaseConfig.Type) diff --git a/conf/config.go b/conf/config.go index ae72e4f4..fd829275 100644 --- a/conf/config.go +++ b/conf/config.go @@ -9,6 +9,7 @@ type Database struct { Name string `json:"name"` TablePrefix string `json:"table_prefix"` DBFile string `json:"db_file"` + SslMode string `json:"ssl_mode"` } type Scheme struct { @@ -41,6 +42,7 @@ func DefaultConfig() *Config { Port: 0, TablePrefix: "x_", DBFile: "data/data.db", + SslMode: "disable", }, Cache: CacheConfig{ Expiration: 60,