Create a gist now

Instantly share code, notes, and snippets.

What would you like to do?
ConverterCallAdapterFactory uses Retrofit's CallAdapter to pass the okhttp3.Request to a serializer like Jsoup. Idea and corrections from https://github.com/square/retrofit/issues/2267/.
/*
* Copyright (C) 2017 Eric Cochran
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import java.io.IOException;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import okhttp3.Request;
import okhttp3.ResponseBody;
import retrofit2.Call;
import retrofit2.CallAdapter;
import retrofit2.Callback;
import retrofit2.Response;
import retrofit2.Retrofit;
public final class ConverterCallAdapterFactory extends CallAdapter.Factory {
public interface ResponseBodyConverter<T> {
T convert(Request request, ResponseBody body) throws IOException;
interface Factory {
ResponseBodyConverter<?> responseBodyConverter(Type type, Annotation[] annotations,
Retrofit retrofit);
}
}
private final ResponseBodyConverter.Factory converterFactory;
public ConverterCallAdapterFactory(ResponseBodyConverter.Factory converterFactory) {
this.converterFactory = converterFactory;
}
@Override
public CallAdapter<?, ?> get(Type returnType, Annotation[] annotations, Retrofit retrofit) {
@SuppressWarnings("unchecked") final CallAdapter<Object, ?> delegate =
(CallAdapter<Object, ?>) retrofit.nextCallAdapter(this, returnType, annotations);
@SuppressWarnings("unchecked") final ResponseBodyConverter<Object> converter =
(ResponseBodyConverter<Object>) converterFactory.responseBodyConverter(
delegate.responseType(), annotations, retrofit);
return new CallAdapter<ResponseBody, Object>() {
@Override public Type responseType() {
return ResponseBody.class;
}
@Override public Object adapt(Call<ResponseBody> call) {
return delegate.adapt(new ConverterCall<>(call, converter));
}
};
}
private static final class ConverterCall<T> implements Call<T> {
private final Call<ResponseBody> delegate;
final ResponseBodyConverter<T> converter;
ConverterCall(Call<ResponseBody> delegate, ResponseBodyConverter<T> converter) {
this.delegate = delegate;
this.converter = converter;
}
@Override public Response<T> execute() throws IOException {
Response<ResponseBody> response = delegate.execute();
okhttp3.Response raw = response.raw();
if (raw.isSuccessful()) {
return Response.success(converter.convert(delegate.request(), response.body()), raw);
}
return Response.error(raw.body(), raw);
}
@Override public void enqueue(final Callback<T> callback) {
delegate.enqueue(new Callback<ResponseBody>() {
@Override
public void onResponse(Call<ResponseBody> call, Response<ResponseBody> response) {
okhttp3.Response raw = response.raw();
Response<T> converted;
if (raw.isSuccessful()) {
try {
converted =
Response.success(converter.convert(call.request(), response.body()), raw);
} catch (IOException e) {
callback.onFailure(ConverterCall.this, e);
return;
}
} else {
converted = Response.error(raw.body(), raw);
}
callback.onResponse(ConverterCall.this, converted);
}
@Override public void onFailure(Call<ResponseBody> call, Throwable t) {
callback.onFailure(ConverterCall.this, t);
}
});
}
@Override public boolean isExecuted() {
return delegate.isExecuted();
}
@Override public void cancel() {
delegate.cancel();
}
@Override public boolean isCanceled() {
return delegate.isCanceled();
}
@Override public Call<T> clone() {
return new ConverterCall<>(delegate.clone(), converter);
}
@Override public Request request() {
return delegate.request();
}
}
}
Retrofit retrofit = new Retrofit.Builder().baseUrl(server.url("/"))
.addCallAdapterFactory(new ConverterCallAdapterFactory(
new ConverterCallAdapterFactory.ResponseBodyConverter.Factory() {
@Override
public ConverterCallAdapterFactory.ResponseBodyConverter<?> responseBodyConverter(
final Type type, Annotation[] annotations, Retrofit retrofit) {
return new ConverterCallAdapterFactory.ResponseBodyConverter<Object>() {
@Override public Object convert(Request request, ResponseBody body)
throws IOException {
return Jsoup.parse(body.string(), request.url().toString());
}
};
}
}))
.build();
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment