Alex Matsuev
b2d6d32c86
All checks were successful
Build and release project / Explore-Gitea-Actions (push) Successful in 51s
96 lines
2.1 KiB
Go
96 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/VictoriaMetrics/metrics"
|
|
)
|
|
|
|
// Register various metrics.
|
|
// Metric name may contain labels in Prometheus format - see below.
|
|
var (
|
|
// Register counter without labels.
|
|
// requestsTotal = metrics.NewCounter("requests_total")
|
|
|
|
requestsTotal = metrics.GetOrCreateCounter("requests_total")
|
|
|
|
// Register summary with a single label.
|
|
requestDuration = metrics.NewSummary(`requests_duration_seconds{path="/foobar/baz"}`)
|
|
|
|
// Register gauge with two labels.
|
|
// queueSize = metrics.NewGauge(`queue_size{queue="foobar",topic="baz"}`, func() float64 {
|
|
// return float64(foobarQueue.Len())
|
|
// })
|
|
|
|
// Register histogram with a single label.
|
|
// responseSize = metrics.NewHistogram(`response_size{path="/foo/bar"}`)
|
|
)
|
|
|
|
func main() {
|
|
metrics.ExposeMetadata(true)
|
|
|
|
http.HandleFunc("/", Recovery(Metrics(rootHandler)))
|
|
|
|
// Expose the registered metrics at `/metrics` path.
|
|
http.HandleFunc("/metrics", Recovery(metricsHandler))
|
|
|
|
http.HandleFunc("/hello", Recovery(Metrics(helloHandler)))
|
|
|
|
go func() {
|
|
if err := http.ListenAndServe(":8080", nil); err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
}()
|
|
|
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer cancel()
|
|
|
|
<-ctx.Done()
|
|
|
|
}
|
|
|
|
func rootHandler(w http.ResponseWriter, req *http.Request) {
|
|
w.Write([]byte("Simple GoLang metrics service"))
|
|
}
|
|
|
|
// metricsHandler ...
|
|
func metricsHandler(w http.ResponseWriter, req *http.Request) {
|
|
metrics.WritePrometheus(w, true)
|
|
}
|
|
|
|
// helloHandler ...
|
|
func helloHandler(w http.ResponseWriter, req *http.Request) {
|
|
w.Write([]byte("Test /hello location"))
|
|
}
|
|
|
|
// Metrics ...
|
|
func Metrics(h http.HandlerFunc) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
requestsTotal.Inc()
|
|
|
|
startTime := time.Now()
|
|
|
|
h(w, r)
|
|
|
|
requestDuration.UpdateDuration(startTime)
|
|
}
|
|
}
|
|
|
|
// Recovery ...
|
|
func Recovery(h http.HandlerFunc) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}()
|
|
|
|
h(w, r)
|
|
}
|
|
}
|