From 170860c14a7778ff141a136899e52ba20ee6607b Mon Sep 17 00:00:00 2001
From: Dmitrii Stepanov <d.stepanov@yadro.com>
Date: Wed, 29 Jan 2025 11:44:20 +0300
Subject: [PATCH] [#1608] logger: Add IO tag logging

Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
---
 pkg/util/logger/log.go | 35 +++++++++++++++--------------------
 1 file changed, 15 insertions(+), 20 deletions(-)

diff --git a/pkg/util/logger/log.go b/pkg/util/logger/log.go
index 269e07d90..413b1d9aa 100644
--- a/pkg/util/logger/log.go
+++ b/pkg/util/logger/log.go
@@ -4,37 +4,32 @@ import (
 	"context"
 
 	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/tracing"
+	qos "git.frostfs.info/TrueCloudLab/frostfs-qos/tagging"
 	"go.uber.org/zap"
 )
 
 func (l *Logger) Debug(ctx context.Context, msg string, fields ...zap.Field) {
-	if traceID := tracing.GetTraceID(ctx); traceID != "" {
-		l.z.Debug(msg, append(fields, zap.String("trace_id", traceID))...)
-		return
-	}
-	l.z.Debug(msg, fields...)
+	l.z.Debug(msg, appendContext(ctx, fields...)...)
 }
 
 func (l *Logger) Info(ctx context.Context, msg string, fields ...zap.Field) {
-	if traceID := tracing.GetTraceID(ctx); traceID != "" {
-		l.z.Info(msg, append(fields, zap.String("trace_id", traceID))...)
-		return
-	}
-	l.z.Info(msg, fields...)
+	l.z.Info(msg, appendContext(ctx, fields...)...)
 }
 
 func (l *Logger) Warn(ctx context.Context, msg string, fields ...zap.Field) {
-	if traceID := tracing.GetTraceID(ctx); traceID != "" {
-		l.z.Warn(msg, append(fields, zap.String("trace_id", traceID))...)
-		return
-	}
-	l.z.Warn(msg, fields...)
+	l.z.Warn(msg, appendContext(ctx, fields...)...)
 }
 
 func (l *Logger) Error(ctx context.Context, msg string, fields ...zap.Field) {
-	if traceID := tracing.GetTraceID(ctx); traceID != "" {
-		l.z.Error(msg, append(fields, zap.String("trace_id", traceID))...)
-		return
-	}
-	l.z.Error(msg, fields...)
+	l.z.Error(msg, appendContext(ctx, fields...)...)
+}
+
+func appendContext(ctx context.Context, fields ...zap.Field) []zap.Field {
+	if traceID := tracing.GetTraceID(ctx); traceID != "" {
+		fields = append(fields, zap.String("trace_id", traceID))
+	}
+	if ioTag, ioTagDefined := qos.IOTagFromContext(ctx); ioTagDefined {
+		fields = append(fields, zap.String("io_tag", ioTag))
+	}
+	return fields
 }