Merge pull request #215 from prusnak/psql-unix-prefix

Enable "unix:" prefix for PostgreSQL connection string
getAlby · Jul 22, 2022 · 2bdd71c · 2bdd71c
2 parents c6a56b5 + 487524c
commit 2bdd71c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/db/db.go b/db/db.go
@@ -14,11 +14,11 @@ import (
 func Open(dsn string) (*bun.DB, error) {
 	var db *bun.DB
 	switch {
-	case strings.HasPrefix(dsn, "postgres"):
+	case strings.HasPrefix(dsn, "postgres://") || strings.HasPrefix(dsn, "postgresql://") || strings.HasPrefix(dsn, "unix://"):
 		dbConn := sql.OpenDB(pgdriver.NewConnector(pgdriver.WithDSN(dsn)))
 		db = bun.NewDB(dbConn, pgdialect.New())
 	default:
-		return nil, fmt.Errorf("Invalid database connection string %s, only postgres is supported", dsn)
+		return nil, fmt.Errorf("Invalid database connection string %s, only (postgres|postgresql|unix):// is supported", dsn)
 	}
 
 	db.AddQueryHook(bundebug.NewQueryHook(