From 487524c236f5dd06313c1999ab315c698d502a9e Mon Sep 17 00:00:00 2001 From: Pavol Rusnak Date: Thu, 21 Jul 2022 17:33:05 +0200 Subject: [PATCH] Enable "unix:" prefix for PostgreSQL connection string --- db/db.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/db/db.go b/db/db.go index d2458b5..a194ae7 100644 --- 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(