mirror of
https://github.com/NotAShelf/goblin.git
synced 2024-11-22 21:31:07 +00:00
51 lines
1.5 KiB
Go
51 lines
1.5 KiB
Go
package metrics
|
|
|
|
import (
|
|
"github.com/gorilla/mux"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
)
|
|
|
|
var (
|
|
pasteLengthHistogram = promauto.NewHistogramVec(
|
|
prometheus.HistogramOpts{
|
|
Namespace: "goblin",
|
|
Subsystem: "paste",
|
|
Name: "length",
|
|
Help: "Length of pastes",
|
|
Buckets: prometheus.ExponentialBuckets(1, 2, 10), // Example buckets
|
|
},
|
|
[]string{"status"}, // Labels for success/failure
|
|
)
|
|
|
|
pasteCreatedCounter = promauto.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Namespace: "goblin",
|
|
Subsystem: "paste",
|
|
Name: "created_total",
|
|
Help: "Total number of pastes created",
|
|
},
|
|
[]string{"status"}, // Labels for success/failure
|
|
)
|
|
)
|
|
|
|
// ObservePasteLength records the length of pastes.
|
|
func ObservePasteLength(status string, length float64) {
|
|
pasteLengthHistogram.WithLabelValues(status).Observe(length)
|
|
}
|
|
|
|
// IncrementPasteCreatedCounter increments the paste creation counter.
|
|
func IncrementPasteCreatedCounter(status string) {
|
|
pasteCreatedCounter.WithLabelValues(status).Inc()
|
|
}
|
|
|
|
// InitPrometheus initializes Prometheus metrics and registers a handler for /metrics endpoint.
|
|
func InitPrometheus(router *mux.Router) {
|
|
// Register Prometheus metrics collectors here
|
|
// For example, you can use promauto to define and register metrics
|
|
|
|
// Register Prometheus HTTP handler for /metrics endpoint with the Mux router
|
|
router.Handle("/metrics", promhttp.Handler())
|
|
}
|