Java 中一个你绝对没用过,但是很好用的数据结构

Java 中常规的集合工具,相比大家都熟练于胸,但是如果说有一个集合类你不一定知道或者说肯定没用过,你相不相信呢?今天跟大家介绍的就是 java.util.EnumMap,也是 java.util 包下面的一个集合类,同样的也有对应的的 java.util.EnumSet,下面我们看一下吧。

用过的小伙伴在评论扣 1,没用过扣个 0,让阿粉看看有多少小伙伴用过

MapSet 结构在我们日常工作的使用的特别多,经常会用来存放数据或者参数传递,不过有些场景在使用 Map 的时候,不知道大家会不会感受到一丝丝的不安,毕竟 Map 的数据设置我们没办法控制,完全不知道别人会 put 一些什么样的数据进去,或者说如果某些场景我们 Map 的数据 Key 的类型和个数是固定,那在这种情况的下,我们如何提升系统的安全性和性能呢?

这个时候我们就可以考虑使用 EnumMapEnumMap 顾名思义首先是一个 Map,其次它的 key 只能是枚举,大家都知道枚举中的实例个数是固定的,而且还是预编译的,所以在很大程度上保证了数据的安全性,同时也可以提升一定的性能。

EnumMap

下面我们来看下如何使用 EnumMap,首先我们需要创建一个枚举 Color

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
package com.ziyou.demo.enums;
/**
 * <br>
 * <b>Function:</b><br>
 * <b>Author:</b>@author ziyou<br>
 * <b>Date:</b>2022-04-17 15:11<br>
 * <b>Desc:</b>无<br>
 */
public enum Color {
  BLUE("blue", "蓝色"),
  RED("red", "红色"),
  ;
  public String color;
  public String desc;

  Color(String color, String desc) {
    this.color = color;
    this.desc = desc;
  }
}

在创建一个测试类

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
package com.ziyou.demo.enums;

import java.util.EnumMap;

/**
 * <br>
 * <b>Function:</b><br>
 * <b>Author:</b>@author ziyou<br>
 * <b>Date:</b>2022-04-17 15:12<br>
 * <b>Desc:</b>无<br>
 */
public class ColorTest {

  public static void main(String[] args) {
    EnumMap<Color, String> enumMap = new EnumMap<>(Color.class);
    enumMap.put(Color.RED, "我是一个红色枚举");
    enumMap.put(Color.BLUE, "我是一个蓝色枚举");
    System.out.println(enumMap.get(Color.BLUE));
  }
}

我们可以看到构造 EnumMap 的时候需要传入一个枚举类,后续的 putget 都跟普通的 Map 一样,只不过这个时候 put 的时候 key 必须是该枚举实例了。接下来我们看下 EnumMapputget 方法是如何实现的,查看 JDK 源码我们可以看到

1
2
3
4
5
6
7
8
9
10
    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);
    }

在进行 put 的时候,会先进行类型检查,如果说传进来的不是枚举或者说不是在构造的时候指定的枚举,这里就会抛出异常。当类型检查通过以后,会通过枚举的 ordinal() 方法获取该枚举实例的索引,这个方法会返回一个 int 值,返回的值跟枚举在编写的时候的顺序有关系,比如说我们上面创建的 Color 枚举,Color.BLUE.ordinal() 会返回 0,Color.RED.ordinal() 会返回 1。拿到索引过后,就会在对应的数组位置上放上 value 值。

获取数据的时候就更简单了,直接通过 key 获取到索引,然后从数组中拿取数据即可。

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

可以看到整个 EnumMapputget 的效率是非常高的,都是在一维数组中直接根据索引定向处理。所以后续大家在类似的场景中可以尝试使用这种方式来提升性能。

EnumSet

说完了 EnumMap 我们再来看看 EnumSetEnumSet 是一个用来操作 Enum 的集合,是一个抽象类,它有两个继承类,JumboEnumSetRegularEnumSet。在使用的时候,需要确定枚举类型。通过下面的方式可以创建一个空的 EnumSet,在后续进行使用。

1
2
3
4
5
6
  public static void main(String[] args) {
    EnumSet<Color> enumSet = EnumSet.noneOf(Color.class);
    enumSet.add(Color.BLUE);
    enumSet.add(Color.RED);
    System.out.println(enumSet.size());
  }

EnumSet 的构造方式相对会多一点,我们可以创建空的集合,同时我们也可以直接根据创建一个完整的集合,没必要创建空的然后再进行 add 操作,如下所示

1
2
3
4
  public static void main(String[] args) {
    EnumSet<Color> enumSet = EnumSet.allOf(Color.class);
    System.out.println(enumSet.size());
  }

另外前面提到会使用到枚举的 ordinal() 方式,所以我们在构造 EnumSet 的时候还可以只构造指定两个枚举范围之间的所有枚举值,这里要注意 range 方法的第二个参数的枚举不能在第一个枚举前面。

1
EnumSet.range(Color.BLUE,Color.RED);

还可以通过 EnumSetof 方法来构造指定的枚举集合,通过源码我们可以发现不管是通过什么方法进行构造,底层都是先构造一个空集合,然后将对应的枚举元素添加进行。构造空集合的实现逻辑如下,这里我们可以看到,当枚举个数大于 64 的时候,采用的是 JumboEnumSet 这个子类,否则都是 RegularEnumSet 这个子类,正常来说一个枚举的实例个数超过 64 的会比较少吧。

1
2
3
4
5
6
7
8
9
10
    public static <E extends Enum<E>> EnumSet<E> noneOf(Class<E> elementType) {
        Enum<?>[] universe = getUniverse(elementType);
        if (universe == null)
            throw new ClassCastException(elementType + " not an enum");

        if (universe.length <= 64)
            return new RegularEnumSet<>(elementType, universe);
        else
            return new JumboEnumSet<>(elementType, universe);
    }

其他方法的使用跟普通的 Set 都一样,就不一一赘述了,感兴趣的小伙伴可以在工作中用起来。

总结

今天给大家介绍了两个不常用,但是很好用的 JDK 集合类,在某些场景中还是可以提升一定的性能的,小伙伴有使用过这两个集合吗?在评论区告诉阿粉!

Java Geek Tech wechat
欢迎订阅 Java 极客技术,这里分享关于 Java 的一切。