Add setting for DB sslmode

camptocamp · Jan 20, 2020 · 218e73e · 218e73e
1 parent 1a89a08
commit 218e73e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/config/config.go b/config/config.go
@@ -24,6 +24,7 @@ type DBConfig struct {
 	User         string `long:"db-user" env:"DB_USER" yaml:"user" description:"Database user." default:"gorm"`
 	Password     string `long:"db-password" env:"DB_PASSWORD" yaml:"password" description:"Database password."`
 	Name         string `long:"db-name" env:"DB_NAME" yaml:"name" description:"Database name." default:"gorm"`
+	SSLMode      string `long:"db-sslmode" yaml:"sslmode" description:"Database SSL mode." default:"disable"`
 	NoSync       bool   `long:"no-sync" yaml:"no-sync" description:"Do not sync database."`
 	SyncInterval uint16 `long:"sync-interval" yaml:"sync-interval" description:"DB sync interval (in minutes)" default:"1"`
 }

diff --git a/db/db.go b/db/db.go
@@ -8,8 +8,8 @@ import (
 	"strings"
 
 	"github.com/camptocamp/terraboard/config"
-	"github.com/camptocamp/terraboard/types"
 	"github.com/camptocamp/terraboard/state"
+	"github.com/camptocamp/terraboard/types"
 	"github.com/hashicorp/terraform/states"
 	"github.com/hashicorp/terraform/states/statefile"
 	log "github.com/sirupsen/logrus"
@@ -29,7 +29,7 @@ var pageSize = 20
 // Init setups up the Database and a pointer to it
 func Init(config config.DBConfig, debug bool) *Database {
 	var err error
-	connString := fmt.Sprintf("host=%s port=%d user=%s dbname=%s sslmode=disable password=%s", config.Host, config.Port, config.User, config.Name, config.Password)
+	connString := fmt.Sprintf("host=%s port=%d user=%s dbname=%s sslmode=%s password=%s", config.Host, config.Port, config.User, config.Name, config.SSLMode, config.Password)
 	db, err := gorm.Open("postgres", connString)
 	if err != nil {
 		log.Fatal(err)