Support postgres:// schema connection URLs

GreenmaskIO · Feb 6, 2025 · 9483d00 · 9483d00
1 parent 8eb1ca1
commit 9483d00
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/internal/db/postgres/pgdump/pgdump.go b/internal/db/postgres/pgdump/pgdump.go
@@ -108,7 +108,7 @@ type Options struct {
 
 func (o *Options) GetPgDSN() (string, error) {
 	// URI or Standard format
-	if strings.HasPrefix(o.DbName, "postgresql://") || strings.Contains(o.DbName, "=") {
+	if strings.HasPrefix(o.DbName, "postgresql://") || strings.HasPrefix(o.DbName, "postgres://") || strings.Contains(o.DbName, "=") {
 		return o.DbName, nil
 	}
 

diff --git a/internal/db/postgres/pgrestore/pgrestore.go b/internal/db/postgres/pgrestore/pgrestore.go
@@ -138,7 +138,7 @@ func (o *Options) ToDataSectionSettings() *DataSectionSettings {
 }
 
 func (o *Options) GetPgDSN() (string, error) {
-	if strings.HasPrefix(o.DbName, "postgresql://") || strings.Contains(o.DbName, "=") {
+	if strings.HasPrefix(o.DbName, "postgresql://") || strings.HasPrefix(o.DbName, "postgres://") || strings.Contains(o.DbName, "=") {
 		return o.DbName, nil
 	}