SpringBoot - Web请求头中添加参数

SpringBoot - Web请求头中添加参数

1. 自定义包装类继承HttpServletRequest

   public class HeaderMapRequestWrapper extends HttpServletRequestWrapper {
       public HeaderMapRequestWrapper(HttpServletRequest request) {
           super(request);
       }
   
       /**
        * Key忽略大小写
        */
       private final Map<String, String> headerMap = new LinkedCaseInsensitiveMap<>(Locale.ENGLISH);
   
       public void addHeader(String name, String value) {
           headerMap.put(name, value);
       }
   
       @Override
       public String getHeader(String name) {
           String headerValue = super.getHeader(name);
           if (headerMap.containsKey(name)) {
               headerValue = headerMap.get(name);
           }
           return headerValue;
       }
   
       @Override
       public Enumeration<String> getHeaderNames() {
           List<String> names = Collections.list(super.getHeaderNames());
           for (String name : headerMap.keySet()) {
               names.add(name);
           }
           return Collections.enumeration(names);
       }
   
       @Override
       public Enumeration<String> getHeaders(String name) {
           List<String> values = Collections.list(super.getHeaders(name));
           if (headerMap.containsKey(name)) {
               values.add(headerMap.get(name));
           }
           return Collections.enumeration(values);
       }
   }
   

2. 过滤器使用自定义包装类

   public class SubmitInfoFilter implements Filter {
   
       public SubmitInfoFilter() {
          
       }
   
       @Override
       public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
           HttpServletRequest httpServletRequest = (HttpServletRequest) request;
           HttpServletResponse httpServletResponse = (HttpServletResponse) response;
           // 自定义包装类包装HttpServletRequest并且传入下一个拦截器中
           HeaderMapRequestWrapper wrapper = new HeaderMapRequestWrapper(httpServletRequest);
           wrapper.addHeader("token", "token");
           chain.doFilter(wrapper, response);
       }
   
       @Override
       public void init(FilterConfig filterConfig) throws ServletException {
   
       }
   
       @Override
       public void destroy() {
   
       }
   }