RepoMirrors/postgres_exporter
1
0
Fork 0
mirror of https://github.com/prometheus-community/postgres_exporter synced 2025-02-02 03:21:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Localize the shared database connection to the Exporter object.

Browse Source 
This allows the integration tests to properly exercise configuration parameters.
This commit is contained in:
Will Rouesnel 2017-08-03 23:12:10 +10:00
parent 41cb2e7a68
commit c692b4e76a
1 changed files with 21 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
postgres_exporter.go
View File

@ -28,8 +28,6 @@ import (
// (semantic version)-(commitish) form.
var Version = "0.0.1"
var sharedDBConn *sql.DB
var (
listenAddress = flag.String(
"web.listen-address", ":9187",
@ -673,6 +671,11 @@ type Exporter struct {
duration, error prometheus.Gauge
totalScrapes prometheus.Counter
// dbDsn is the connection string used to establish the dbConnection
dbDsn string
// dbConnection is used to allow re-using the DB connection between scrapes
dbConnection *sql.DB
// Last version used to calculate metric map. If mismatch on scrape,
// then maps are recalculated.
lastMapVersion semver.Version
@ -923,8 +926,16 @@ func (e *Exporter) checkMapVersions(ch chan<- prometheus.Metric, db *sql.DB) err
return nil
}
func getDB(conn string) (*sql.DB, error) {
if sharedDBConn == nil {
func (e *Exporter) getDB(conn string) (*sql.DB, error) {
// Has dsn changed?
if (e.dbConnection != nil) && (e.dsn != e.dbDsn) {
err := e.dbConnection.Close()
log.Warnln("Error while closing obsolete DB connection:", err)
e.dbConnection = nil
e.dbDsn = ""
}
if e.dbConnection == nil {
d, err := sql.Open("postgres", conn)
if err != nil {
return nil, err
@ -935,10 +946,12 @@ func getDB(conn string) (*sql.DB, error) {
}
d.SetMaxOpenConns(1)
d.SetMaxIdleConns(1)
sharedDBConn = d
e.dbConnection = d
e.dbDsn = e.dsn
log.Infoln("Established new database connection.")
}
return sharedDBConn, nil
return e.dbConnection, nil
}
func (e *Exporter) scrape(ch chan<- prometheus.Metric) {
@ -949,10 +962,11 @@ func (e *Exporter) scrape(ch chan<- prometheus.Metric) {
e.error.Set(0)
e.totalScrapes.Inc()
db, err := getDB(e.dsn)
db, err := e.getDB(e.dsn)
if err != nil {
loggableDsn := "could not parse DATA_SOURCE_NAME"
if pDsn, pErr := url.Parse(e.dsn); pErr != nil {
log.Debugln("Blanking password for loggable DSN:", e.dsn)
pDsn.User = url.UserPassword(pDsn.User.Username(), "xxx")
loggableDsn = pDsn.String()
}