com.tukeof.common.rest.okhttp.RetrofitUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-rest Show documentation
a encapsulated restful java library
The newest version!
package com.tukeof.common.rest.okhttp;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.thoughtworks.xstream.XStream;
import com.tukeof.common.rest.gson.GsonUtil;
import com.tukeof.common.rest.xstream.XStreamUtil;
import com.tukeof.common.rest.xstream.XStreamXmlConverterFactory;
import lombok.extern.slf4j.Slf4j;
import okhttp3.OkHttpClient;
import okhttp3.ResponseBody;
import retrofit2.Call;
import retrofit2.CallAdapter;
import retrofit2.Converter;
import retrofit2.Response;
import retrofit2.Retrofit;
import retrofit2.converter.gson.GsonConverterFactory;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

@Slf4j
public class RetrofitUtil {

    private static HashMap instances = new HashMap<>();

    private static Gson gson = new GsonBuilder().create();
    private static Converter.Factory gsonFactory =
            GsonConverterFactory.create(gson);
    private static XStreamXmlConverterFactory xstreamFactory;

    private static Converter.Factory buildGsonFactory() {
        return gsonFactory;
    }

    private static Converter.Factory buildXStreamFactory() {
        if (xstreamFactory == null) {
            XStream xStream = XStreamUtil.newXStream();
            xstreamFactory = XStreamXmlConverterFactory.create(xStream);
        }
        return xstreamFactory;
    }

    // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    public static  T createService(Retrofit retrofit, Class clazz) {
        return retrofit.create(clazz);
    }

    public static  T info(Call call) throws IOException {
        Response response = null;
        response = call.execute();
        if (response.isSuccessful()) {
            T responseBody = response.body();
            String responseContent = null;
            if (responseBody instanceof ResponseBody) {
                responseContent = ((ResponseBody) responseBody).string();
            } else {
                responseContent = GsonUtil.toJson(responseBody);
            }
            log.info("---------------- success:\t{}", responseContent);
            return responseBody;
        } else {
            log.info("---------------- http status code:\t{}", response.code());
            log.info("---------------- http response message:\t{}", response.message());
            String errorStrig = response.errorBody() == null ? "null" : response.errorBody().string();
            log.info("---------------- http error body:\t{}", errorStrig);
            throw new IOException("---------------- request error");
        }
    }

    // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    public static Retrofit getInstance4Json(String baseUrl) {
        return getInstance(baseUrl, buildGsonFactory());
    }

    public static Retrofit getInstance4Json(String baseUrl, Map headers) {
        return getInstance(baseUrl, buildGsonFactory(), headers);
    }

    public static Retrofit getInstance4Xml(String baseUrl) {
        return getInstance(baseUrl, buildXStreamFactory());
    }

    public static Retrofit getInstance4Xml(String baseUrl, Map headers) {
        return getInstance(baseUrl, buildXStreamFactory(), headers);
    }

    // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    public static Retrofit getInstance(String baseUrl, Converter.Factory factory) {
        return getInstance(baseUrl, factory, null, null);
    }

    public static Retrofit getInstance(String baseUrl, Converter.Factory factory, Map headers) {
        return getInstance(baseUrl, factory, null, headers);
    }

    public static Retrofit getInstance(String baseUrl, Converter.Factory converterFactory, CallAdapter.Factory callAdapterFactory) {
        return getInstance(baseUrl, converterFactory, callAdapterFactory, null);
    }

    public static Retrofit getInstance(
            String baseUrl, Converter.Factory converterFactory, CallAdapter.Factory callAdapterFactory, Map headers) {
        if (instances.containsKey(baseUrl)) return instances.get(baseUrl);

        Retrofit.Builder builder = new Retrofit.Builder()
                .baseUrl(baseUrl)
                .client(OkHttpUtil.newClient(headers, null, null));

        if (converterFactory != null) builder.addConverterFactory(converterFactory);
        if (callAdapterFactory != null) builder.addCallAdapterFactory(callAdapterFactory);

        Retrofit retrofit = builder.build();
        instances.put(baseUrl, retrofit);
        return retrofit;
    }

}