OAuth自动刷新续签Token

RuoYi

2020-08-01 43bc0ca39be664ffaeea6256a9f2160f5a8f7e40

 ruoyi-gateway/src/main/java/com/ruoyi/gateway/filter/ValidateCodeFilter.java

@@ -3,6 +3,7 @@
import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.cloud.gateway.filter.GatewayFilter;

import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;

import org.springframework.http.HttpHeaders;

import org.springframework.http.server.reactive.ServerHttpRequest;

import org.springframework.http.server.reactive.ServerHttpResponse;

import org.springframework.stereotype.Component;

@@ -25,6 +26,16 @@
    @Autowired

    private ValidateCodeService validateCodeService;



    private static final String BASIC_ = "Basic ";



    private static final String CODE = "code";



    private static final String UUID = "uuid";

    
    private static final String GRANT_TYPE = "grant_type";

    
    private static final String REFRESH_TOKEN = "refresh_token";



    @Override

    public GatewayFilter apply(Object config)

    {

@@ -36,14 +47,30 @@
            {

                return chain.filter(exchange);

            }

            
            // 刷新token请求，不处理

            String grantType = request.getQueryParams().getFirst(GRANT_TYPE);

            if (StringUtils.containsIgnoreCase(request.getURI().getPath(), AUTH_URL) && StringUtils.containsIgnoreCase(grantType, REFRESH_TOKEN))

            {

                return chain.filter(exchange);

            }



            // 消息头存在内容，且不存在验证码参数，不处理

            String header = request.getHeaders().getFirst(HttpHeaders.AUTHORIZATION);

            if (StringUtils.isNotEmpty(header) && StringUtils.startsWith(header, BASIC_)

                    && !request.getQueryParams().containsKey(CODE) && !request.getQueryParams().containsKey(UUID))

            {

                return chain.filter(exchange);

            }

            try

            {

                validateCodeService.checkCapcha(request.getQueryParams().getFirst("code"),

                        request.getQueryParams().getFirst("uuid"));

                validateCodeService.checkCapcha(request.getQueryParams().getFirst(CODE),

                        request.getQueryParams().getFirst(UUID));

            }

            catch (Exception e)

            {

                ServerHttpResponse response = exchange.getResponse();

                response.getHeaders().add("Content-Type", "application/json;charset=UTF-8");

                return exchange.getResponse().writeWith(

                        Mono.just(response.bufferFactory().wrap(JSON.toJSONBytes(AjaxResult.error(e.getMessage())))));

            }