fix: Fixing up postgres connection strings (#96)

This commit is contained in:
Mukundan Kidambi 2021-09-24 02:49:41 -07:00 committed by GitHub
parent 8d43829bc5
commit a85eafd58c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -3,6 +3,7 @@ package postgres
import (
"database/sql"
"fmt"
"net/url"
"path"
"strconv"
@ -31,8 +32,11 @@ type DBConnection struct {
func Generate(destDir string, dbConn DBConnection, genTemplate ...template.Template) (err error) {
defer utils.ErrorCatch(&err)
connectionString := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=%s %s",
dbConn.Host, strconv.Itoa(dbConn.Port), dbConn.User, dbConn.Password, dbConn.DBName, dbConn.SslMode, dbConn.Params)
if dbConfig.SchemaName == "" {
dbConfig.SchemaName = "public"
}
connectionString := fmt.Sprintf("postgresql://%s:%s@%s:%s/%s?sslmode=%s&search_path=%s",
dbConn.User, url.QueryEscape(dbConn.Password), dbConn.Host, strconv.Itoa(dbConn.Port), dbConn.DBName, dbConn.SslMode, dbConn.SchemaName)
db := openConnection(connectionString)
defer utils.DBClose(db)