以编程方式控制 @RestController 可用性

Question

是否可以通过编程方式控制@RestController 以启用或禁用它？我不想只在每个@RequestMapping 方法中编写代码来做某种if (!enabled) { return 404Exception; }

我见过this question，但这仅在启动时有效。我真正需要的是允许我多次启用或禁用控制器的东西。

我想过不同的方法，但不知道哪些方法在春天可行。

1. 实际上控制容器（在我的例子中是码头），因此对特定端点的请求被禁用
2. 不知何故控制RequestMappingHandlerMapping，因为它似乎是那个在url和控制器之间进行映射的类
3. 控制@RestController组件的生命周期，以便我可以随意创建和销毁它，但是我不确定如何触发到端点的映射

Answer 1

如果最终结果是当您决定应该禁用特定端点时想要以 404 响应，那么您可以编写一个拦截器来检查您的启用条件是否为假，如果是，则相应地设置响应。

例如：

@Component
public class ConditionalRejectionInterceptor extends HandlerInterceptorAdapter {

    @Override
    public boolean preHandle(HttpServletRequest request,
            HttpServletResponse response, Object handler) throws Exception {
        String requestUri = request.getRequestURI();
        if (shouldReject(requestUri)) {
            response.setStatus(HttpStatus.NOT_FOUND.value());
            return false;
        }
        return super.preHandle(request, response, handler);
    }

    private boolean shouldReject(String requestUri) {
        // presumably you have some mechanism of inferring or discovering whether 
        // the endpoint represented by requestUri should be allowed or disallowed
        return ...;
    }
}

在 Spring Boot 中，注册自己的拦截器只涉及实现 WebMvcConfigurerAdapter。例如：

@Configuration
public class CustomWebMvcConfigurer extends WebMvcConfigurerAdapter {

  @Autowired 
  private HandlerInterceptor conditionalRejectionInterceptor;

  @Override
  public void addInterceptors(InterceptorRegistry registry) {
    // you can use .addPathPatterns(...) here to limit this interceptor to specific endpoints
    // this could be used to replace any 'conditional on the value of requestUri' code in the interceptor
    registry.addInterceptor(conditionalRejectionInterceptor);
  }
}