✨ add sslmode for postgres

DerekTso · Jan 19, 2022 · 68f1ccf · 68f1ccf
1 parent 54272db
commit 68f1ccf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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,