当前位置: 代码迷 >> java >> 如何反序列化EnumMap
  详细解决方案

如何反序列化EnumMap

热度:64   发布时间:2023-07-25 19:39:15.0

我正在试图弄清楚如何反序列化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)如何使抽象版本工作? 谢谢!

默认情况下, 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());
    }
}

也可以看看:

根据例外,似乎Gson能够将输入反序列化为LinkedHashMap 因此,您可以使用反序列化结果初始化实例,而不是赋值:

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