prometheus/postgres_exporter
1
0
Fork 0
mirror of https://github.com/prometheus-community/postgres_exporter.git synced 2025-07-31 20:44:25 +03:00
Code Activity

Add collector interface

Browse Source 
Uses node_exporter style collector registration

Signed-off-by: Joe Adams <github@joeadams.io>
This commit is contained in:
Joe Adams
2022-02-14 21:12:27 -05:00
parent 1981623d9d
commit c3b0206369
5 changed files with 342 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
collector/pg_database.go
View File

@ -1,4 +1,4 @@
// Copyright 2021 The Prometheus Authors
// Copyright 2022 The Prometheus Authors
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
@ -15,15 +15,21 @@ package collector
import (
"context"
"database/sql"
"github.com/go-kit/log"
"github.com/prometheus/client_golang/prometheus"
)
type PGDatabaseCollector struct{}
func init() {
registerCollector("database", defaultEnabled, NewPGDatabaseCollector)
}
func NewPGDatabaseCollector() *PGDatabaseCollector {
return &PGDatabaseCollector{}
type PGDatabaseCollector struct {
log log.Logger
}
func NewPGDatabaseCollector(logger log.Logger) (Collector, error) {
return &PGDatabaseCollector{log: logger}, nil
}
var pgDatabase = map[string]*prometheus.Desc{
@ -34,14 +40,17 @@ var pgDatabase = map[string]*prometheus.Desc{
),
}
func (PGDatabaseCollector) Update(ctx context.Context, db *sql.DB, server string) ([]prometheus.Metric, error) {
metrics := []prometheus.Metric{}
func (PGDatabaseCollector) Update(ctx context.Context, server *server, ch chan<- prometheus.Metric) error {
db, err := server.GetDB()
if err != nil {
return err
}
rows, err := db.QueryContext(ctx,
`SELECT pg_database.datname
,pg_database_size(pg_database.datname)
FROM pg_database;`)
if err != nil {
return metrics, err
return err
}
defer rows.Close()
@ -49,15 +58,16 @@ func (PGDatabaseCollector) Update(ctx context.Context, db *sql.DB, server string
var datname string
var size int64
if err := rows.Scan(&datname, &size); err != nil {
return metrics, err
return err
}
metrics = append(metrics, prometheus.MustNewConstMetric(
ch <- prometheus.MustNewConstMetric(
pgDatabase["size_bytes"],
prometheus.GaugeValue, float64(size), datname, server,
))
prometheus.GaugeValue, float64(size), datname, server.GetName(),
)
}
if err := rows.Err(); err != nil {
return metrics, err
return err
}
return metrics, nil
return nil
}