Support both postgresql:// and posgres:// url prefix

2015-07-14 22:42:46 -05:00
parent c12853d0b0
commit 785108002e
3 changed files with 14 additions and 2 deletions
--- a/pkg/connection/connection_string.go
+++ b/pkg/connection/connection_string.go
@@ -28,7 +28,7 @@ func FormatUrl(opts command.Options) (string, error) {
 	url := opts.Url

 	// Make sure to only accept urls in a standard format
-	if !strings.Contains(url, "postgres://") {
+	if !strings.HasPrefix(url, "postgres://") && !strings.HasPrefix(url, "postgresql://") {
 		return "", errors.New("Invalid URL. Valid format: postgres://user:password@host:port/db?sslmode=mode")
 	}