Возврат GZip-ответа с помощью Spring Cloud Gateway зависает

Я использую Spring Cloud Gateway для пересылки запросов в серверную службу. Поскольку некоторые из наших внутренних сервисов отправляют свои результаты в сжатом виде, определение глобального фильтра для разархивирования перед обработкой в ​​маршрутах и ​​повторного сжатия в формате gzip перед отправкой клиенту кажется хорошей идеей. Таким образом, нет необходимости делать это для каждого маршрута. Использование различных идей, которые есть в Stack Overflow, привело к следующему определению gzip результата этапов обработки.

@Component
public class GlobalGZipFilter implements GlobalFilter, Ordered {

    private static final Logger log = LoggerFactory.getLogger(GlobalGZipFilter.class);

    @Override
    public int getOrder() {
        return -2; 
    }



    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {

        ServerHttpResponse originalResponse = exchange.getResponse();

        DataBufferFactory bufferFactory = originalResponse.bufferFactory();
        ServerHttpResponseDecorator decoratedResponse = new ServerHttpResponseDecorator(originalResponse) {

            @Override
            public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
                log.info("Content should be GZipped: {}", isGZipped(originalResponse));
                if (isGZipped(originalResponse) && body instanceof Flux) {
                    Flux<? extends DataBuffer> flux = (Flux<? extends DataBuffer>) body;

                    return super.writeWith(flux.buffer().map(dataBuffers -> {
                        ByteOutputStream outputStream = new ByteOutputStream();
                        for (DataBuffer i : dataBuffers) {
                            byte[] array = new byte[i.readableByteCount()];
                            i.read(array);
                            outputStream.write(array);
                        }

                        String resultResponse = new String(outputStream.getBytes());

                        return bufferFactory.wrap(zipString(resultResponse));
                    }));
                }

                return super.writeWith(body); // if body is not a flux. never got there.
            }
        };

        return chain
                .filter(exchange
                        .mutate()
                        .response(decoratedResponse)
                        .build()); // replace response with decorator
    }

Процесс состоит в том, чтобы собрать буферы, содержащие результат, преобразовать его в один полный byte[] и сжать этот byte[] с помощью gzip.

Кажется, что результат обработан правильно, но только после остановки приложения результат отображается в клиенте. Почему-то транзакция не заканчивается до закрытия. Что-то нужно для подтверждения результата или текущий выбранный поток просто неверен?


spring-cloud-gateway
person Wieki    schedule 05.02.2019    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Проблема не столько в обработке потока или данных. Я забыл учесть, что при сжатии возвращаемого содержимого длина ответа будет другой. Добавление длины содержимого на основе содержимого, сжатого с помощью gzip, поможет:

  String resultResponse = new String(outputStream.getBytes());
                        byte[] zippedResponse = zipString(resultResponse);

                        originalResponse.getHeaders().setContentLength(zippedResponse.length);

                        return bufferFactory.wrap(zippedResponse);
person Wieki    schedule 06.02.2019