SpringCloud Webflux過濾器增加header傳遞方式

更新時間：2023年02月14日 09:54:46 作者：低頭確望水中月亮

這篇文章主要介紹了SpringCloud Webflux過濾器增加header傳遞方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

Webflux過濾器增加header傳遞

webflux 使用異步非阻塞的方式，傳遞Header還需要我們自己增加代碼傳遞：

定義過濾器，便攜下面的配置可以嘗試一下header的傳遞到后面的服務中。

@Component
public class AuthorizationFilter implements WebFilter {
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
        ServerHttpRequest mutatedRequest = exchange.getRequest().mutate().header(HttpHeaders.AUTHORIZATION, "Bearer " + authHeader).build();
        ServerWebExchange mutatedExchange = exchange.mutate().request(mutatedRequest).build();
        return chain.filter(mutatedExchange);
    }
}
 
 
## Kotlin 的方式
@Component
class AuthorizationFilter : WebFilter {
    override fun filter(exchange: ServerWebExchange, chain: WebFilterChain): Mono<Void> {
        val mutatedRequest = exchange.request.mutate().header(HttpHeaders.AUTHORIZATION, "Bearer $authHeader").build()
        val mutatedExchange = exchange.mutate().request(mutatedRequest).build()
        return chain.filter(mutatedExchange)
    }
}

Webflux過濾器使用

下面是簡單處理token的例子

@Component
@Order(value = 2)
public class AuthFilter implements WebFilter {
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
        ServerHttpRequest request =  exchange.getRequest();
        //ServerHttpResponse response =  exchange.getResponse();

        String token = request.getHeaders().getFirst("token");
        if(null == token){
                /*http頭中沒有appKey，修改請求的目標url為/auth/error
                  request.mutate返回一個請求構建器(builder design pattern)，path方法修改請求的url，build方法返回新的request
               */
            ServerHttpRequest authErrorReq = request.mutate().path("/auth/error").build();
            //erverWebExchange.mutate類似，構建一個新的ServerWebExchange
            ServerWebExchange authErrorExchange = exchange.mutate().request(authErrorReq).build();
            return chain.filter(authErrorExchange);
        } else{ 
            return chain.filter(exchange);
        }
    }
}

webFlux將指定的錯誤信息通過json格式返回給前端

exchange.getResponse().writeWith(Flux.just(exchange.getResponse().bufferFactory().wrap(JsonUtil.toJson(new BaseErrorVO().fail(e.getMessageCode(),e.getSystemMessage())).getBytes())));