Adding query-id header to the list of headers.

o19s · Feb 22, 2024 · 5445b3a · 5445b3a
1 parent a7ae948
commit 5445b3a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/java/org/opensearch/ubl/UserBehaviorLoggingPlugin.java b/src/main/java/org/opensearch/ubl/UserBehaviorLoggingPlugin.java
@@ -55,6 +55,7 @@ public class UserBehaviorLoggingPlugin extends Plugin implements ActionPlugin {
     public Collection<RestHeaderDefinition> getRestHeaders() {
         return List.of(
                 new RestHeaderDefinition(HeaderConstants.EVENT_STORE_HEADER.getHeader(), false),
+                new RestHeaderDefinition(HeaderConstants.QUERY_ID_HEADER.getHeader(), false),
                 new RestHeaderDefinition(HeaderConstants.SESSION_ID_HEADER.getHeader(), false),
                 new RestHeaderDefinition(HeaderConstants.USER_ID_HEADER.getHeader(), false)
         );
@@ -64,6 +65,7 @@ public Collection<RestHeaderDefinition> getRestHeaders() {
     public Collection<String> getTaskHeaders() {
         return List.of(
                 HeaderConstants.EVENT_STORE_HEADER.getHeader(),
+                HeaderConstants.QUERY_ID_HEADER.getHeader(),
                 HeaderConstants.SESSION_ID_HEADER.getHeader(),
                 HeaderConstants.USER_ID_HEADER.getHeader()
         );