自定義枚舉 --- Gson轉換

經過Restful接口返回的JSON數據默認是枚舉的名字，可是使用自定義枚舉時，通常統一使用自定義的code來表明。因此須要自定義HttpMessageConverter

CodedTypeTypeAdapter

import com.google.gson.*;
import com.utils.mybatis.CodedEnum;

import java.lang.reflect.Type;

/**
 * CodedEnum在GSON中的轉換規則，使用code，而不是字符
 *
 * @param <E>
 * @author tenmao
 */
public class CodedTypeTypeAdapter<E extends Enum<E> & CodedEnum> implements JsonSerializer<E>, JsonDeserializer<E> {
    @Override
    public E deserialize(JsonElement jsonElement, Type type, JsonDeserializationContext jsonDeserializationContext) throws JsonParseException {
        if (type instanceof Class) {
            @SuppressWarnings("unchecked")
            Class<E> klass = (Class<E>) type;
            return CodedEnum.codeOf(klass, jsonElement.getAsInt()).orElse(null);
        } else {
            throw new RuntimeException(String.format("json %s cannot convert to type %s", jsonElement, type));
        }
    }

    @Override
    public JsonElement serialize(E e, Type type, JsonSerializationContext jsonSerializationContext) {
        return new JsonPrimitive(e.getCode());
    }
}

HttpMessageConverter

import com.google.gson.GsonBuilder;
import com.tenmao.utils.mybatis.CodedEnum;
import com.tenmao.utils.mybatis.converter.CodedTypeTypeAdapter;
import lombok.extern.slf4j.Slf4j;
import org.reflections.Reflections;
import org.reflections.scanners.SubTypesScanner;
import org.springframework.http.converter.json.GsonHttpMessageConverter;

import java.util.Set;

/**
 * @author tenmao
 * @since 2017/12/1
 */
@Slf4j
public class HttpMessageConverter extends GsonHttpMessageConverter {
    public HttpMessageConverter() {
        final GsonBuilder builder = new GsonBuilder();
        final Reflections reflections = new Reflections("com.tenmao", new SubTypesScanner(true));
        final Set<String> allClasses = reflections.getStore().getSubTypesOf(CodedEnum.class.getName());
        for (String klass : allClasses) {
            try {
                final Class<?> aClass = Class.forName(klass);
                builder.registerTypeAdapter(aClass, new CodedTypeTypeAdapter<>());
            } catch (ClassNotFoundException e) {
                log.error("fail to register for gson", e);
            }
        }
        setGson(builder.create());
    }
}

spring-mvc.xml

<mvc:annotation-driven>
      <mvc:message-converters>
            <bean class="com.tenmao.web.mvc.support.HttpMessageConverter" />
      </mvc:message-converters>
</mvc:annotation-driven>

完成

實現上述步驟後，只要實現接口CodedEnum的自定義枚舉均可以自動轉換爲其code值

自定義枚舉系列

• 自定義枚舉 --- MyBatis字段映射
• 自定義枚舉 --- Swagger文檔展現

做者：十毛tenmao
連接：https://www.jianshu.com/p/e25ecc0c5762
來源：簡書
簡書著做權歸做者全部，任何形式的轉載都請聯繫做者得到受權並註明出處。