api/internal/endpoint/handler/http/proxy/register.go

39 lines
898 B
Go
Raw Normal View History

2021-01-13 17:07:04 +00:00
package proxy
2020-04-08 22:59:22 +00:00
import (
"github.com/prometheus/client_golang/prometheus"
"gitlab.com/inetmock/inetmock/internal/endpoint"
"gitlab.com/inetmock/inetmock/pkg/logging"
"gitlab.com/inetmock/inetmock/pkg/metrics"
2020-04-08 22:59:22 +00:00
"go.uber.org/zap"
"gopkg.in/elazarl/goproxy.v1"
2020-04-08 22:59:22 +00:00
)
var (
handlerNameLblName = "handler_name"
requestDurationHistogram *prometheus.HistogramVec
)
func AddHTTPProxy(registry endpoint.HandlerRegistry) (err error) {
var logger logging.Logger
if logger, err = logging.CreateLogger(); err != nil {
return
}
2020-04-08 22:59:22 +00:00
logger = logger.With(
zap.String("protocol_handler", name),
2020-04-08 22:59:22 +00:00
)
if requestDurationHistogram, err = metrics.Histogram(name, "request_duration", "", nil, handlerNameLblName); err != nil {
return
}
registry.RegisterHandler(name, func() endpoint.ProtocolHandler {
return &httpProxy{
logger: logger,
proxy: goproxy.NewProxyHttpServer(),
}
})
return
2020-04-08 22:59:22 +00:00
}