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

Enable "unix:" prefix for PostgreSQL connection string
This commit is contained in:
kiwiidb
2022-07-22 09:17:56 +02:00
committed by GitHub

View File

@@ -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(