Я использую Spring Gateway с приложением Spring WebFlux. В моем случае мне нужно вернуть ошибку HTTP 417, если определенные критерии не выполняются, перед вызовом какой-либо службы ниже по течению. По какой-то причине он не возвращает ошибку; по сути ничего не возвращает .. запрос просто висит. Это мой код.

@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;
        }
    }
}

Есть ли способ исправить это?

0
nicholasnet 4 Дек 2020 в 01:45

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;
        }
    }
}

0
nicholasnet 4 Дек 2020 в 22:23