6.33 使用EnumMap替代使用枚举量的ordinal做索引

假设有如下类Food（食物），枚举类Type定义Food的类型：

public class Food {public enum Type{FRUIT,VEGETABLE}private final Type type;private final String name;public Food(String name,Type type){this.name=name;this.type=type;}public String getName(){return name;}public Type getType(){return type;}}

?

以下数组包含了多种食物，如下：

private Food[] foods;@BeforeClasspublic void init(){Food apple=new Food("apple",Type.FRUIT);Food pear=new Food("pear",Type.FRUIT);Food cabbage=new Food("cabbage",Type.VEGETABLE);Food potato=new Food("potato",Type.VEGETABLE);foods=new Food[]{apple,cabbage,pear,potato};}

?

假设现在需要根据Food类型对其进行分类，把相同类型的Food放入同一个HashSet对象里，其中一种做法如下：

@Testpublic void testFood(){//根据Food类型分类,存放到相应的Set里Set<Food>[] foodsByType=new Set[Type.values().length];for(int i=0,len=foodsByType.length;i<len;i++)foodsByType[i]=new HashSet<Food>();for(Food food:foods){//使用了Type枚举量的ordinal()作为索引值，取出相应的Set对象foodsByType[food.getType().ordinal()].add(food);}for(Type type:Type.values()){Set<Food> foods=foodsByType[type.ordinal()];for(Food food:foods)Assert.assertEquals(food.getType(), type);}}

?

以上代码其中一个缺点是只能使用int类型的索引值从数组里取出相应的HashSet对象，可能导致ArrayIndexOutOfBoundsException异常。另一种更好的做法如下：

@Testpublic void testFoodMap(){Map<Type, Set<Food>> foodsByType=new EnumMap<Food.Type, Set<Food>>(Type.class);for(Type type:Type.values())foodsByType.put(type, new HashSet<Food>());//使用Type枚举量做为key值for(Food food:foods)foodsByType.get(food.getType()).add(food);for(Type type:Type.values()){Set<Food> foods=foodsByType.get(type);for(Food food:foods)Assert.assertEquals(food.getType(), type);}}

?

以上代码EnumMap使用枚举量Type作为key，因此更不容易出错。EnumMap底层使用数组维护key，因此与第一种做法在性能上相差无几。

?

?

?