diff --git a/api/handler/util.go b/api/handler/util.go index abc8acb..625b1b0 100644 --- a/api/handler/util.go +++ b/api/handler/util.go @@ -15,7 +15,6 @@ import ( "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/middleware" "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/internal/logs" cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id" - "go.opentelemetry.io/otel/trace" "go.uber.org/zap" ) @@ -42,10 +41,7 @@ func (h *handler) logAndSendError(ctx context.Context, w http.ResponseWriter, lo zap.String("user", reqInfo.User), zap.Error(err)} fields = append(fields, additional...) - if traceID, err := trace.TraceIDFromHex(reqInfo.TraceID); err == nil && traceID.IsValid() { - fields = append(fields, zap.String("trace_id", reqInfo.TraceID)) - } - h.reqLogger(ctx).Error(logs.RequestFailed, fields...) // consider using h.reqLogger (it requires accept context.Context or http.Request) + h.reqLogger(ctx).Error(logs.RequestFailed, fields...) } func handleDeleteMarker(w http.ResponseWriter, err error) error { diff --git a/api/middleware/response.go b/api/middleware/response.go index 94b633d..e86151f 100644 --- a/api/middleware/response.go +++ b/api/middleware/response.go @@ -11,7 +11,6 @@ import ( "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/api/errors" "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/internal/logs" "git.frostfs.info/TrueCloudLab/frostfs-s3-gw/internal/version" - "go.opentelemetry.io/otel/trace" "go.uber.org/zap" ) @@ -332,10 +331,6 @@ func LogSuccessResponse(l *zap.Logger) Func { fields = append(fields, zap.String("user", reqInfo.User)) } - if traceID, err := trace.TraceIDFromHex(reqInfo.TraceID); err == nil && traceID.IsValid() { - fields = append(fields, zap.String("trace_id", reqInfo.TraceID)) - } - reqLogger.Info(logs.RequestEnd, fields...) }) } diff --git a/api/middleware/tracing.go b/api/middleware/tracing.go index 7597700..28b54ea 100644 --- a/api/middleware/tracing.go +++ b/api/middleware/tracing.go @@ -9,16 +9,21 @@ import ( "go.opentelemetry.io/otel/attribute" semconv "go.opentelemetry.io/otel/semconv/v1.18.0" "go.opentelemetry.io/otel/trace" + "go.uber.org/zap" ) // Tracing adds tracing support for requests. // Must be placed after prepareRequest middleware. -func Tracing() Func { +func Tracing(log *zap.Logger) Func { return func(h http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { appCtx, span := StartHTTPServerSpan(r, "REQUEST S3") - reqInfo := GetReqInfo(r.Context()) - reqInfo.TraceID = span.SpanContext().TraceID().String() + + if traceID := span.SpanContext().TraceID(); traceID.IsValid() { + reqLog := reqLogOrDefault(appCtx, log).With(zap.String("trace_id", traceID.String())) + appCtx = SetReqLogger(appCtx, reqLog) + } + lw := &traceResponseWriter{ResponseWriter: w, ctx: appCtx, span: span} h.ServeHTTP(lw, r.WithContext(appCtx)) }) diff --git a/api/router.go b/api/router.go index ec8c27b..8285908 100644 --- a/api/router.go +++ b/api/router.go @@ -134,7 +134,7 @@ func NewRouter(cfg Config) *chi.Mux { s3middleware.Request(cfg.Log, cfg.MiddlewareSettings), middleware.ThrottleWithOpts(cfg.Throttle), middleware.Recoverer, - s3middleware.Tracing(), + s3middleware.Tracing(cfg.Log), s3middleware.Metrics(cfg.Log, cfg.Handler.ResolveCID, cfg.Metrics, cfg.MiddlewareSettings), s3middleware.LogSuccessResponse(cfg.Log), s3middleware.Auth(cfg.Center, cfg.Log),