vid-app-common/src/main/java/org/onap/vid/controller/filter/PromiseEcompRequestIdFilter.java

   1 package org.onap.vid.controller.filter;
   2
   3
   4 import com.google.common.collect.ImmutableList;
   5 import org.apache.commons.lang3.StringUtils;
   6 import org.openecomp.portalsdk.core.logging.logic.EELFLoggerDelegate;
   7 import org.springframework.web.filter.GenericFilterBean;
   8
   9 import javax.servlet.FilterChain;
  10 import javax.servlet.ServletException;
  11 import javax.servlet.ServletRequest;
  12 import javax.servlet.ServletResponse;
  13 import javax.servlet.annotation.WebFilter;
  14 import javax.servlet.http.HttpServletRequest;
  15 import javax.servlet.http.HttpServletRequestWrapper;
  16 import javax.servlet.http.HttpServletResponse;
  17 import java.io.IOException;
  18 import java.util.Collections;
  19 import java.util.Enumeration;
  20 import java.util.UUID;
  21
  22 import static org.openecomp.portalsdk.core.util.SystemProperties.ECOMP_REQUEST_ID;
  23
  24 @WebFilter(urlPatterns = "/*")
  25 public class PromiseEcompRequestIdFilter extends GenericFilterBean {
  26
  27     private final static EELFLoggerDelegate LOGGER = EELFLoggerDelegate.getLogger(PromiseEcompRequestIdFilter.class);
  28     private final static String REQUEST_ID_RESPONSE_HEADER = ECOMP_REQUEST_ID + "-echo";
  29
  30
  31     @Override
  32     public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
  33             throws IOException, ServletException {
  34
  35         if (request instanceof HttpServletRequest) {
  36             request = wrapIfNeeded(request);
  37
  38             if (response instanceof HttpServletResponse) {
  39                 final String actualRequestId = ((HttpServletRequest) request).getHeader(ECOMP_REQUEST_ID);
  40                 ((HttpServletResponse) response).addHeader(REQUEST_ID_RESPONSE_HEADER, actualRequestId);
  41             }
  42         }
  43
  44         chain.doFilter(request, response);
  45     }
  46
  47     public static ServletRequest wrapIfNeeded(ServletRequest request) {
  48         final HttpServletRequest httpRequest = (HttpServletRequest) request;
  49         final String originalRequestId = httpRequest.getHeader(ECOMP_REQUEST_ID);
  50
  51         if (StringUtils.isEmpty(originalRequestId)) {
  52             request = new PromiseEcompRequestIdRequestWrapper(httpRequest);
  53         }
  54
  55         return request;
  56     }
  57
  58     private static class PromiseEcompRequestIdRequestWrapper extends HttpServletRequestWrapper {
  59
  60         private final UUID requestId;
  61
  62         PromiseEcompRequestIdRequestWrapper(HttpServletRequest request) {
  63             super(request);
  64             requestId = UUID.randomUUID();
  65         }
  66
  67         @Override
  68         public String getHeader(String name) {
  69             return isRequestIdHeaderName(name) ?
  70                     requestId.toString() : super.getHeader(name);
  71         }
  72
  73         @Override
  74         public Enumeration<String> getHeaders(String name) {
  75             if (isRequestIdHeaderName(name)) {
  76                 return Collections.enumeration(Collections.singleton(requestId.toString()));
  77             } else {
  78                 return super.getHeaders(name);
  79             }
  80         }
  81
  82         @Override
  83         public Enumeration<String> getHeaderNames() {
  84             return Collections.enumeration(ImmutableList.<String>builder()
  85                     .add(ECOMP_REQUEST_ID)
  86                     .addAll(Collections.list(super.getHeaderNames()))
  87                     .build());
  88         }
  89
  90         private boolean isRequestIdHeaderName(String name) {
  91             return ECOMP_REQUEST_ID.equalsIgnoreCase(name);
  92         }
  93     }
  94 }