Как zip-сжать HTTP-запрос с помощью Spring RestTemplate?

как gzip HTTP запрос, создано org.springframework.web.client.RestTemplate?

Я использую Spring 4.2.6 С весны Boot 1.3.5 (Java SE, а не Android или Javascript в веб-браузере).

Я делаю некоторые действительно крупные!--3--> запросы, и я хочу!--5-->запроса для сжатия.

3 ответов


я предлагаю два решения, один проще без потоковой передачи и один, который поддерживает потоковую передачу.

если вы не требуется потоковая передача используйте пользовательский ClientHttpRequestInterceptor, функция Spring.

RestTemplate rt = new RestTemplate();
rt.setInterceptors(Collections.singletonList(interceptor));

здесь interceptor можно:

ClientHttpRequestInterceptor interceptor = new ClientHttpRequestInterceptor() {

    @Override
    public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution)
            throws IOException {
        request.getHeaders().add("Content-Encoding", "gzip");
        byte[] gzipped = getGzip(body);
        return execution.execute(request, gzipped);
    } 
 }

getGzip Я скопировал

    private byte[] getGzip(byte[] body) throws IOException {

        ByteArrayOutputStream byteStream = new ByteArrayOutputStream();
        try {
            GZIPOutputStream zipStream = new GZIPOutputStream(byteStream);
            try {
                zipStream.write(body);
            } finally {
                zipStream.close();
            }
        } finally {
            byteStream.close();
        }

        byte[] compressedData = byteStream.toByteArray();
        return compressedData;

    }

после настройки перехватчика все запросы будут сжаты.

недостатком этого подхода является что он не поддерживает потоковую передачу как ClientHttpRequestInterceptor получает содержимое как byte[]

если вы требуется потоковая передача создать пользовательское ClientHttpRequestFactory, сказал GZipClientHttpRequestFactory, и используйте его так:

    SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
    requestFactory.setBufferRequestBody(false);
    ClientHttpRequestFactory gzipRequestFactory = new GZipClientHttpRequestFactory(requestFactory);
    RestTemplate rt = new RestTemplate(gzipRequestFactory);

здесь GZipClientHttpRequestFactory - это:

public class GZipClientHttpRequestFactory extends AbstractClientHttpRequestFactoryWrapper {

    public GZipClientHttpRequestFactory(ClientHttpRequestFactory requestFactory) {
        super(requestFactory);
    }

    @Override
    protected ClientHttpRequest createRequest(URI uri, HttpMethod httpMethod, ClientHttpRequestFactory requestFactory)
            throws IOException {
        ClientHttpRequest delegate = requestFactory.createRequest(uri, httpMethod);
        return new ZippedClientHttpRequest(delegate);
    }

}

и ZippedClientHttpRequest - это:

public class ZippedClientHttpRequest extends WrapperClientHttpRequest
{
    private GZIPOutputStream zip;

    public ZippedClientHttpRequest(ClientHttpRequest delegate) {
        super(delegate);
        delegate.getHeaders().add("Content-Encoding", "gzip");
        // here or in getBody could add content-length to avoid chunking
        // but is it available ? 
        // delegate.getHeaders().add("Content-Length", "39");

    }

    @Override
    public OutputStream getBody() throws IOException {
        final OutputStream body = super.getBody();
        zip = new GZIPOutputStream(body);
        return zip;
    }

    @Override
    public ClientHttpResponse execute() throws IOException {
        if (zip!=null) zip.close();
        return super.execute();
    }

}

и наконец WrapperClientHttpRequest - это:

public class WrapperClientHttpRequest implements ClientHttpRequest {

    private final ClientHttpRequest delegate;

    protected WrapperClientHttpRequest(ClientHttpRequest delegate) {
        super();
        if (delegate==null)
            throw new IllegalArgumentException("null delegate");
        this.delegate = delegate;
    }

    protected final ClientHttpRequest getDelegate() {
        return delegate;
    }

    @Override
    public OutputStream getBody() throws IOException {
        return delegate.getBody();
    }

    @Override
    public HttpHeaders getHeaders() {
        return delegate.getHeaders();
    }

    @Override
    public URI getURI() {
        return delegate.getURI();
    }

    @Override
    public HttpMethod getMethod() {
        return delegate.getMethod();
    }

    @Override
    public ClientHttpResponse execute() throws IOException {
        return delegate.execute();
    }
}

этот подход создает запрос с поблочного кодирования, это можно изменить настройка заголовка content length, если известен размер.

преимущество ClientHttpRequestInterceptor и/или на заказ ClientHttpRequestFactory подход заключается в том, что он работает с любым методом RestTemplate. Альтернативный подход, проходящий RequestCallback возможно только с execute методы, это потому, что другие методы RestTemplate внутренне создают свои собственные RequestCallback(ы), которые производят содержимое.

кстати, кажется, что нет мало поддержка распаковки запроса gzip на сервере. Также связано:отправка данных gzipped в WebRequest? к Zip Бомба вопрос. Я думаю, вам придется написать код для него.


основная идея-создать requestCallback, который будет копировать данные, которые вы хотите отправить из gzipOutputStream на request поток.

RequestCallback requestCallback = new RequestCallback() {
    @Override
    public void doWithRequest(ClientHttpRequest request) throws IOException {
        GZIPOutputStream gzipOutputStream;
        try {
            gzipOutputStream = new GZIPOutputStream(request.getBody());
        } catch (IOException ignored) {
            return;
        }

        request.getHeaders().add("Content-Type", "application/octet-stream");
        request.getHeaders().add("Content-Encoding", "gzip");

        try {
            String data = "Test data.";
            gzipOutputStream.write(data.getBytes(StandardCharsets.UTF_8));
            gzipOutputStream.flush(); // Optional in this example.
            gzipOutputStream.finish();
        } catch (IOException ignored) {
        }
    }
};

теперь вы можете использовать его следующим образом:

RestTemplate restTemplate = new RestTemplate();
SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
requestFactory.setBufferRequestBody(false);
restTemplate.setRequestFactory(requestFactory);

ResponseExtractor<String> responseExtractor = new HttpMessageConverterExtractor<>(String.class,
        restTemplate.getMessageConverters());
String response = restTemplate.execute("http://localhost:8080/gzip.php", HttpMethod.POST, requestCallback,
        responseExtractor);

System.out.println(response);

ссылки:

  1. простой пример сжатия HTTP-запроса через gzip с использованием Spring RestTemplate
  2. как пересылать большие файлы с помощью RestTemplate?

далее к приведенному выше ответу от @TestoTestini, если мы воспользуемся синтаксисом Java 7+ ' s 'try-with-resources', так как оба ByteArrayOutputStream и GZIPOutputStream реализуйте closeable (), мы можем сжать функцию getGzip в следующее:

private byte[] getGzip(byte[] body) throws IOException {

    try (ByteArrayOutputStream byteStream = new ByteArrayOutputStream()) {
        try (GZIPOutputStream zipStream = new GZIPOutputStream(byteStream)) {
            zipStream.write(body);
        }
        byte[] compressedData = byteStream.toByteArray();
        return compressedData;
    }

}

(Я не мог найти способ комментировать исходный ответ @TestoTestini и сохранить вышеуказанный формат кода, следовательно, этот ответ).