如何反序列化EnumMap

时间:2019-03-03 06:16:54

标签: java gson deserialization enum-map

我试图弄清楚如何反序列化EnumMap。到目前为止,我一直在使用Gson库进行其他所有操作,并取得了成功。事实证明这很困难。

这是一个基本概念:

import java.lang.reflect.Type;
import com.google.gson.reflect.TypeToken;
import com.google.gson.Gson;

enum FRUIT {
  APPLE, BANANA
}
EnumMap<FRUIT, String> fruitMap;
Gson gson = new Gson();

public void setFruitMap(String jsonString){
  Type typeToken = new TypeToken<EnumMap<FRUIT, String>>(){}.getType();
  fruitMap = gson.fromJson(jsonString, typeToken);
}

String fruitMapString = "{ \"BANANA\":\"tasty!\", \"APPLE\":\"gross!\" }";

setFruitMap(fruitMapString); //Error occurs here.

assertEquals("tasty!", fruitMap.get(FRUIT.BANANA));

运行上面的代码时,我得到一个java.lang.ClassCastException: java.util.LinkedHashMap cannot be cast to java.util.EnumMap

所以在我看来,Gson库不是在创建EnumMap,而是在创建LinkedHashMap之后尝试进行转换。

所以我想我要制定自己的反序列化逻辑。这是有效的实现。但是..有点儿简陋。

public JsonDeserializer<EnumMap<FRUIT, String>> deserializeFruitMap(){
  return new JsonDeserializer<EnumMap<FRUIT, String>>(){
    @Override
    public EnumMap<FRUIT, String> deserialize(JsonElement element, Type typeOf, JsonDerializationContext context){
      Type token = new TypeToken<HashMap<String, String>>(){}.getType();
      HashMap<String, String> tempMap = context.deserialize(element, token);
      EnumMap<FRUIT, String> fruitMap = new EnumMap<>(FRUIT.class);
      for(Entry<String, String> entry : tempMap.entrySet){
        fruitMap.put(FRUIT.valueOf(entry.getKey()), entry.getValue());
      }
      return fruitMap;
    }
  };
}

它可以工作,但不是很漂亮。这是非常具体的。我真的很想将其抽象为...

public <K extends Enum<K>, V> JsonDeserializer<EnumMap<K, V>> deserializeEnumMap(){
  return new JsonDeserializer<EnumMap<K, V>>(){
    @Override
    public EnumMap<K, V> deserialize(JsonElement element, Type typeOf, JsonDerializationContext context){
      Type token = new TypeToken<HashMap<String, String>>(){}.getType();
      HashMap<String, String> tempMap = context.deserialize(element, token);
      EnumMap<K, V> enumMap = new EnumMap<>(K.class); //This doesn't work.
      for(Entry<String, String> entry : tempMap.entrySet){
        fruitMap.put(K.valueOf(entry.getKey()), entry.getValue());
      }
      return enumMap;
    }
  };
}

有人在以下方面有任何想法吗:1)如何改进第一种方法? 2)如何使抽象版本起作用?谢谢!

2 个答案:

答案 0 :(得分:3)

默认情况下,Gson无法识别EnumMap。它像常规Map一样对待它。但是此实现没有Gson可以使用的默认构造函数。我们需要提供如下的InstanceCreator

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.InstanceCreator;
import com.google.gson.reflect.TypeToken;

import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.EnumMap;

public class JsonApp {

    enum FRUIT {
        APPLE, BANANA
    }

    enum Apple {
        LIGOL,
        MUTSU
    }

    public static void main(String[] args) {
        Gson gson = new GsonBuilder()
                .registerTypeAdapter(EnumMap.class, new InstanceCreator<EnumMap>() {
                    @Override
                    public EnumMap createInstance(Type type) {
                        Type[] types = (((ParameterizedType) type).getActualTypeArguments());
                        return new EnumMap((Class<?>) types[0]);
                    }
                })
                .create();

        String jsonString = "{ \"BANANA\":\"tasty!\", \"APPLE\":\"gross!\" }";
        String jsonString1 = "{ \"LIGOL\":\"red!\", \"MUTSU\":\"green!\" }";

        Type enumMap1 = new TypeToken<EnumMap<FRUIT, String>>() {}.getType();
        System.out.println(gson.fromJson(jsonString, enumMap1).toString());

        Type enumMap2 = new TypeToken<EnumMap<Apple, String>>() {}.getType();
        System.out.println(gson.fromJson(jsonString1, enumMap2).toString());
    }
}

另请参阅:

答案 1 :(得分:2)

根据例外情况,Gson似乎能够将输入反序列化为LinkedHashMap。因此,您可以使用反序列化结果来初始化实例,而无需进行赋值:

public void setFruitMap(String jsonString){
  Type typeToken = new TypeToken<Map<FRUIT, String>>(){}.getType();
  fruitMap = new EnumMap<>(gson.fromJson(jsonString, typeToken));
}