api/plugins/http_mock/init.go

package http_mock

import (
	"github.com/baez90/inetmock/pkg/api"
	"github.com/baez90/inetmock/pkg/logging"
	"github.com/baez90/inetmock/pkg/metrics"
	"github.com/prometheus/client_golang/prometheus"
	"go.uber.org/zap"
)

var (
	totalRequestCounter      *prometheus.CounterVec
	requestDurationHistogram *prometheus.HistogramVec
)

func init() {
	var err error
	var logger logging.Logger
	if logger, err = logging.CreateLogger(); err != nil {
		panic(err)
	}
	logger = logger.With(
		zap.String("protocol_handler", name),
	)
	if totalRequestCounter, err = metrics.Counter(
		name,
		"total_requests",
		"",
		handlerNameLblName,
		ruleMatchedLblName,
	); err != nil {
		panic(err)
	}
	if requestDurationHistogram, err = metrics.Histogram(
		name,
		"request_duration",
		"",
		nil,
		handlerNameLblName,
	); err != nil {
		panic(err)
	}
	api.Registry().RegisterHandler(name, func() api.ProtocolHandler {
		return &httpHandler{
			logger: logger,
		}
	})
}