add media service

internal/infrastructure/grpc/interceptor.go

@@ -0,0 +1,59 @@
+package grpc
+
+import (
+	"context"
+	"time"
+
+	"lendry-erp/media/pkg/logger"
+
+	"github.com/google/uuid"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/metadata"
+)
+
+// RequestLoggerInterceptor логирует время выполнения каждого gRPC запроса
+func RequestLoggerInterceptor(
+	ctx context.Context,
+	req interface{},
+	info *grpc.UnaryServerInfo,
+	handler grpc.UnaryHandler,
+) (interface{}, error) {
+
+	start := time.Now()
+	resp, err := handler(ctx, req)
+
+	status := "✅"
+	if err != nil {
+		status = "❌"
+	}
+
+	logger.Info("%s %s %v", status, info.FullMethod, time.Since(start))
+	return resp, err
+}
+
+// TraceIDInterceptor добавляет уникальный ID для отслеживания запроса
+func TraceIDInterceptor(
+	ctx context.Context,
+	req interface{},
+	info *grpc.UnaryServerInfo,
+	handler grpc.UnaryHandler,
+) (interface{}, error) {
+	md, ok := metadata.FromIncomingContext(ctx)
+	if !ok {
+		md = metadata.New(nil)
+	}
+
+	ids := md.Get("x-trace-id")
+	var traceID string
+	if len(ids) == 0 {
+		traceID = uuid.New().String()
+		md.Set("x-trace-id", traceID)
+	} else {
+		traceID = ids[0]
+	}
+
+	ctx = metadata.NewIncomingContext(ctx, md)
+	ctx = context.WithValue(ctx, "traceID", traceID)
+
+	return handler(ctx, req)
+}