diff --git a/README.md b/README.md index 92f6c62..94ff6b4 100644 --- a/README.md +++ b/README.md @@ -74,6 +74,7 @@ Application Options: --host= Server hostname or IP (localhost) --port= Server port (5432) --user= Database user (postgres) + --pass= Password for user --db= Database name (postgres) --ssl= SSL option (disable) --listen= HTTP server listen port (8080) diff --git a/main.go b/main.go index 56fb186..53355f4 100644 --- a/main.go +++ b/main.go @@ -19,6 +19,7 @@ var options struct { Host string `long:"host" description:"Server hostname or IP" default:"localhost"` Port int `long:"port" description:"Server port" default:"5432"` User string `long:"user" description:"Database user" default:"postgres"` + Pass string `long:"pass" description:"Password for user"` DbName string `long:"db" description:"Database name" default:"postgres"` Ssl string `long:"ssl" description:"SSL option" default:"disable"` HttpPort uint `long:"listen" description:"HTTP server listen port" default:"8080"` @@ -36,12 +37,18 @@ func getConnectionString() string { return options.Url } - return fmt.Sprintf( + str := fmt.Sprintf( "host=%s port=%d user=%s dbname=%s sslmode=%s", options.Host, options.Port, options.User, options.DbName, options.Ssl, ) + + if options.Pass != "" { + str += fmt.Sprintf(" password=%s", options.Pass) + } + + return str } func initClient() {