0

我需要将自定义标头注入到 Spring Boot 应用程序收到的每个请求中，为此，我编写了一些代码，但它似乎没有完成它的工作。简而言之，我已经实现了 Filter 接口并定义了 doFilter 方法，扩展了 HttpServletRequestWrapper 类，并重写了 getHeader() 和 getHeaderNames() 方法以考虑我从属性文件中读取的自定义标头。

但是，当我进入控制器并检查请求时，我没有得到通过 MyReqWrapper 设置的自定义标头。下面是代码，我也尝试在 Stackoverflow 中搜索它，但在这里找不到关于什么是/可能是错误的解决方案。有人可以指出我正确的方向吗？

另外，请告诉我如何测试自定义标头是否实际设置。

这是过滤器的实现

@Component
@Order(Ordered.HIGHEST_PRECEDENCE)
public class ReqFilter implements Filter {
    private static final String CUSTOMHEADERENABLED = "customheadersenabled";
    private static final String CUSTOMHEADERCOUNT = "customheaderscount";
    
    @Autowired
    private Environment env;

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
        //
    }

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {

        try {
            boolean customHeadersEnabled = Boolean.parseBoolean(env.getProperty(CUSTOMHEADERENABLED, "false"));
            int count = Integer.parseInt(env.getProperty(CUSTOMHEADERCOUNT, "0"));

            if (customHeadersEnabled && count > 0) {
                MyReqWrapper myReq = new MyReqWrapper((HttpServletRequest) servletRequest);
                myReq.processMyHeaders(count, env);
                filterChain.doFilter(customRequest, servletResponse);
            } else {
                filterChain.doFilter(servletRequest, servletResponse);
            }
        }
        catch(ServletException ex){
            throw ex;
        }
    }

    @Override
    public void destroy() {
        //
    }
}

这是扩展 HttpServletRequestWrapper 的自定义请求包装器

final class MyReqWrapper extends HttpServletRequestWrapper {
    private static final String CUSTOMHEADERPREFIX = "header1";
    private final Map<String, String> myHeaders;

    public MyReqWrapper(HttpServletRequest request) {
        super(request);
        myHeaders = new HashMap<>();
    }

    @Override
    public String getHeader(String name) {
        String headerValue = myHeaders.get(name);

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

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

        Enumeration<String> headerNames = ((HttpServletRequest) getRequest()).getHeaderNames();
        while (headerNames.hasMoreElements()) {
            String n = headerNames.nextElement();
            set.add(n);
        }
        return Collections.enumeration(set);
    }

    public void processMyHeaders(int headerCount, Environment env) {
        while(headerCount > 0){
            String [] headerKeyValue = Objects.requireNonNull(env.getProperty(String.format("%1$s%2$s", CUSTOMHEADERPREFIX, headerCount--)))
                    .split(":");
            this.myHeaders.put(headerKeyValue[0], headerKeyValue[1]);
        }
    }
}
4

1 回答 1

0

这已经为我解决了，我忘了用答案来更新它。

所以问题是我HttpServletRequest在 ReqFilter 和控制器类中使用来自两个不同命名空间的类，即一个来自"org.apache.catalina.servlet4preview.http.HttpServletRequest"和另一个来自"javax.servlet.http.HttpServletRequest"。

一旦我在两个文件中使用了统一命名空间，我就可以从控制器类中访问头文件。

于 2020-12-15T07:45:04.860 回答