aai-traversal/src/main/java/org/onap/aai/interceptors/post/ResponseHeaderManipulation.java

   1 package org.onap.aai.interceptors.post;
   2
   3 import java.io.IOException;
   4
   5 import javax.annotation.Priority;
   6 import javax.ws.rs.container.ContainerRequestContext;
   7 import javax.ws.rs.container.ContainerResponseContext;
   8 import javax.ws.rs.container.ContainerResponseFilter;
   9
  10 import org.onap.aai.interceptors.AAIContainerFilter;
  11 import org.onap.aai.interceptors.AAIHeaderProperties;
  12
  13 @Priority(AAIResponseFilterPriority.HEADER_MANIPULATION)
  14 public class ResponseHeaderManipulation extends AAIContainerFilter implements ContainerResponseFilter {
  15
  16
  17         @Override
  18         public void filter(ContainerRequestContext requestContext, ContainerResponseContext responseContext)
  19                         throws IOException {
  20
  21                 updateResponseHeaders(requestContext, responseContext);
  22
  23         }
  24
  25         private void updateResponseHeaders(ContainerRequestContext requestContext,
  26                         ContainerResponseContext responseContext) {
  27                 responseContext.getHeaders().add(AAIHeaderProperties.AAI_TX_ID, requestContext.getProperty(AAIHeaderProperties.AAI_TX_ID));
  28         }
  29
  30 }