1. 介绍

MapUtils 是 Apache Commons Collections 项目中的一个工具类。

它提供了一些实用方法和修饰器,用于处理 java.util.Map 和 java.util.SortedMap 实例。

2. 配置

添加 mvn 依赖:

<dependency>
    <groupId>org.apache.commons</groupId>
    <artifactId>commons-collections4</artifactId>
    <version>4.5.0-M2</version>
</dependency>

3. 工具方法

3.1. 通过 Array 创建 Map

首先我们初始化一个一维数组、和一个二维数组:

public class MapUtilsTest {
    private String[][] color2DArray = new String[][] {
        {"RED", "#FF0000"},
        {"GREEN", "#00FF00"},
        {"BLUE", "#0000FF"}
    };
    private String[] color1DArray = new String[] {
        "RED", "#FF0000",
        "GREEN", "#00FF00",
        "BLUE", "#0000FF"
    };
    private Map<String, String> colorMap;

    //...
}

下面演示如何从一个二维数组中创建一个map:

@Test
public void whenCreateMapFrom2DArray_theMapIsCreated() {
    this.colorMap = MapUtils.putAll(
      new HashMap<>(), this.color2DArray);

    assertThat(
      this.colorMap, 
      is(aMapWithSize(this.color2DArray.length)));

    assertThat(this.colorMap, hasEntry("RED", "#FF0000"));
    assertThat(this.colorMap, hasEntry("GREEN", "#00FF00"));
    assertThat(this.colorMap, hasEntry("BLUE", "#0000FF"));
}

一维数组:

@Test
public void whenCreateMapFrom1DArray_theMapIsCreated() {
    this.colorMap = MapUtils.putAll(
      new HashMap<>(), this.color1DArray);

    assertThat(
      this.colorMap, 
      is(aMapWithSize(this.color1DArray.length / 2)));

    assertThat(this.colorMap, hasEntry("RED", "#FF0000"));
    assertThat(this.colorMap, hasEntry("GREEN", "#00FF00"));
    assertThat(this.colorMap, hasEntry("BLUE", "#0000FF"));
}

3.2. 打印 Map 内容

经常我们调试的时候,希望打印整个map:

@Test
public void whenVerbosePrintMap_thenMustPrintFormattedMap() {
    MapUtils.verbosePrint(System.out, "Optional Label", this.colorMap);
}

输出结果:

Optional Label = 
{
    RED = #FF0000
    BLUE = #0000FF
    GREEN = #00FF00
}

也可使用 *debugPrint()*,它可以额外地打印出数值的数据类型。

3.3. 取值

MapUtils 提供了一些 null-safe 的get方法。

例如,使用 getString() 获取 String 类型的值,String值是通过toString()获得的。如果值为null或者类型转换失败,可以指定返回默认值:

@Test
public void whenGetKeyNotPresent_thenMustReturnDefaultValue() {
    String defaultColorStr = "COLOR_NOT_FOUND";
    String color = MapUtils
      .getString(this.colorMap, "BLACK", defaultColorStr);

    assertEquals(color, defaultColorStr);
}

这些方法是 null 安全的,可以安全的处理null map:

@Test
public void whenGetOnNullMap_thenMustReturnDefaultValue() {
    String defaultColorStr = "COLOR_NOT_FOUND";
    String color = MapUtils.getString(null, "RED", defaultColorStr);

    assertEquals(color, defaultColorStr);
}

3.4. 反转 Map

我们还可以轻松地反转一个 map,即将 key和value进行调换:

@Test
public void whenInvertMap_thenMustReturnInvertedMap() {
    Map<String, String> invColorMap = MapUtils.invertMap(this.colorMap);

    int size = invColorMap.size();
    Assertions.assertThat(invColorMap)
      .hasSameSizeAs(colorMap)
      .containsKeys(this.colorMap.values().toArray(new String[] {}))
      .containsValues(this.colorMap.keySet().toArray(new String[] {}));
}

结果:

{
    #00FF00 = GREEN
    #FF0000 = RED
    #0000FF = BLUE
}

如果源 map 存在多个key关联相同的值,则在反转后,其中一个值将随机成为一个key。

3.5. 空值检查

isEmpty() 用于判断 Map 是否为null或元素为空。

safeAddToMap() 方法防止将 null 添加到 map 中

4. Map 修饰器

4.1. 固定大小 Map

fixedSizeMap() 会返回一个固定大小的map。可以更改,但无法新增或删除元素:

@Test(expected = IllegalArgumentException.class)
public void whenCreateFixedSizedMapAndAdd_thenMustThrowException() {
    Map<String, String> rgbMap = MapUtils
      .fixedSizeMap(MapUtils.putAll(new HashMap<>(), this.color1DArray));

    rgbMap.put("ORANGE", "#FFA500");
}

4.2. Predicated Map

predicatedMap() 方法返回一个 map,确保所有的元素都符合提供的 predicate:

@Test(expected = IllegalArgumentException.class)
public void whenAddDuplicate_thenThrowException() {
    Map<String, String> uniqValuesMap 
      = MapUtils.predicatedMap(this.colorMap, null, 
        PredicateUtils.uniquePredicate());

    uniqValuesMap.put("NEW_RED", "#FF0000");
}

上面,我们使用 PredicateUtils.uniquePredicate() ,任何尝试将重复值插入此 map 中都将抛出 IllegalArgumentException 异常。

我们也可实现 Predicate 接口来自定义异常。

4.3. Lazy Map

lazyMap() 返回一个 map,其中的值在被请求时才会被初始化。

如果传递给该 map 的 Map.get(Object) 方法的键在 map 中不存在,将使用 Transformer 实例来创建一个新对象,并与请求的键关联:

@Test
public void whenCreateLazyMap_theMapIsCreated() {
    Map<Integer, String> intStrMap = MapUtils.lazyMap(
      new HashMap<>(),
      TransformerUtils.stringValueTransformer());

    assertThat(intStrMap, is(anEmptyMap()));

    intStrMap.get(1);
    intStrMap.get(2);
    intStrMap.get(3);

    assertThat(intStrMap, is(aMapWithSize(3)));
}

5. 总结

本文我们探索了 Apache Commons Collections 的 MapUtils 类,并介绍了各种实用方法和修饰器,可以简化各种常见的 map 操作。

文中代码可在 GitHub 上找到。


« 上一篇: Java周报,186