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