2021-01-13 17:07:04 +00:00
|
|
|
package mock
|
2020-04-01 22:58:44 +00:00
|
|
|
|
|
|
|
import (
|
2020-10-02 09:56:48 +00:00
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
2020-12-26 13:11:49 +00:00
|
|
|
"gitlab.com/inetmock/inetmock/pkg/api"
|
|
|
|
"gitlab.com/inetmock/inetmock/pkg/metrics"
|
2020-04-01 22:58:44 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
name = "dns_mock"
|
|
|
|
)
|
|
|
|
|
2020-10-02 09:56:48 +00:00
|
|
|
var (
|
|
|
|
handlerNameLblName = "handler_name"
|
|
|
|
totalHandledRequestsCounter *prometheus.CounterVec
|
|
|
|
unhandledRequestsCounter *prometheus.CounterVec
|
|
|
|
requestDurationHistogram *prometheus.HistogramVec
|
|
|
|
)
|
|
|
|
|
2020-12-26 13:11:49 +00:00
|
|
|
func AddDNSMock(registry api.HandlerRegistry) (err error) {
|
2020-10-02 09:56:48 +00:00
|
|
|
if totalHandledRequestsCounter, err = metrics.Counter(
|
|
|
|
name,
|
|
|
|
"handled_requests_total",
|
|
|
|
"",
|
|
|
|
handlerNameLblName,
|
|
|
|
); err != nil {
|
2020-12-26 13:11:49 +00:00
|
|
|
return
|
2020-10-02 09:56:48 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if unhandledRequestsCounter, err = metrics.Counter(
|
|
|
|
name,
|
|
|
|
"unhandled_requests_total",
|
|
|
|
"",
|
|
|
|
handlerNameLblName,
|
|
|
|
); err != nil {
|
2020-12-26 13:11:49 +00:00
|
|
|
return
|
2020-10-02 09:56:48 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if requestDurationHistogram, err = metrics.Histogram(
|
|
|
|
name,
|
|
|
|
"request_duration",
|
|
|
|
"",
|
|
|
|
nil,
|
|
|
|
handlerNameLblName,
|
|
|
|
); err != nil {
|
2020-12-26 13:11:49 +00:00
|
|
|
return
|
2020-10-02 09:56:48 +00:00
|
|
|
}
|
|
|
|
|
2020-12-26 13:11:49 +00:00
|
|
|
registry.RegisterHandler(name, func() api.ProtocolHandler {
|
2021-01-07 21:00:12 +00:00
|
|
|
return &dnsHandler{}
|
2020-04-01 22:58:44 +00:00
|
|
|
})
|
2020-12-26 13:11:49 +00:00
|
|
|
|
|
|
|
return
|
2020-04-01 22:58:44 +00:00
|
|
|
}
|