Как 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);
ссылки:
далее к приведенному выше ответу от @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 и сохранить вышеуказанный формат кода, следовательно, этот ответ).