option to use pgx instead pq (pressly#283)

EBSCOIS · Nov 24, 2021 · 4148551 · 4148551
1 parent 1f07599
commit 4148551
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/db.go b/db.go
@@ -22,7 +22,7 @@ func OpenDBWithDriver(driver string, dbstring string) (*sql.DB, error) {
 	}
 
 	switch driver {
-	case "postgres", "sqlite3", "mysql", "sqlserver", "clickhouse":
+	case "postgres", "pgx", "sqlite3", "mysql", "sqlserver", "clickhouse":
 		return sql.Open(driver, dbstring)
 	default:
 		return nil, fmt.Errorf("unsupported driver %s", driver)

diff --git a/dialect.go b/dialect.go
@@ -25,7 +25,7 @@ func GetDialect() SQLDialect {
 // SetDialect sets the SQLDialect
 func SetDialect(d string) error {
 	switch d {
-	case "postgres":
+	case "postgres", "pgx":
 		dialect = &PostgresDialect{}
 	case "mysql":
 		dialect = &MySQLDialect{}