Spring boot HandlerInterceptor 负载均衡

Question

我正在使用 Spring Boot 实现（某种）负载平衡 HandlerInterceptor。

public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
    String uri = request.getRequestURI();
    if (shouldUseServer1(uri)) {
        response.sendRedirect(server1Uri);
    } else {
        response.sendRedirect(server2Uri);
    }
}

我们的想法是，基于 url，我们要么重定向到一个服务，要么重定向到另一个服务。该应用程序没有任何明确的RequestMappings（目前）。

现在的问题是，当调用拦截器时，请求被重定向到默认的 Spring 错误处理程序。因此，HttpServletRequest 中存储的 URI 被 /error 替换（有效地拒绝了对原始 URI 的访问）。

有没有办法在请求被重新路由到错误处理程序（或获取原始 uri）之前拦截请求？

Answer 1

编辑：

由于 Spring MVC 处理没有映射的请求的方式，您要么需要一个过滤器：

@Component
public class CustomFilter implements Filter {

    public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain)
            throws IOException, ServletException {

        HttpServletRequest request = (HttpServletRequest) req;
        HttpServletResponse response = (HttpServletResponse) res;

        request.getSession().setAttribute("ORIGINAL_REQUEST_URI", request.getRequestURI());
        chain.doFilter(request, response);

        // alternatively, ignore the last 2 lines
        // and just do your redirects from here 
        // and don't continue the filter chain
    }

  @Override
  public void destroy() {}

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

}

否则，如果您不想依赖会话 you'll need to make the DispatcherServlet throw an exception in case no handler mapping is found，然后从 @ControllerAdvice 错误处理程序发送重定向：

@ControllerAdvice
class NoHandlerFoundExceptionExceptionHandler {

  @ExceptionHandler(value = NoHandlerFoundException.class)
  public ModelAndView
  defaultErrorHandler(HttpServletRequest req, NoHandlerFoundException e) throws Exception {
    String uri = // resolve the URI
    return new ModelAndView("redirect:" + uri);
  }
}

为避免重复，您可能希望有一个公共类，您将从拦截器和错误处理程序中调用它。