网关支持黑名单配置

RuoYi

2020-06-13 bb305dfb6fa8f457c9d4797d69a485cedc198081

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

@@ -1,10 +1,9 @@
package com.ruoyi.gateway.filter;



import com.ruoyi.common.core.utils.web.WebUtils;

import com.ruoyi.gateway.config.properties.IgnoreClientProperties;

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;

@@ -27,8 +26,11 @@
    @Autowired

    private ValidateCodeService validateCodeService;



    @Autowired

    private IgnoreClientProperties ignoreClient;

    private static final String BASIC_ = "Basic ";



    private static final String CODE = "code";



    private static final String UUID = "uuid";



    @Override

    public GatewayFilter apply(Object config)

@@ -41,17 +43,18 @@
            {

                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

            {

                // swagger的oauth2.0验证码放行操作

                String[] clientInfos = WebUtils.getClientId(request);

                if (ignoreClient.getClients().contains(clientInfos[0]))

                {

                    return chain.filter(exchange);

                }



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

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

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

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

            }

            catch (Exception e)

            {