Feat: configure audit fields and metrics for onix adapter and add local configuration for onix adapterZ

core/module/handler/http_metric.go

@@ -0,0 +1,120 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"strconv"
+	"sync"
+	"time"
+
+	"github.com/beckn-one/beckn-onix/pkg/telemetry"
+	"github.com/google/uuid"
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/metric"
+)
+
+type HTTPMetrics struct {
+	HttpRequestCount metric.Int64Counter
+}
+
+var (
+	httlMetricsInstance *HTTPMetrics
+	httpMetricsOnce     sync.Once
+	httpMetricsErr      error
+)
+
+func newHTTPMetrics() (*HTTPMetrics, error) {
+
+	meter := otel.GetMeterProvider().Meter(telemetry.ScopeName,
+		metric.WithInstrumentationVersion(telemetry.ScopeVersion))
+	m := &HTTPMetrics{}
+	var err error
+
+	if m.HttpRequestCount, err = meter.Int64Counter(
+		"onix_http_request_count",
+		metric.WithDescription("Total HTTP requests by status, route, method, role and calle "),
+		metric.WithUnit("1"),
+	); err != nil {
+		return nil, fmt.Errorf("onix_http_request_count: %w", err)
+	}
+
+	return m, nil
+}
+
+func GetHTTPMetrics(ctx context.Context) (*HTTPMetrics, error) {
+	httpMetricsOnce.Do(func() {
+		httlMetricsInstance, httpMetricsErr = newHTTPMetrics()
+	})
+	return httlMetricsInstance, httpMetricsErr
+}
+
+// StatusClass returns the HTTP status class string (e.g. 200 -> "2xx").
+func StatusClass(statusCode int) string {
+	switch {
+	case statusCode >= 100 && statusCode < 200:
+		return "1xx"
+	case statusCode >= 200 && statusCode < 300:
+		return "2xx"
+	case statusCode >= 300 && statusCode < 400:
+		return "3xx"
+	case statusCode >= 400 && statusCode < 500:
+		return "4xx"
+	default:
+		return "5xx"
+	}
+}
+
+func RecordHTTPRequest(ctx context.Context, statusCode int, action, role, caller string) {
+	m, err := GetHTTPMetrics(ctx)
+	if err != nil || m == nil {
+		return
+	}
+	status := StatusClass(statusCode)
+	attributes := []attribute.KeyValue{
+		telemetry.AttrHTTPStatus.String(status),
+		telemetry.AttrAction.String(action),
+		telemetry.AttrRole.String(role),
+		telemetry.AttrCaller.String(caller),
+	}
+
+	metric_code := action + "_api_total_count"
+	category := "NetworkHealth"
+	if action == "/search" || action == "/discovery" {
+		category = "Discovery"
+	}
+	attributes = append(attributes, specHttpMetricAttr(metric_code, category)...) //TODO: need to update as per the furthur discussion
+	m.HttpRequestCount.Add(ctx, 1, metric.WithAttributes(attributes...))
+}
+
+type responseRecorder struct {
+	http.ResponseWriter
+	statusCode int
+	written    bool
+	record     func()
+}
+
+func (r *responseRecorder) WriteHeader(statusCode int) {
+	if !r.written {
+		r.written = true
+		r.statusCode = statusCode
+		if r.record != nil {
+			r.record()
+		}
+	}
+	r.ResponseWriter.WriteHeader(statusCode)
+}
+
+func specHttpMetricAttr(metricCode, category string) []attribute.KeyValue {
+
+	granularity, frequency := telemetry.GetNetworkMetricsConfig()
+	return []attribute.KeyValue{
+		telemetry.AttrMetricUUID.String(uuid.New().String()),
+		telemetry.AttrMetricCode.String(metricCode),
+		telemetry.AttrMetricCategory.String(category),
+		telemetry.AttrMetricGranularity.String(granularity),
+		telemetry.AttrMetricFrequency.String(frequency),
+		telemetry.AttrObservedTimeUnixNano.String(strconv.FormatInt(time.Now().UnixNano(), 10)),
+	}
+}