/
connector.go
42 lines (36 loc) · 1.03 KB
/
connector.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
package sqldrv
import (
"database/sql"
"database/sql/driver"
"fmt"
"net/url"
"github.com/jackc/pgx/v5/stdlib"
)
// NewDB is a convenience function for creating a *sql.DB from a DB URL and application_name.
func NewDB(urlStr, appName string) (*sql.DB, error) {
c, err := NewConnector(urlStr, appName)
if err != nil {
return nil, err
}
return sql.OpenDB(c), nil
}
// AppURL will add the application_name parameter to the provided URL.
func AppURL(urlStr, appName string) (string, error) {
u, err := url.Parse(urlStr)
if err != nil {
return "", fmt.Errorf("parse db url: %w", err)
}
q := u.Query()
q.Set("application_name", appName)
q.Set("enable_seqscan", "off")
u.RawQuery = q.Encode()
return u.String(), nil
}
// NewConnector will create a new driver.Connector with retry enabled and the provided application_name.
func NewConnector(urlStr, appName string) (driver.Connector, error) {
urlStr, err := AppURL(urlStr, appName)
if err != nil {
return nil, err
}
return NewRetryDriver(&stdlib.Driver{}, 10).OpenConnector(urlStr)
}