feat: add config to enable/disable postgres ssl mode

This commit is contained in:
Vivek R 2019-05-27 17:16:46 +05:30
parent 08b9d153da
commit 09117426ee
3 changed files with 6 additions and 4 deletions

View file

@ -35,7 +35,7 @@ upload_uri = "/uploads"
# Maximum concurrent workers that will attempt to send messages
# simultaneously. This should depend on the number of CPUs the
# machine has and also the number of simultaenous e-mails the
# mail server will
# mail server will
concurrency = 100
# The number of errors (eg: SMTP timeouts while e-mailing) a running
@ -51,6 +51,7 @@ port = 5432
user = "listmonk"
password = ""
database = "listmonk"
ssl_mode = "disable"
# TQekh4quVgGc3HQ

View file

@ -148,7 +148,8 @@ func main() {
viper.GetInt("db.port"),
viper.GetString("db.user"),
viper.GetString("db.password"),
viper.GetString("db.database"))
viper.GetString("db.database"),
viper.GetString("db.ssl_mode"))
if err != nil {
logger.Fatalf("error connecting to DB: %v", err)
}

View file

@ -79,9 +79,9 @@ type Queries struct {
}
// connectDB initializes a database connection.
func connectDB(host string, port int, user, pwd, dbName string) (*sqlx.DB, error) {
func connectDB(host string, port int, user, pwd, dbName string, sslMode string) (*sqlx.DB, error) {
db, err := sqlx.Connect("postgres",
fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s", host, port, user, pwd, dbName))
fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=%s", host, port, user, pwd, dbName, sslMode))
if err != nil {
return nil, err
}