如何反序列化 EnumMap

Question

我想弄清楚如何反序列化一个 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）如何使抽象版本工作？谢谢！

Answer 1

默认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());
    }
}

也可以看看：

• 将 java.lang.reflect.Type 转换为 Class clazz
• 使用 Gson 和自定义序列化来序列化枚举映射
• GSON fromJson 返回 LinkedHashMap 而不是 EnumMap