io.freefair.spring.okhttp.client.OkHttpClientResponse Maven / Gradle / Ivy
package io.freefair.spring.okhttp.client;
import kotlin.Pair;
import lombok.RequiredArgsConstructor;
import okhttp3.Headers;
import okhttp3.Response;
import okhttp3.ResponseBody;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatusCode;
import org.springframework.http.client.ClientHttpResponse;
import java.io.InputStream;
/**
* OkHttp based {@link ClientHttpResponse} implementation.
*
* @author Lars Grefer
* @see OkHttpClientRequest
*/
@RequiredArgsConstructor
public class OkHttpClientResponse implements ClientHttpResponse {
private final Response okHttpResponse;
private HttpHeaders springHeaders;
@Override
public HttpStatusCode getStatusCode() {
return HttpStatusCode.valueOf(okHttpResponse.code());
}
@Override
public String getStatusText() {
return okHttpResponse.message();
}
@Override
public void close() {
ResponseBody body = okHttpResponse.body();
if (body != null) {
body.close();
}
}
@Override
public InputStream getBody() {
ResponseBody body = okHttpResponse.body();
if (body != null) {
return body.byteStream();
} else {
return InputStream.nullInputStream();
}
}
@Override
public HttpHeaders getHeaders() {
if (springHeaders == null) {
springHeaders = convertHeaders(okHttpResponse.headers());
}
return springHeaders;
}
/**
* Converts the given {@link Headers OkHttp Headers} to {@link HttpHeaders Spring Web HttpHeaders}
*/
static HttpHeaders convertHeaders(Headers okHttpHeaders) {
HttpHeaders springHeaders = new HttpHeaders();
for (Pair header : okHttpHeaders) {
springHeaders.add(header.getFirst(), header.getSecond());
}
return springHeaders;
}
}