Gateway網(wǎng)關(guān)自定義攔截器不可重復(fù)讀取數(shù)據(jù)怎么解決

本篇內(nèi)容主要講解“Gateway網(wǎng)關(guān)自定義攔截器不可重復(fù)讀取數(shù)據(jù)怎么解決”，感興趣的朋友不妨來(lái)看看。本文介紹的方法操作簡(jiǎn)單快捷，實(shí)用性強(qiáng)。下面就讓小編來(lái)帶大家學(xué)習(xí)“Gateway網(wǎng)關(guān)自定義攔截器不可重復(fù)讀取數(shù)據(jù)怎么解決”吧!

Gateway網(wǎng)關(guān)自定義攔截器的不可重復(fù)讀取數(shù)據(jù)

最近在開(kāi)發(fā)gateway網(wǎng)關(guān)時(shí)，通過(guò)自定義攔截器對(duì)某些接口的數(shù)據(jù)進(jìn)行處理，發(fā)現(xiàn)，無(wú)法讀取到數(shù)據(jù)。經(jīng)過(guò)查詢，發(fā)現(xiàn)在Spring5的webflux編程或者普通web編程中，只能從request中獲取body一次，后面無(wú)法再獲取。

參考網(wǎng)上的方法先通過(guò)全局過(guò)濾器把body先緩存起來(lái)。

這個(gè)網(wǎng)上有很多例子：

/**
 * @author zhong
 * @date 2022/1/6 - 15:34
 * 
 */
@Configuration
public class CacheBodyGlobalFilter implements Ordered, GlobalFilter {
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        //判斷是否時(shí)認(rèn)證接口
        if (!AuthUtils.isAuthorPath(exchange.getRequest())) {
            //不是認(rèn)證接口
            return chain.filter(exchange);
        } else {
            //指定接口可建數(shù)據(jù)從新讀出構(gòu)建
            return DataBufferUtils.join(exchange.getRequest().getBody())
                    .flatMap(dataBuffer -> {
                        DataBufferUtils.retain(dataBuffer);
                        Flux<DataBuffer> cachedFlux = Flux
                                .defer(() -> Flux.just(dataBuffer.slice(0, dataBuffer.readableByteCount())));
                        ServerHttpRequest mutatedRequest = new ServerHttpRequestDecorator(exchange.getRequest()) {
                            @Override
                            public Flux<DataBuffer> getBody() {
                                return cachedFlux;
                            }
                        };
                        //exchange.getAttributes().put(CACHE_REQUEST_BODY_OBJECT_KEY, cachedFlux);
                        return chain.filter(exchange.mutate().request(mutatedRequest).build());
                    });
        }
    }
 
    @Override
    public int getOrder() {
        return Ordered.HIGHEST_PRECEDENCE;
    }
}

CacheBodyGlobalFilter這個(gè)全局過(guò)濾器的目的就是把原有的request請(qǐng)求中的body內(nèi)容讀出來(lái)，并且使用ServerHttpRequestDecorator這個(gè)請(qǐng)求裝飾器對(duì)request進(jìn)行包裝，重寫(xiě)getBody方法，并把包裝后的請(qǐng)求放到過(guò)濾器鏈中傳遞下去。這樣后面的過(guò)濾器中再使用exchange.getRequest().getBody()來(lái)獲取body時(shí)，實(shí)際上就是調(diào)用的重載后的getBody方法，獲取的最先已經(jīng)緩存了的body數(shù)據(jù)。這樣就能夠?qū)崿F(xiàn)body的多次讀取了。

這個(gè)過(guò)濾器的order設(shè)置的是Ordered.HIGHEST_PRECEDENCE，即最高優(yōu)先級(jí)的過(guò)濾器。優(yōu)先級(jí)設(shè)置這么高的原因是某些系統(tǒng)內(nèi)置的過(guò)濾器可能也會(huì)去讀body。

但是在測(cè)試中我發(fā)現(xiàn)，如果我在請(qǐng)求接口中如果沒(méi)有body內(nèi)容，就會(huì)導(dǎo)致程序只能執(zhí)行CacheBodyGlobalFilter這一個(gè)攔截器，而無(wú)法執(zhí)行其他攔截器(自定義的和默認(rèn)的)。而且接口返回200，這和我的預(yù)期時(shí)不一致的。

通過(guò)測(cè)試發(fā)現(xiàn)，原因是，按照以上代碼執(zhí)行，如果是body為null，Conten-Type也為空，所以沒(méi)有執(zhí)行代碼中的flatMap()方法.所以也就沒(méi)有執(zhí)行后面的調(diào)用鏈。

chain.filter(exchange.mutate().request(mutatedRequest).build());

解決辦法，在獲取到數(shù)據(jù)流時(shí)，如果數(shù)據(jù)流為null，我們可以構(gòu)建一個(gè)空的數(shù)據(jù)流，這也也就能執(zhí)行我們后面的攔截器。

利用Flux.defaultIfEmpty(defaultDataBuffer);的方法可以實(shí)現(xiàn)這個(gè)功能。

public final Flux<T> defaultIfEmpty(T defaultV) {
   return onAssembly(new FluxDefaultIfEmpty<>(this, defaultV));
}

代碼如下： 

@Configuration
public class CacheBodyGlobalFilter implements Ordered, GlobalFilter {
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        //判斷是否時(shí)認(rèn)證接口
        if (!AuthUtils.isAuthorPath(exchange.getRequest())) {
            //不是認(rèn)證接口
            return chain.filter(exchange);
        } else {
            //指定接口可建數(shù)據(jù)從新讀出構(gòu)建
 
            //當(dāng)body中沒(méi)有緩存時(shí)，只會(huì)執(zhí)行這一個(gè)攔截器， 原因是fileMap中的代碼沒(méi)有執(zhí)行，所以需要在波多野為空時(shí)構(gòu)建一個(gè)空的緩存
            DefaultDataBufferFactory defaultDataBufferFactory = new DefaultDataBufferFactory();
            DefaultDataBuffer defaultDataBuffer = defaultDataBufferFactory.allocateBuffer(0);
            //構(gòu)建新數(shù)據(jù)流， 當(dāng)body為空時(shí)，構(gòu)建空流
            Flux<DataBuffer> bodyDataBuffer = exchange.getRequest().getBody().defaultIfEmpty(defaultDataBuffer);
            return DataBufferUtils.join(bodyDataBuffer)
                    .flatMap(dataBuffer -> {
                        DataBufferUtils.retain(dataBuffer);
                        Flux<DataBuffer> cachedFlux = Flux
                                .defer(() -> Flux.just(dataBuffer.slice(0, dataBuffer.readableByteCount())));
                        ServerHttpRequest mutatedRequest = new ServerHttpRequestDecorator(exchange.getRequest()) {
                            @Override
                            public Flux<DataBuffer> getBody() {
                                return cachedFlux;
                            }
                        };
                        //exchange.getAttributes().put(CACHE_REQUEST_BODY_OBJECT_KEY, cachedFlux);
                        return chain.filter(exchange.mutate().request(mutatedRequest).build());
                    });
        }
    }
 
    @Override
    public int getOrder() {
        return Ordered.HIGHEST_PRECEDENCE;
    }
}

當(dāng)然這個(gè)是在我們的攔截的接口存在沒(méi)有body，但是也要執(zhí)行后面的攔截器的情況下。如果說(shuō)接口必須要有body。那我們可以在CacheBodyGlobalFilter 這攔截器中直接返回指定的內(nèi)容。這個(gè)我還沒(méi)有測(cè)試過(guò)。

統(tǒng)一網(wǎng)關(guān)Gateway

一、為什么需要網(wǎng)關(guān)

網(wǎng)關(guān)功能：

• 身份認(rèn)證和權(quán)限校驗(yàn)

• 服務(wù)路由、負(fù)載均衡

• 請(qǐng)求限流

二、搭建網(wǎng)關(guān)服務(wù)

搭建網(wǎng)關(guān)服務(wù)的步驟：

1、創(chuàng)建新的module,引入SpringCloudGateway的依賴和nacos的服務(wù)發(fā)現(xiàn)依賴

<!--網(wǎng)關(guān)依賴-->
<dependency>
    <groupId>org.springframework.cloud</groupId>
    <artifactId>spring-cloud-starter-gateway</artifactId>
</dependency>
<!--nacos服務(wù)發(fā)現(xiàn)依賴-->
<dependency>
    <groupId>com.alibaba.cloud</groupId>
    <artifactId>spring-cloud-starter-alibaba-nacos-discovery</artifactId> 
</dependency>

2、編寫(xiě)路由配置及nacos地址

server:
  port: 10010 # 網(wǎng)關(guān)端口
spring:
  application:
    name: gateway # 服務(wù)名稱
  cloud:
    nacos:
      server-addr: localhost:8848 # nacos地址
    gateway:
      routes: # 網(wǎng)關(guān)路由配置
        - id: user-service # 路由id，自定義，只要唯一即可
          # uri: http://127.0.0.1:8081 # 路由的目標(biāo)地址 http就是固定地址
          uri: lb://userservice # 路由的目標(biāo)地址 lb就是負(fù)載均衡，后面跟服務(wù)名稱
          predicates: # 路由斷言，也就是判斷請(qǐng)求是否符合路由規(guī)則的條件
            - Path=/user/** # 這個(gè)是按照路徑匹配，只要以/user/開(kāi)頭就符合要求

3、搭建步驟總結(jié)：

創(chuàng)建項(xiàng)目，引入nacos服務(wù)發(fā)現(xiàn)和gateway依賴

配置application.yml，包括服務(wù)基本信息、nacos地址、路由

路由配置包括：

• 路由id：路由的唯一標(biāo)示

• 路由目標(biāo)（uri）：路由的目標(biāo)地址，http代表固定地址，lb代表根據(jù)服務(wù)名負(fù)載均衡

• 路由斷言（predicates）：判斷路由的規(guī)則，

• 路由過(guò)濾器（filters）：對(duì)請(qǐng)求或響應(yīng)做處理

Gateway執(zhí)行流程：

三、路由斷言工廠

上文中，在配置文件中寫(xiě)的斷言規(guī)則只是字符串，這些字符串會(huì)被Predicate Factory讀取并處理，轉(zhuǎn)變?yōu)槁酚膳袛嗟臈l件！例如Path=/user/**是按照路徑匹配，這個(gè)規(guī)則是由org.springframework.cloud.gateway.handler.predicate.PathRoutePredicateFactory類來(lái)處理的，像這樣的斷言工廠在SpringCloudGateway還有十幾個(gè)

Spring提供了11種基本的Predicate工廠：

四、路由過(guò)濾器

路由過(guò)濾器 GatewayFilter

GatewayFilter是網(wǎng)關(guān)中提供的一種過(guò)濾器，可以對(duì)進(jìn)入網(wǎng)關(guān)的請(qǐng)求和微服務(wù)返回的響應(yīng)做處理：

過(guò)濾器工廠 GatewayFilterFactory

Spring提供了31種不同的路由過(guò)濾器工廠。例如：

過(guò)濾器工廠案例：給所有進(jìn)入userservice的請(qǐng)求添加一個(gè)請(qǐng)求頭

給所有進(jìn)入userservice的請(qǐng)求添加一個(gè)請(qǐng)求頭：Truth=xiaobai is headsome

spring:
 cloud:
   gateway:
     routes: # 網(wǎng)關(guān)路由配置
       - id: user-service
         uri: lb://userservice
         predicates:
           - Path=/user/**
         filters: # 過(guò)濾器
           - AddRequestHeader=Truth, xiaobai is headsome # 添加請(qǐng)求頭

默認(rèn)過(guò)濾器

如果要對(duì)所有的路由都生效，則可以將過(guò)濾器工廠寫(xiě)到default(默認(rèn)過(guò)濾器)下。格式如下：

spring:
  application:
    name: gateway # 服務(wù)名稱
  cloud:
    nacos:
      server-addr: localhost:8848 # nacos地址
    gateway:
      routes: # 網(wǎng)關(guān)路由配置
        - id: user-service 
          uri: lb://userservice  
          predicates: 
            - Path=/user/** 
        - id: order-service
          uri: lb://orderservice
          predicates:
            - Path=/order/**
        default-filters: # 默認(rèn)過(guò)濾器，會(huì)對(duì)所有的路由請(qǐng)求都生效
          - AddRequestHeader=Truth, xiaobai is headsome # 添加請(qǐng)求頭

全局過(guò)濾器 GlobalFilter

全局過(guò)濾器的作用也是處理一切進(jìn)入網(wǎng)關(guān)的請(qǐng)求和微服務(wù)響應(yīng)，與GatewayFilter的作用一樣。

區(qū)別在于GatewayFilter通過(guò)配置定義，處理邏輯是固定的。而GlobalFilter的邏輯需要自己寫(xiě)代碼實(shí)現(xiàn)。

定義方式是實(shí)現(xiàn)GlobalFilter接口。

public interface GlobalFilter {
   /**
    *  處理當(dāng)前請(qǐng)求，有必要的話通過(guò){@link GatewayFilterChain}將請(qǐng)求交給下一個(gè)過(guò)濾器處理
    *
    * @param exchange 請(qǐng)求上下文，里面可以獲取Request、Response等信息
    * @param chain 用來(lái)把請(qǐng)求委托給下一個(gè)過(guò)濾器 
    * @return {@code Mono<Void>} 返回標(biāo)示當(dāng)前過(guò)濾器業(yè)務(wù)結(jié)束
    */
        Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain);
     }

全局過(guò)濾器案例：定義全局過(guò)濾器，攔截并判斷用戶身份 

需求：

定義全局過(guò)濾器，攔截請(qǐng)求，判斷請(qǐng)求的參數(shù)是否滿足下面條件：

參數(shù)中是否有authorization，authorization參數(shù)值是否為admin

如果同時(shí)滿足則放行，否則攔截

自定義類，實(shí)現(xiàn)GlobalFilter接口，添加@Order注解

@Order(-1)
@Component
public class AuthorizeFilter implements GlobalFilter {
    @Override
        public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
         // 1.獲取請(qǐng)求參數(shù)
            MultiValueMap<String, String> params = exchange.getRequest().getQueryParams();
        // 2.獲取authorization參數(shù)
            String auth = params.getFirst("authorization");
        // 3.校驗(yàn)
            if ("admin".equals(auth)) {
        // 放行
            return chain.filter(exchange);
  }
       // 4.攔截
       // 4.1.禁止訪問(wèn)                                                                                                
            exchange.getResponse().setStatusCode(HttpStatus.FORBIDDEN);
       // 4.2.結(jié)束處理
            return exchange.getResponse().setComplete();                       
   }
}

過(guò)濾器執(zhí)行順序

請(qǐng)求進(jìn)入網(wǎng)關(guān)會(huì)碰到三類過(guò)濾器：當(dāng)前路由的過(guò)濾器、DefaultFilter、GlobalFilter

請(qǐng)求路由后，會(huì)將當(dāng)前路由過(guò)濾器和DefaultFilter、GlobalFilter，合并到一個(gè)過(guò)濾器鏈（集合）中，排序后依次執(zhí)行每個(gè)過(guò)濾器

• 每一個(gè)過(guò)濾器都必須指定一個(gè)int類型的order值，order值越小，優(yōu)先級(jí)越高，執(zhí)行順序越靠前。

• GlobalFilter通過(guò)實(shí)現(xiàn)Ordered接口，或者添加@Order注解來(lái)指定order值，由我們自己指定

• 路由過(guò)濾器和defaultFilter的order由Spring指定，默認(rèn)是按照聲明順序從1遞增。

• 當(dāng)過(guò)濾器的order值一樣時(shí)，會(huì)按照 defaultFilter > 路由過(guò)濾器 > GlobalFilter的順序執(zhí)行。

五、跨域問(wèn)題處理

跨域：域名不一致就是跨域，主要包括：

• 域名不同： www.taobao.com 和 www.taobao.org 和 www.jd.com 和 miaosha.jd.com

• 域名相同，端口不同：localhost:8080和localhost8081

跨域問(wèn)題：瀏覽器禁止請(qǐng)求的發(fā)起者與服務(wù)端發(fā)生跨域ajax請(qǐng)求，請(qǐng)求被瀏覽器攔截的問(wèn)題

解決方案：CORS

網(wǎng)關(guān)處理跨域采用的同樣是CORS方案，并且只需要簡(jiǎn)單配置即可實(shí)現(xiàn)：

spring:
  cloud:
    gateway:
      globalcors: # 全局的跨域處理
        add-to-simple-url-handler-mapping: true # 解決options請(qǐng)求被攔截問(wèn)題
        corsConfigurations:
          '[/**]':
	        allowedOrigins: # 允許哪些網(wǎng)站的跨域請(qǐng)求 
	          - "http://localhost:8090"
	          - "http://www.leyou.com"
	        allowedMethods: # 允許的跨域ajax的請(qǐng)求方式
	          - "GET"
	          - "POST"
	          - "DELETE"
	          - "PUT"
	          - "OPTIONS"
	        allowedHeaders: "*" # 允許在請(qǐng)求中攜帶的頭信息
	        allowCredentials: true # 是否允許攜帶cookie
	        maxAge: 360000 # 這次跨域檢測(cè)的有效期

到此，相信大家對(duì)“Gateway網(wǎng)關(guān)自定義攔截器不可重復(fù)讀取數(shù)據(jù)怎么解決”有了更深的了解，不妨來(lái)實(shí)際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！