2021-01-13 17:07:04 +00:00
|
|
|
package interceptor
|
2020-04-01 02:08:21 +00:00
|
|
|
|
|
|
|
import (
|
2020-12-26 13:11:49 +00:00
|
|
|
"sync"
|
|
|
|
|
2020-04-25 23:18:35 +00:00
|
|
|
"github.com/google/uuid"
|
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/logging"
|
|
|
|
"gitlab.com/inetmock/inetmock/pkg/metrics"
|
2020-04-01 02:08:21 +00:00
|
|
|
"go.uber.org/zap"
|
|
|
|
)
|
|
|
|
|
2020-10-02 09:56:48 +00:00
|
|
|
var (
|
|
|
|
labelNames = []string{"handler_name"}
|
|
|
|
handledRequestCounter *prometheus.CounterVec
|
|
|
|
openConnectionsGauge *prometheus.GaugeVec
|
|
|
|
requestDurationHistogram *prometheus.HistogramVec
|
|
|
|
)
|
|
|
|
|
2020-12-26 13:11:49 +00:00
|
|
|
func AddTLSInterceptor(registry api.HandlerRegistry) (err error) {
|
2020-10-02 09:56:48 +00:00
|
|
|
var logger logging.Logger
|
|
|
|
if logger, err = logging.CreateLogger(); err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2020-04-01 02:08:21 +00:00
|
|
|
logger = logger.With(
|
2020-10-02 09:56:48 +00:00
|
|
|
zap.String("protocol_handler", name),
|
2020-04-01 02:08:21 +00:00
|
|
|
)
|
|
|
|
|
2020-10-02 09:56:48 +00:00
|
|
|
if handledRequestCounter, err = metrics.Counter(name, "handled_requests", "", labelNames...); err != nil {
|
2020-12-26 13:11:49 +00:00
|
|
|
return
|
2020-10-02 09:56:48 +00:00
|
|
|
}
|
|
|
|
if openConnectionsGauge, err = metrics.Gauge(name, "open_connections", "", labelNames...); 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, labelNames...); err != nil {
|
2020-12-26 13:11:49 +00:00
|
|
|
|
2020-10-02 09:56:48 +00:00
|
|
|
}
|
|
|
|
|
2020-12-26 13:11:49 +00:00
|
|
|
registry.RegisterHandler(name, func() api.ProtocolHandler {
|
2020-04-01 02:08:21 +00:00
|
|
|
return &tlsInterceptor{
|
|
|
|
logger: logger,
|
2021-01-02 16:24:06 +00:00
|
|
|
currentConnectionsCount: new(sync.WaitGroup),
|
2020-04-25 23:18:35 +00:00
|
|
|
currentConnections: make(map[uuid.UUID]*proxyConn),
|
2020-06-24 10:25:34 +00:00
|
|
|
connectionsMutex: &sync.Mutex{},
|
2020-04-01 02:08:21 +00:00
|
|
|
}
|
2020-04-25 22:22:45 +00:00
|
|
|
})
|
2020-12-26 13:11:49 +00:00
|
|
|
return
|
2020-04-01 02:08:21 +00:00
|
|
|
}
|