枚举详解之EnumSet、EnumMap用法

枚举简单例子

/**
 * @author shuliangzhao
 * @Title: Color
 * @ProjectName design-parent
 * @Description: TODO * @date 2019/6/12 22:35 */ public enum Color { YELLOW, RED, BLUE, PURPLE, BLACK; }

/**
 * @author shuliangzhao
 * @Title: ColorEnumTest
 * @ProjectName design-parent
 * @Description: TODO * @date 2019/6/12 22:37 */ public class ColorEnumTest { public static void main(String[] args) { Color color = Color.BLUE; switch (color) { case RED: System.out.println("红色"); break; case BLUE: System.out.println("蓝色"); break; case PURPLE: System.out.println("紫色"); break; case BLACK: System.out.println("黑色"); break; case YELLOW: System.out.println("黄色"); break; default: System.out.println("蓝色"); break; } } }

测试结果

image.png

Color枚举的本质就是一个类，编译器会自动为我们生成Color类，通过反编译得到该类如下：

final class Color extends Enum { //编译器为我们添加的静态的values()方法 public static Color[] values() { return (Color[])$VALUES.clone(); } //编译器为我们添加的静态的valueOf()方法，注意间接调用了Enum也类的valueOf方法 public static Color valueOf(String s) { return (Color)Enum.valueOf(com/sl/emun/Color, s); } //私有构造函数 private Color(String s, int i) { super(s, i); } //前面定义的7种枚举实例 public static final Color YELLOW; public static final Color RED; public static final Color BLUE; public static final Color PURPLE; public static final Color BLACK; private static final Color $VALUES[]; static { //实例化枚举实例 YELLOW = new Color("YELLOW", 0); RED = new Color("RED", 1); BLUE = new Color("BLUE", 2); PURPLE = new Color("PURPLE", 3); BLACK = new Color("BLACK", 4); $VALUES = (new Color[] { YELLOW, RED, BLUE, PURPLE, BLACK }); } }

可以看出每个枚举类型颜色是该Color类的一个实例对象，该构成方式和单例模式有些类似，故可以用只有一个枚举类型的枚举作为单例模式，而且枚举的构造器由编译器管理安全性十分高，既可以防止反射破解也可以防止反序列破解。

EnumMap用法

/**
 * @author shuliangzhao
 * @Title: EnumMapTest
 * @ProjectName design-parent
 * @Description: TODO * @date 2019/6/12 22:49 */ public class EnumMapTest { public static void main(String[] args) { EnumMap<Color,String> enumMap = new EnumMap(Color.class); enumMap.put(Color.BLACK,"黑色"); enumMap.put(Color.BLUE,"蓝色"); System.out.println(enumMap); System.out.println(enumMap.get(Color.BLUE)); } }

运行结果

image.png

EnumMap put方法

public V put(K key, V value) {
        typeCheck(key);

        int index = key.ordinal(); Object oldValue = vals[index]; vals[index] = maskNull(value); if (oldValue == null) size++; return unmaskNull(oldValue); }

首先调用typeCheck检查键的类型，如果类型不对，会抛出异常。类型正确的话，调用ordinal获取索引index，并将值value放入值数组vals[index]中。EnumMap允许值为null，为了区别null值与没有值，EnumMap将null值包装成了一个特殊的对象，有两个辅助方法用于null的打包和解包，打包方法为maskNull，解包方法为unmaskNull。

get方法

public V get(Object key) {
        return (isValidKey(key) ? unmaskNull(vals[((Enum<?>)key).ordinal()]) : null); }

键有效的话，通过ordinal方法取索引，然后直接在值数组vals里找。isValidKey的代码与typeCheck类似，但是返回boolean值而不是抛出异常。

以上就是EnumMap的基本实现原理，内部有两个数组，长度相同，一个表示所有的键，一个表示对应的值，值为null表示没有该键值对，键都有一个对应的索引，根据索引可直接访问和操作其键和值，效率很高。

EnumSet

EnumSet这是一个用来操作Enum的集合，是一个抽象类，它有两个继承类：JumboEnumSet和RegularEnumSet。在使用的时候，需要确定枚举类型。它的特点也是速度非常快，为什么速度很快呢？因为每次add的时候，每个枚举值只占一个长整型的一位。
EnumSet.noneOf()方法创建一个空的set

public class EnumSetTest {

    public static void main(String[] args) { EnumSet<Color> enumSet = EnumSet.noneOf(Color.class); System.out.println(enumSet); enumSet.add(Color.BLUE); enumSet.add(Color.PURPLE); System.out.println(enumSet); } }

运行结果

image.png

EnumSet.allOf()方法创建一个满的set

/**
 * @author shuliangzhao
 * @Title: EnumSetTest
 * @ProjectName design-parent
 * @Description: TODO * @date 2019/6/12 23:24 */ public class EnumSetTest { public static void main(String[] args) { /*EnumSet<Color> enumSet = EnumSet.noneOf(Color.class); System.out.println(enumSet); enumSet.add(Color.BLUE); enumSet.add(Color.PURPLE); System.out.println(enumSet);*/ EnumSet<Color> enumSet = EnumSet.allOf(Color.class); System.out.println(enumSet); } }

运行结果

image.png

EnumSet.range创建指定范围set

public class EnumSetTest {

    public static void main(String[] args) { /*EnumSet<Color> enumSet = EnumSet.noneOf(Color.class); System.out.println(enumSet); enumSet.add(Color.BLUE); enumSet.add(Color.PURPLE); System.out.println(enumSet); EnumSet<Color> enumSet = EnumSet.allOf(Color.class); System.out.println(enumSet);*/ EnumSet<Color> enumSet = EnumSet.range(Color.YELLOW,Color.BLUE); System.out.println(enumSet); } }

image.png

EnumSet.complementOf补集创建set

public class EnumSetTest {

    public static void main(String[] args) { /*EnumSet<Color> enumSet = EnumSet.noneOf(Color.class); System.out.println(enumSet); enumSet.add(Color.BLUE); enumSet.add(Color.PURPLE); System.out.println(enumSet); EnumSet<Color> enumSet = EnumSet.allOf(Color.class); System.out.println(enumSet);*/ EnumSet<Color> enumSet = EnumSet.range(Color.YELLOW,Color.BLUE); System.out.println(enumSet); EnumSet<Color> enumSet1 = EnumSet.complementOf(enumSet); System.out.println(enumSet1); } }

运行结果

image.png

EnumSet.copyOf复制创建set

public class EnumSetTest {

    public static void main(String[] args) { /*EnumSet<Color> enumSet = EnumSet.noneOf(Color.class); System.out.println(enumSet); enumSet.add(Color.BLUE); enumSet.add(Color.PURPLE); System.out.println(enumSet); EnumSet<Color> enumSet = EnumSet.allOf(Color.class); System.out.println(enumSet); EnumSet<Color> enumSet = EnumSet.range(Color.YELLOW,Color.BLUE); System.out.println(enumSet); EnumSet<Color> enumSet1 = EnumSet.complementOf(enumSet); System.out.println(enumSet1);*/ EnumSet<Color> enumSet = EnumSet.range(Color.YELLOW,Color.BLUE); System.out.println(enumSet); EnumSet<Color> enumSet1 = EnumSet.copyOf(enumSet); System.out.println(enumSet1); } }

运行结果

image.png

EnumSet.copyOf复制创建集合

public class EnumSetTest {

    public static void main(String[] args) { /*EnumSet<Color> enumSet = EnumSet.noneOf(Color.class); System.out.println(enumSet); enumSet.add(Color.BLUE); enumSet.add(Color.PURPLE); System.out.println(enumSet); EnumSet<Color> enumSet = EnumSet.allOf(Color.class); System.out.println(enumSet); EnumSet<Color> enumSet = EnumSet.range(Color.YELLOW,Color.BLUE); System.out.println(enumSet); EnumSet<Color> enumSet1 = EnumSet.complementOf(enumSet); System.out.println(enumSet1); EnumSet<Color> enumSet = EnumSet.range(Color.YELLOW,Color.BLUE); System.out.println(enumSet); EnumSet<Color> enumSet1 = EnumSet.copyOf(enumSet); System.out.println(enumSet1);*/ List<Color> colors = new ArrayList<>(); colors.add(Color.PURPLE); colors.add(Color.BLUE); colors.add(Color.BLUE); System.out.println(colors); EnumSet<Color> enumSet = EnumSet.copyOf(colors); System.out.println(enumSet); } }

运行结果

image.png