You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.2 KiB
49 lines
1.2 KiB
4 months ago
|
package prommetrics
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/prometheus/client_golang/prometheus"
|
||
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||
|
|
"strconv"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
apiCounter = prometheus.NewCounterVec(
|
||
|
|
prometheus.CounterOpts{
|
||
|
|
Name: "api_count",
|
||
|
|
Help: "Total number of API calls",
|
||
|
|
},
|
||
|
|
[]string{"path", "method", "code"},
|
||
|
|
)
|
||
|
|
httpCounter = prometheus.NewCounterVec(
|
||
|
|
prometheus.CounterOpts{
|
||
|
|
Name: "http_count",
|
||
|
|
Help: "Total number of HTTP calls",
|
||
|
|
},
|
||
|
|
[]string{"path", "method", "status"},
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
func ApiInit(prometheusPort int) error {
|
||
|
|
apiRegistry := prometheus.NewRegistry()
|
||
|
|
cs := append(
|
||
|
|
baseCollector,
|
||
|
|
apiCounter,
|
||
|
|
httpCounter,
|
||
|
|
)
|
||
|
|
return Init(apiRegistry, prometheusPort, commonPath, promhttp.HandlerFor(apiRegistry, promhttp.HandlerOpts{}), cs...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func APICall(path string, method string, apiCode int) {
|
||
|
|
apiCounter.With(prometheus.Labels{"path": path, "method": method, "code": strconv.Itoa(apiCode)}).Inc()
|
||
|
|
}
|
||
|
|
|
||
|
|
func HttpCall(path string, method string, status int) {
|
||
|
|
httpCounter.With(prometheus.Labels{"path": path, "method": method, "status": strconv.Itoa(status)}).Inc()
|
||
|
|
}
|
||
|
|
|
||
|
|
//func ApiHandler() http.Handler {
|
||
|
|
// return promhttp.InstrumentMetricHandler(
|
||
|
|
// apiRegistry, promhttp.HandlerFor(apiRegistry, promhttp.HandlerOpts{}),
|
||
|
|
// )
|
||
|
|
//}
|