telegraf/plugins/inputs/pgbouncer/pgbouncer.go

182 lines
3.9 KiB
Go
Raw Normal View History

2018-08-01 22:44:10 +00:00
package pgbouncer
import (
"bytes"
"github.com/influxdata/telegraf"
"github.com/influxdata/telegraf/internal"
"github.com/influxdata/telegraf/plugins/inputs"
"github.com/influxdata/telegraf/plugins/inputs/postgresql"
_ "github.com/jackc/pgx/stdlib" // register driver
2018-08-01 22:44:10 +00:00
)
type PgBouncer struct {
postgresql.Service
}
var ignoredColumns = map[string]bool{"user": true, "database": true, "pool_mode": true,
"avg_req": true, "avg_recv": true, "avg_sent": true, "avg_query": true,
}
var sampleConfig = `
## specify address via a url matching:
## postgres://[pqgotest[:password]]@localhost[/dbname]\
## ?sslmode=[disable|verify-ca|verify-full]
## or a simple string:
## host=localhost user=pqotest password=... sslmode=... dbname=app_production
##
## All connection parameters are optional.
##
address = "host=localhost user=pgbouncer sslmode=disable"
`
func (p *PgBouncer) SampleConfig() string {
return sampleConfig
}
func (p *PgBouncer) Description() string {
return "Read metrics from one or many pgbouncer servers"
}
func (p *PgBouncer) Gather(acc telegraf.Accumulator) error {
var (
err error
query string
columns []string
)
query = `SHOW STATS`
rows, err := p.DB.Query(query)
if err != nil {
return err
}
defer rows.Close()
// grab the column information from the result
if columns, err = rows.Columns(); err != nil {
return err
}
for rows.Next() {
tags, columnMap, err := p.accRow(rows, acc, columns)
if err != nil {
return err
}
fields := make(map[string]interface{})
for col, val := range columnMap {
_, ignore := ignoredColumns[col]
if !ignore {
fields[col] = *val
}
}
acc.AddFields("pgbouncer", fields, tags)
}
query = `SHOW POOLS`
poolRows, err := p.DB.Query(query)
if err != nil {
return err
}
defer poolRows.Close()
// grab the column information from the result
if columns, err = poolRows.Columns(); err != nil {
return err
}
for poolRows.Next() {
tags, columnMap, err := p.accRow(poolRows, acc, columns)
if err != nil {
return err
}
if user, ok := columnMap["user"]; ok {
if s, ok := (*user).(string); ok && s != "" {
tags["user"] = s
}
2018-08-01 22:44:10 +00:00
}
if poolMode, ok := columnMap["pool_mode"]; ok {
if s, ok := (*poolMode).(string); ok && s != "" {
tags["pool_mode"] = s
}
2018-08-01 22:44:10 +00:00
}
fields := make(map[string]interface{})
for col, val := range columnMap {
_, ignore := ignoredColumns[col]
if !ignore {
fields[col] = *val
}
}
acc.AddFields("pgbouncer_pools", fields, tags)
}
return poolRows.Err()
}
type scanner interface {
Scan(dest ...interface{}) error
}
func (p *PgBouncer) accRow(row scanner, acc telegraf.Accumulator, columns []string) (map[string]string,
map[string]*interface{}, error) {
var columnVars []interface{}
var dbname bytes.Buffer
// this is where we'll store the column name with its *interface{}
columnMap := make(map[string]*interface{})
for _, column := range columns {
columnMap[column] = new(interface{})
}
// populate the array of interface{} with the pointers in the right order
for i := 0; i < len(columnMap); i++ {
columnVars = append(columnVars, columnMap[columns[i]])
}
// deconstruct array of variables and send to Scan
err := row.Scan(columnVars...)
if err != nil {
return nil, nil, err
}
if columnMap["database"] != nil {
// extract the database name from the column map
dbname.WriteString((*columnMap["database"]).(string))
} else {
dbname.WriteString("postgres")
}
var tagAddress string
tagAddress, err = p.SanitizedAddress()
if err != nil {
return nil, nil, err
}
// Return basic tags and the mapped columns
return map[string]string{"server": tagAddress, "db": dbname.String()}, columnMap, nil
}
func init() {
inputs.Add("pgbouncer", func() telegraf.Input {
return &PgBouncer{
Service: postgresql.Service{
MaxIdle: 1,
MaxOpen: 1,
MaxLifetime: internal.Duration{
Duration: 0,
},
IsPgBouncer: true,
},
}
})
}