spring-webflux - 直接从 Spring Gateway 发送来自预过滤器的错误响应

Question

我正在将 Spring Gateway 与 Spring WebFlux 应用程序一起使用。417在我的情况下，如果在下游调用任何服务之前不满足某些条件，我需要返回 HTTP错误。由于某种原因，它没有返回错误；实际上它没有返回任何东西..请求只是挂起。这是我的代码。

@Component
public class GlobalGatewayPreFilter extends AbstractGatewayFilterFactory<GlobalGatewayPreFilter.Config> {

    public GlobalGatewayPreFilter() {
        super(Config.class);
    }

    @Override
    public GatewayFilter apply(Config config) {
        return (exchange, chain) -> exchange
                .getPrincipal()
                .filter(principal -> principal instanceof UserAuthenticationToken)
                .cast(UserAuthenticationToken.class)
                .filter(token -> (token.getUser().isVerified()))
                .defaultIfEmpty(setErrorResponse(exchange)).flatMap(chain::filter);
    }

    private ServerWebExchange setErrorResponse(ServerWebExchange exchange) {
        exchange.getResponse().setStatusCode(HttpStatus.EXPECTATION_FAILED);
        exchange.getResponse().setComplete();
        return exchange;
    }

    public static class Config {
        private String name;

        public String getName() {
            return this.name;
        }

        public void setName(String name) {
            this.name = name;
        }
    }
}

有没有什么办法解决这一问题？

标签： spring-webfluxspring-cloud-gateway

Answer 1

好的，我能够通过使用以下方法来解决这个问题。我在这里发帖，希望有些人会从中受益。

@Component
public class GlobalGatewayPreFilter extends AbstractGatewayFilterFactory<GlobalGatewayPreFilter.Config> {

    public GlobalGatewayPreFilter() {
        super(Config.class);
    }

    @Override
    public GatewayFilter apply(Config config) {
        return (exchange, chain) -> exchange
                .getPrincipal()
                .filter(principal -> principal instanceof UserAuthenticationToken)
                .cast(UserAuthenticationToken.class)
                .filter(token -> (token.getUser().isVerified()))
                .switchIfEmpty(Mono.defer(() -> setErrorResponse(exchange.getResponse()).setComplete().then(Mono.empty())))
                .flatMap(chain::filter);
    }

    private ServerHttpResponse setErrorResponse(ServerHttpResponse serverHttpResponse) {
        serverHttpResponse.setStatusCode(HttpStatus.EXPECTATION_FAILED);
        return serverHttpResponse;
    }

    public static class Config {
        private String name;

        public String getName() {
            return this.name;
        }

        public void setName(String name) {
            this.name = name;
        }
    }
}