如何在 HandlerInterceptorAdapter 中添加 HttpServletRequest 標頭？

Question

我正在嘗試在我的請求中添加一個 Authorization 標頭，作為我們切換環境時的臨時解決方法。 我試圖在擴展 HandlerInterceptorAdapter 的攔截器中處理它。

我使用了這里創建的 MutableHttpServletRequest 類，以便能夠將標頭添加到請求中，但似乎沒有任何方法可以實際修改攔截器中返回的請求。 有什么想法嗎？

編輯：或者我必須在過濾器中執行此操作嗎？

Answer 1

HttpServletRequest 類型對象是只讀的，為此您應該：

-> 創建一個擴展 HttpServletRequestWrapper 的類（添加一些行為，裝飾器模式）

final public class MutableHttpServletRequest extends HttpServletRequestWrapper {

    private final Map<String, String> customHeaders;

    public MutableHttpServletRequest(HttpServletRequest request){
        super(request);
        this.customHeaders = new HashMap<String, String>();
    }

    public void putHeader(String name, String value){
        this.customHeaders.put(name, value);
    }

    public String getHeader(String name) {
        String headerValue = customHeaders.get(name);

        if (headerValue != null){
            return headerValue;
        }
        return ((HttpServletRequest) getRequest()).getHeader(name);
    }

    public Enumeration<String> getHeaderNames() {
        Set<String> set = new HashSet<String>(customHeaders.keySet());

        @SuppressWarnings("unchecked")
        Enumeration<String> e = ((HttpServletRequest) getRequest()).getHeaderNames();
        while (e.hasMoreElements()) {
            String n = e.nextElement();
            set.add(n);
        }
        return Collections.enumeration(set);
    }
}

-> 創建您的過濾器類

public class CustomGatewayFilter implements Filter {

    @Override
    public void doFilter(ServletRequest request, ServletResponse response,
                         FilterChain chain) throws IOException, ServletException {
        HttpServletRequest req = (HttpServletRequest) request;
        MutableHttpServletRequest mutableRequest = new MutableHttpServletRequest(req);
        mutableRequest.putHeader("referer", "custom value");
        chain.doFilter(mutableRequest, response);
    }
}

-> 並在您的配置類中添加

 .and().addFilterAfter(new CustomGatewayFilter(), ConcurrentSessionFilter.class)

Answer 2

public class YourInterceptor extends HandlerInterceptorAdapter {    

    @Override
    public boolean preHandle(HttpServletRequest request,
            HttpServletResponse response, Object handler) throws Exception {
        SomeModel model = new SomeModel();
        request.setAttribute("someValue", model );
        request.addHeader("xxx","asecret");

        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request,
            HttpServletResponse response, Object handler,
            ModelAndView modelAndView) throws Exception {
        //...
         response.addHeader("dummy-header, "dummy-value");
    }

    @Override
    public void afterCompletion(HttpServletRequest request,
            HttpServletResponse response, Object handler, Exception ex)
            throws Exception {
        SomeModel model = (Long) request.getAttribute("someValue");

    }

}