微服务下计划一个注解标识是否须要登录

需求

现在我们是微服务体系，须要计划一个注解 @RequiredLogin ，当标识这个注解时表现体系须要登录才气继承操纵。
实现思绪

起首，须要明白我们要拦截的是从欣赏器过来的哀求，服务之间的相互调用是不须要拦截的（比如 Feign 调用）。下图是一些交互环境。
 

我们在网关的全局过滤器中添加一个标识 X（表现他是从欣赏器过来的），然后在每次 Feign 远程调用之前标识一下 Y（表现是内部远程调用），末了在每个服务的拦截器（这个可以抽出通用拦截器设置）中拦截哀求头中的信息举行判断即可。

代码实现

界说注解

2. @Target({ElementType.METHOD})
3. @Retention(RetentionPolicy.RUNTIME)
4. @Documented
5. public @interface RequireLogin {
6. }

复制代码

网关的全局过滤器

2. /**
3. * 定义全局过滤器，功能如下:
4. * 1.把客户端真实IP通过请求同的方式传递给微服务
5. * 2.在请求头中添加FEIGN_REQUEST的请求头，值为0，标记请求不是Feign调用，而是客户端调用
6. * 3.刷新Token的有效时间
7. */
8. @Component
9. public class CommonFilter implements GlobalFilter {
10.     @Autowired
11.     private StringRedisTemplate redisTemplate;
12.     @Override
13.     public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
14.         /**
15.          * pre拦截逻辑
16.          * 在请求去到微服务之前，做了两个处理
17.          * 1.把客户端真实IP通过请求同的方式传递给微服务
18.          * 2.在请求头中添加FEIGN_REQUEST的请求头，值为0，标记请求不是Feign调用，而是客户端调用
19.          */
20.         ServerHttpRequest request = exchange.getRequest().mutate().
21.                 header(CommonConstants.REAL_IP,exchange.getRequest().getRemoteAddress().getHostString()).
22.                 header(CommonConstants.FEIGN_REQUEST_KEY,CommonConstants.FEIGN_REQUEST_FALSE).
23.                 build();
24.         return chain.filter(exchange.mutate().request(request).build()).then(Mono.fromRunnable(()->{
25.             /**
26.              * post拦截逻辑
27.              * 在请求执行完微服务之后,需要刷新token在redis的时间
28.              * 判断token不为空 && Redis还存在这个token对于的key,这时候需要延长Redis中对应key的有效时间.
29.              */
30.             String token,redisKey;
31.             if(!StringUtils.isEmpty(token = exchange.getRequest().getHeaders().getFirst(CommonConstants.TOKEN_NAME))
32.                     && redisTemplate.hasKey(redisKey = CommonRedisKey.USER_TOKEN.getRealKey(token))){
33.                 redisTemplate.expire(redisKey, CommonRedisKey.USER_TOKEN.getExpireTime(), CommonRedisKey.USER_TOKEN.getUnit());
34.             }
35.         }));
36.     }
37. }

复制代码

Feign 远程调用的前置拦截

2. /**
3. * Feign调用时添加标记为此时是通过Feign进行调用（不需要登录）
4. */
5. public class FeignRequestInterceptor implements RequestInterceptor {
6.     @Override
7.     public void apply(RequestTemplate template) {
8.         template.header(CommonConstants.FEIGN_REQUEST_KEY,CommonConstants.FEIGN_REQUEST_TRUE);
9.     }
10. }

复制代码

每个哀求都要走的通用登录拦截器

2. public class RequireLoginInterceptor implements HandlerInterceptor {
3.     private StringRedisTemplate redisTemplate;
4.     public RequireLoginInterceptor(StringRedisTemplate redisTemplate){
5.         this.redisTemplate = redisTemplate;
6.     }
7.     @Override
8.     public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
9.         if(handler instanceof HandlerMethod){
10.             HandlerMethod handlerMethod = (HandlerMethod) handler;
11.             String feignRequest = request.getHeader(CommonConstants.FEIGN_REQUEST_KEY);
12.             //如果是feign请求，直接放行
13.             if(!StringUtils.isEmpty(feignRequest) && CommonConstants.FEIGN_REQUEST_TRUE.equals(feignRequest)){
14.                 return true;
15.             }
16.             //如果不是Feign请求，判断是否有贴RequireLogin注解
17.             if(handlerMethod.getMethodAnnotation(RequireLogin.class)!=null){
18.                 response.setContentType("application/json;charset=utf-8");
19.                 String token = request.getHeader(CommonConstants.TOKEN_NAME);
20.                 if(StringUtils.isEmpty(token)){
21.                     response.getWriter().write(JSON.toJSONString(Result.error(CommonCodeMsg.TOKEN_INVALID)));
22.                     return false;
23.                 }
24.                 String phone = JSON.parseObject(redisTemplate.opsForValue().get(CommonRedisKey.USER_TOKEN.getRealKey(token)),String.class);
25.                 if(phone==null){
26.                     response.getWriter().write(JSON.toJSONString(Result.error(CommonCodeMsg.TOKEN_INVALID)));
27.                     return false;
28.                 }
29.             }
30.         }
31.         return true;
32.     }
33. }

复制代码