io.github.ximutech.spore.retrofit.JacksonConverterFactory Maven / Gradle / Ivy

Go to download
package io.github.ximutech.spore.retrofit;

import com.fasterxml.jackson.databind.JavaType;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.ObjectReader;
import com.fasterxml.jackson.databind.ObjectWriter;
import okhttp3.MediaType;
import okhttp3.RequestBody;
import okhttp3.ResponseBody;
import retrofit2.Converter;
import retrofit2.Retrofit;

import java.io.IOException;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;

/**
 * 支持Jackson的转换器
 *
 * @author ximu
 */
public final class JacksonConverterFactory extends Converter.Factory {
    private String charset = "UTF-8";
    private MediaType MEDIA_TYPE;


    public static JacksonConverterFactory create(ObjectMapper mapper) {
        if (mapper == null) {
            throw new NullPointerException("mapper == null");
        }
        return new JacksonConverterFactory(mapper);
    }

    public static JacksonConverterFactory create(ObjectMapper mapper, String charset) {
        if (mapper == null) {
            throw new NullPointerException("mapper == null");
        }
        if (charset == null) {
            throw new NullPointerException("charset == null");

        }
        return new JacksonConverterFactory(mapper, charset);
    }

    private final ObjectMapper mapper;

    private JacksonConverterFactory(ObjectMapper mapper) {
        MEDIA_TYPE = MediaType.parse("application/json; charset=" + charset);
        this.mapper = mapper;
    }

    private JacksonConverterFactory(ObjectMapper mapper, String charset) {
        this.charset = charset;
        MEDIA_TYPE = MediaType.parse("application/json; charset=" + charset);
        this.mapper = mapper;
    }

    @Override
    public Converter responseBodyConverter(Type type, Annotation[] annotations,
                                                            Retrofit retrofit) {
        JavaType javaType = mapper.getTypeFactory().constructType(type);
        ObjectReader reader = mapper.readerFor(javaType);
        return new JacksonResponseBodyConverter<>(reader);
    }

    @Override
    public Converter requestBodyConverter(Type type,
                                                          Annotation[] parameterAnnotations, Annotation[] methodAnnotations, Retrofit retrofit) {
        JavaType javaType = mapper.getTypeFactory().constructType(type);
        ObjectWriter writer = mapper.writerFor(javaType);
        return new JacksonRequestBodyConverter<>(writer);
    }

    private class JacksonResponseBodyConverter implements Converter {
        private final ObjectReader adapter;

        JacksonResponseBodyConverter(ObjectReader adapter) {
            this.adapter = adapter;
        }

        @Override
        public T convert(ResponseBody value) throws IOException {
            try {
                return adapter.readValue(new String(value.bytes(), charset));
            } finally {
                value.close();
            }
        }
    }

    private class JacksonRequestBodyConverter implements Converter {

        private final ObjectWriter adapter;

        JacksonRequestBodyConverter(ObjectWriter adapter) {
            this.adapter = adapter;
        }

        @Override
        public RequestBody convert(T value) throws IOException {
            byte[] bytes = adapter.writeValueAsBytes(value);
            return RequestBody.create(MEDIA_TYPE, bytes);
        }
    }
}