（WebFlux）001、如何自定义注解实现功能

一、背景

最近在项目又在压测，但基于Http请求类型的校验过多，已有想法把Http请求换成Spring中的WebClient，但是由于不是原配（SpringWebFlux + WebClient），如果采用WebClent....block()这样的实现方式，阻塞获取结果，老是觉得别扭，所以就想把SpringMVC换成SpringWebFlux（新手上路），大胆尝试，直接发车。

现在把换的过程中的问题和解决方式列出来，供大家参考。

SpringBoot版本号： <version>2.6.10</version>

二、正文

在使用SpringMVC实现业务逻辑时，我们经常会采用一些自定义注解，通过自定义注解实现访问URL过滤、鉴权等一些列功能。但是在WebFlux中如何实现呢？

2.1 MVC 实现方案

在MVC中，我们一般直接实现

HandlerInterceptor

，然后实现

HandlerInterceptor#preHandle()

方法即可，然后在其中实现自己的逻辑。代码如下、

先实现自定义注解：

/**
 * <p>校验权限</p>
 *
 * @author [email protected]
 * @since 2022/7/24
 */
@Target({ElementType.TYPE, ElementType.METHOD}) // 可用于方法和类上
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface CheckPermission {

 boolean check() default true;
}


为什么要标记在方法和类上呢？ 那是因为有可能某一个类的方法都需要CheckPermission 或者都不check，直接标记在类上就省事了。

接下来是实现拦截器：

/**
 * <p>拦截器</p>
 *
 * @author [email protected]
 * @since 2022/7/24
 */
@Component
public class CheckPermissionInterceptor implements HandlerInterceptor {

 @Override
 public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

 if (handler instanceof HandlerMethod) {
 HandlerMethod methodHandle = (HandlerMethod) handler;
 CheckPermission permission = methodHandle.getMethodAnnotation(CheckPermission.class);
 if (Objects.isNull(permission)) {
 // handler 所有bean 上面寻找
 permission = AnnotationUtils.findAnnotation(methodHandle.getBeanType(), CheckPermission.class);
 }

 if (Objects.nonNull(permission)) {
 // TODO do something..
 }
 }
 return true;
 }
}


当我们在当前Handler上没有找到注解的的时候，我们就去所在类上寻找，判断是否需要校验。偷鸡，减少代码注解标记量（心理美滋滋）。

2.2 WebFlux 实现方案

MVC的实现方式对我们来说可谓是手到擒来，

闭着眼睛也能把代码敲完（吹嘘居多...）

，这么常见、基础、重要的功能，在WebFlux中是如何实现的呢？

我在翻了好多文档后，终于找到了答案，当然要分享给大家了哇，原文采用的kotlin，我就用JAVA了。

原文出处

在WebFlux中没有

HandlerInterceptor

，我们需要采用提供的

WebFilter

实现功能。注解同上，实现的逻辑代码如下。

/**
 * <p>WebFlux实现方案</p>
 *
 * @author [email protected]
 * @since 2022/7/24
 */
@Component
public class CheckPermissionWebFilter implements WebFilter {
 
 @Autowired
 // 关键，通过RequestMappingHandlerMapping 我们可以获取到MethodHandler
 private RequestMappingHandlerMapping handlerMapping;

 @Override
 public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
 return handlerMapping.getHandler(exchange).switchIfEmpty(chain.filter(exchange))
 .flatMap(handler -> {
 // 熟悉的味道，将handler转换成HandlerMethod
 if (handler instanceof HandlerMethod) {
 HandlerMethod methodHandle = (HandlerMethod) handler;
 CheckPermission permission = methodHandle.getMethodAnnotation(CheckPermission.class);
 if (Objects.isNull(permission)) {
 // handler 所有bean 上面寻找
 permission = AnnotationUtils.findAnnotation(methodHandle.getBeanType(), CheckPermission.class);
 }

 if (Objects.nonNull(permission)) {
 // TODO do something..
 }
 }
 return chain.filter(exchange);
 }
 );
 }
}


可以看到，在实现过程中，最关键的是要知道有

org.springframework.web.reactive.result.method.annotation.RequestMappingHandlerMapping

这个类，知道之后，我们就可以通过其获取到Handler。拿到Handler后就可以按照我们自己的套路来实现具体的业务逻辑了。

2.3 对比

通过对比MVC和WebFlux的实现方式来看，Spring对两种都做了比较好的支持，都是通过获取到HandlerMethod，然后在对具体的逻辑处理。对比后发现，代码相似度达到80%~90%。这也要求我们要做的可能是更多的知道其有的API，这样可以减少我们在对项目做转换时带来的时间成本问题。至于实现，我们可以在熟悉使用时，一步步探索，挖掘实现逻辑，提高知识存储量。

三、总结

尝试新东西的时候是一个不断探索，学习的过程。多看，多搜，多学。