mirror of
https://github.com/crowdsecurity/crowdsec.git
synced 2025-05-11 04:15:54 +02:00
3945a991bd
* alerts * drop CTX from dbclient * lint * pkg/database/alerts: context.TODO() * cscli: context.Background() -> cmd.Context()
128 lines
3.2 KiB
Go
128 lines
3.2 KiB
Go
package v1
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
/*prometheus*/
|
|
var LapiRouteHits = prometheus.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Name: "cs_lapi_route_requests_total",
|
|
Help: "Number of calls to each route per method.",
|
|
},
|
|
[]string{"route", "method"},
|
|
)
|
|
|
|
/*hits per machine*/
|
|
var LapiMachineHits = prometheus.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Name: "cs_lapi_machine_requests_total",
|
|
Help: "Number of calls to each route per method grouped by machines.",
|
|
},
|
|
[]string{"machine", "route", "method"},
|
|
)
|
|
|
|
/*hits per bouncer*/
|
|
var LapiBouncerHits = prometheus.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Name: "cs_lapi_bouncer_requests_total",
|
|
Help: "Number of calls to each route per method grouped by bouncers.",
|
|
},
|
|
[]string{"bouncer", "route", "method"},
|
|
)
|
|
|
|
/*
|
|
keep track of the number of calls (per bouncer) that lead to nil/non-nil responses.
|
|
|
|
while it's not exact, it's a good way to know - when you have a rutpure bouncer - what is the rate of ok/ko answers you got from lapi
|
|
*/
|
|
var LapiNilDecisions = prometheus.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Name: "cs_lapi_decisions_ko_total",
|
|
Help: "Number of calls to /decisions that returned nil result.",
|
|
},
|
|
[]string{"bouncer"},
|
|
)
|
|
|
|
/*hits per bouncer*/
|
|
var LapiNonNilDecisions = prometheus.NewCounterVec(
|
|
prometheus.CounterOpts{
|
|
Name: "cs_lapi_decisions_ok_total",
|
|
Help: "Number of calls to /decisions that returned non-nil result.",
|
|
},
|
|
[]string{"bouncer"},
|
|
)
|
|
|
|
var LapiResponseTime = prometheus.NewHistogramVec(
|
|
prometheus.HistogramOpts{
|
|
Name: "cs_lapi_request_duration_seconds",
|
|
Help: "Response time of LAPI",
|
|
Buckets: []float64{0.005, 0.01, 0.025, 0.05, 0.075, 0.1, 0.2, 0.3, 0.4, 0.5, 0.75, 1},
|
|
},
|
|
[]string{"endpoint", "method"})
|
|
|
|
func PrometheusBouncersHasEmptyDecision(c *gin.Context) {
|
|
bouncer, _ := getBouncerFromContext(c)
|
|
if bouncer != nil {
|
|
LapiNilDecisions.With(prometheus.Labels{
|
|
"bouncer": bouncer.Name,
|
|
}).Inc()
|
|
}
|
|
}
|
|
|
|
func PrometheusBouncersHasNonEmptyDecision(c *gin.Context) {
|
|
bouncer, _ := getBouncerFromContext(c)
|
|
if bouncer != nil {
|
|
LapiNonNilDecisions.With(prometheus.Labels{
|
|
"bouncer": bouncer.Name,
|
|
}).Inc()
|
|
}
|
|
}
|
|
|
|
func PrometheusMachinesMiddleware() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
machineID, _ := getMachineIDFromContext(c)
|
|
if machineID != "" {
|
|
LapiMachineHits.With(prometheus.Labels{
|
|
"machine": machineID,
|
|
"route": c.Request.URL.Path,
|
|
"method": c.Request.Method,
|
|
}).Inc()
|
|
}
|
|
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
func PrometheusBouncersMiddleware() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
bouncer, _ := getBouncerFromContext(c)
|
|
if bouncer != nil {
|
|
LapiBouncerHits.With(prometheus.Labels{
|
|
"bouncer": bouncer.Name,
|
|
"route": c.Request.URL.Path,
|
|
"method": c.Request.Method,
|
|
}).Inc()
|
|
}
|
|
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
func PrometheusMiddleware() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
startTime := time.Now()
|
|
|
|
LapiRouteHits.With(prometheus.Labels{
|
|
"route": c.Request.URL.Path,
|
|
"method": c.Request.Method,
|
|
}).Inc()
|
|
c.Next()
|
|
|
|
elapsed := time.Since(startTime)
|
|
LapiResponseTime.With(prometheus.Labels{"method": c.Request.Method, "endpoint": c.Request.URL.Path}).Observe(elapsed.Seconds())
|
|
}
|
|
}
|