Fix issue with default port in bookmark; cleanup

2016-11-15 21:58:58 -06:00
parent d74bc06cbd
commit 59018287de
2 changed files with 15 additions and 7 deletions
--- a/pkg/bookmarks/bookmarks.go
+++ b/pkg/bookmarks/bookmarks.go
@@ -95,10 +95,15 @@ func GetBookmark(bookmarkPath string, bookmarkName string) (Bookmark, error) {
 	if err != nil {
 		return Bookmark{}, err
 	}
+
 	bookmark, ok := bookmarks[bookmarkName]
 	if !ok {
 		return Bookmark{}, fmt.Errorf("couldn't find a bookmark with name %s", bookmarkName)
 	}
-	return bookmark, nil

+	if bookmark.Port == 0 {
+		bookmark.Port = 5432
+	}
+
+	return bookmark, nil
 }