rushlink/metrics.go

52 lines
1.3 KiB
Go
Raw Normal View History

2019-11-09 15:50:12 +01:00
package rushlink
import (
"log"
"net/http"
"time"
"gitea.hashru.nl/dsprenkels/rushlink/internal/db"
"github.com/gorilla/mux"
"github.com/pkg/errors"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promauto"
"github.com/prometheus/client_golang/prometheus/promhttp"
bolt "go.etcd.io/bbolt"
)
func StartMetricsServer(addr string, db *db.Database) {
var (
_ = promauto.NewGaugeFunc(prometheus.GaugeOpts{
Namespace: "rushlink",
2019-09-15 22:54:07 +02:00
Subsystem: "pastes",
Name: "urls_total",
2019-09-15 22:54:07 +02:00
Help: "The current amount of pastes in the database.",
}, func() float64 {
var metric float64
if err := db.Bolt.View(func(tx *bolt.Tx) error {
2019-09-15 22:54:07 +02:00
bucket := tx.Bucket([]byte("pastes"))
if bucket == nil {
2019-09-15 22:54:07 +02:00
return errors.New("bucket 'pastes' could not be found")
}
metric = float64(bucket.Stats().KeyN)
return nil
}); err != nil {
2019-09-15 22:54:07 +02:00
log.Printf("error: %v", errors.Wrap(err, "fetching pastes_total metric"))
return 0
}
return metric
})
)
router := mux.NewRouter()
router.Handle("/metrics", promhttp.Handler()).Methods("GET")
srv := &http.Server{
Handler: router,
Addr: addr,
WriteTimeout: 15 * time.Second,
ReadTimeout: 15 * time.Second,
}
log.Fatal(srv.ListenAndServe())
}