fix: ignore tracing specific headers when sending requests to deploym…

…ents
epam · Nov 15, 2024 · 021a4d0 · 021a4d0
1 parent 605ac59
commit 021a4d0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/server/src/main/java/com/epam/aidial/core/server/util/ProxyUtil.java b/server/src/main/java/com/epam/aidial/core/server/util/ProxyUtil.java
@@ -36,6 +36,9 @@ public class ProxyUtil {
             .enable(MapperFeature.ACCEPT_CASE_INSENSITIVE_ENUMS)
             .build();
 
+    private static final MultiMap TRACE_HEADERS = MultiMap.caseInsensitiveMultiMap()
+            .add("traceparent", "whatever")
+            .add("tracestate", "whatever");
     private static final MultiMap HOP_BY_HOP_HEADERS = MultiMap.caseInsensitiveMultiMap()
             .add(HttpHeaders.CONNECTION, "whatever")
             .add(HttpHeaders.KEEP_ALIVE, "whatever")
@@ -60,7 +63,7 @@ public static void copyHeaders(MultiMap from, MultiMap to, MultiMap excludeHeade
             String key = entry.getKey();
             String value = entry.getValue();
 
-            if (!HOP_BY_HOP_HEADERS.contains(key) && !excludeHeaders.contains(key)) {
+            if (!HOP_BY_HOP_HEADERS.contains(key) && !TRACE_HEADERS.contains(key) && !excludeHeaders.contains(key)) {
                 to.add(key, value);
             }
         }