通過Restful接口返回的JSON數(shù)據(jù)默認(rèn)是枚舉的名字蓝晒，但是使用自定義枚舉時，一般統(tǒng)一使用自定義的code來代表撵割。所以需要自定義
HttpMessageConverter
CodedTypeTypeAdapter
import com.google.gson.*;
import com.utils.mybatis.CodedEnum;
import java.lang.reflect.Type;
/**
* CodedEnum在GSON中的轉(zhuǎn)換規(guī)則，使用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>
完成
實(shí)現(xiàn)上述步驟后澎蛛，只要實(shí)現(xiàn)接口CodedEnum的自定義枚舉都可以自動轉(zhuǎn)換為其code值
