Merge pull request #52 from kanmu/fix_GetConnectorFromConnStr_for_v4

Fix GetConnectorFromConnStr for v4
kanmu · Dec 4, 2024 · 5be06d2 · 5be06d2
2 parents 7d5be5e + 18d30ba
commit 5be06d2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/db.go b/db.go
@@ -25,14 +25,15 @@ func GetConnectorFromURL(u *url.URL) (driver.Connector, error) {
 	return GetConnectorFromConnStr(u.String())
 }
 
-func GetConnectorFromConnStr(connStr string) (driver.Connector, error) {
+func GetConnectorFromConnStr(connStr string, opts ...stdlib.OptionOpenDB) (driver.Connector, error) {
 	cfg, err := pgx.ParseConfig(connStr)
 
 	if err != nil {
 		return nil, err
 	}
 
-	connector := stdlib.GetConnector(*cfg, stdlib.OptionAfterConnect(PrepareStatements))
+	opts = append(opts, stdlib.OptionAfterConnect(PrepareStatements))
+	connector := stdlib.GetConnector(*cfg, opts...)
 
 	return connector, nil
 }