通过 Filter 与包装 HttpServletRequest 添加自定义 header

有时通过过滤器 Filter 与包装 HttpServletRequest 添加自定义 header；非常有用。有类似如下一些场景：

通过网关转发到应用服务，都会透传一个 long 类型的用户 id；有个灰度方案，通过这个用户 id 取余进行灰度逻辑；想在每个请求都添加一个灰度的请求头 grey-flag。这时就需要向 HttpServletRequest 添加请求。默认 HttpServletRequest 是没有提供类型 addHeader(name, value) 或者 setHeader(name, value) 的方法的。

项目有时候为了简化业务逻辑，想在 Filter 中统一添加一些 Header 信息。比如使用 token 获取到 userId 后，添加到 Header 中，这样各个业务就可以直接使用 userId。

定义 HttpServletRequest 包装类

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.util.*;
 
/**
 *  Filter请求拦截中添加header属性 
 *
 */
public class CustomRequestWrapper extends HttpServletRequestWrapper {
 
    private Map<String, String> headerMap = new HashMap<String, String>();
 
    public CustomRequestWrapper(HttpServletRequest request) {
        super(request);
    }
 
    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);
    }
}

自定义Filter

public class CustomHeaderFilter implements Filter {
 
    @Override    
    public void destroy() {
    
    }
    
    @Override
    public void doFilter(ServletRequest request, ServletResponse response,  FilterChain chain) throws IOException, ServletException {
            HttpServletRequest req = (HttpServletRequest) request;
            CustomRequestWrapper requestWrapper = new CustomRequestWrapper(req);
            //获得请求参数中的token值        
            String token = request.getParamter("token");
            if（!StringUtils.isEntry(token)）{
                  // TODO 处理token调用addHeader方法   
            }        
            chain.doFilter(request, response); 
            // Goes to default servlet.
    }
    
    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
     
    }
}

最后编辑于：2022.07.27 22:40:36