sad/soju
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add Prometheus instrumentation for the database

Browse Source
This commit is contained in:
Simon Ser 2021-11-17 15:40:02 +01:00
parent 5a5c93a756
commit ea98ac042e
4 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/soju/main.go
View File

@ -293,6 +293,10 @@ func main() {
log.Printf("server listening on %q", listen)
}
if db, ok := db.(soju.MetricsCollectorDatabase); ok && srv.MetricsRegistry != nil {
srv.MetricsRegistry.MustRegister(db.MetricsCollector())
}
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP)

7
db.go
View File

@ -6,6 +6,8 @@ import (
"net/url"
"strings"
"time"
"github.com/prometheus/client_golang/prometheus"
)
type Database interface {
@ -28,6 +30,11 @@ type Database interface {
StoreClientDeliveryReceipts(ctx context.Context, networkID int64, client string, receipts []DeliveryReceipt) error
}
type MetricsCollectorDatabase interface {
Database
MetricsCollector() prometheus.Collector
}
func OpenDB(driver, source string) (Database, error) {
switch driver {
case "sqlite3":

6
db_postgres.go
View File

@ -10,6 +10,8 @@ import (
"time"
_ "github.com/lib/pq"
"github.com/prometheus/client_golang/prometheus"
promcollectors "github.com/prometheus/client_golang/prometheus/collectors"
)
const postgresQueryTimeout = 5 * time.Second
@ -148,6 +150,10 @@ func (db *PostgresDB) Close() error {
return db.db.Close()
}
func (db *PostgresDB) MetricsCollector() prometheus.Collector {
return promcollectors.NewDBStatsCollector(db.db, "main")
}
func (db *PostgresDB) Stats(ctx context.Context) (*DatabaseStats, error) {
ctx, cancel := context.WithTimeout(ctx, postgresQueryTimeout)
defer cancel()

6
db_sqlite.go
View File

@ -10,6 +10,8 @@ import (
"time"
_ "github.com/mattn/go-sqlite3"
"github.com/prometheus/client_golang/prometheus"
promcollectors "github.com/prometheus/client_golang/prometheus/collectors"
)
const sqliteQueryTimeout = 5 * time.Second
@ -238,6 +240,10 @@ func (db *SqliteDB) upgrade() error {
return tx.Commit()
}
func (db *SqliteDB) MetricsCollector() prometheus.Collector {
return promcollectors.NewDBStatsCollector(db.db, "main")
}
func (db *SqliteDB) Stats(ctx context.Context) (*DatabaseStats, error) {
db.lock.RLock()
defer db.lock.RUnlock()