Java8-Map-新增方法

Posted on Edited on In Views:

在 Java 8 中的 Map.Entry 接口中增加了 comparingByKey, comparingByValue 方法，它们都返回 Comparator<Map.Entry<K,V>>, Comparator 是一个函数接口，主要是方便 Lambda 表达式的使用。

在 Java 8 中的 Map 接口增加了一些 default 方法，提升了对 key，value 操作的便利性。下面是基本数据的定义，通过这些数据说明新增的一些方法。

1
2
3
4
Map<Integer, String> map = new HashMap<>();
map.put(1, "a");
map.put(2, "b");
map.put(3, "c");

getOrDefault 方法

如果指定的 key 存在，则返回该 key 对应的 value，如果不存在，则返回指定的值。例子如下

1
2
// key为4不存在，输出 d
System.out.println(map.getOrDefault(4, "d"));

forEach 方法

遍历 Map 中的所有 Entry, 对 key, value 进行处理， 接收参数 (K, V) -> void, 例子如下

1
2
// 输出1a, 2b, 3c
map.forEach((key, value) -> System.out.println(key + value));

replaceAll 方法

替换 Map 中所有 Entry 的 value 值，这个值由旧的 key 和 value 计算得出，接收参数 (K, V) -> V, 类似如下代码

1
2
for (Map.Entry<K, V> entry : map.entrySet())
    entry.setValue(function.apply(entry.getKey(), entry.getValue()));

例如如下：

1
2
3
map.replaceAll((key, value) -> (key + 1) + value);
// 输出 12a 23b 34c
map.forEach((key, value) -> System.out.println(key + value));

putIfAbsent 方法

如果 key 关联的 value 不存在，则关联新的 value 值，返回 key 关联的旧的值，类似如下代码

1
2
3
4
5
V v = map.get(key);
if (v == null)
    v = map.put(key, value);

return v;

示例代码如下：

1
2
3
4
5
6
map.putIfAbsent(3, "d");
map.putIfAbsent(4, "d");
// 输出 c
System.out.println(map.get(3));
// 输出 d
System.out.println(map.get(4));

remove 方法

接收 2 个参数，key 和 value，如果 key 关联的 value 值与指定的 value 值相等（equals），则删除这个元素，类似代码如下：

1
2
3
4
5
if (map.containsKey(key) && Objects.equals(map.get(key), value)) {
    map.remove(key);
    return true;
} else
    return false;

示例代码如下：

1
2
3
4
5
6
7
map.remove(1, "b");
// 未删除成功， 输出 a
System.out.println(map.get(1));

map.remove(2, "b");
// 删除成功，输出 null
System.out.println(map.get(2));

replace(K key, V oldValue, V newValue) 方法

如果 key 关联的值与指定的 oldValue 的值相等，则替换成新的 newValue，类似代码如下：

1
2
3
4
5
if (map.containsKey(key) && Objects.equals(map.get(key), value)) {
    map.put(key, newValue);
    return true;
} else
    return false;

示例代码如下

1
2
3
4
5
6
7
map.replace(3, "a", "z");
// 未替换成功，输出 c
System.out.println(map.get(3));

map.replace(1, "a", "z");
// 替换成功， 输出 z
System.out.println(map.get(1));

replace(K key, V value) 方法

如果 map 中存在 key，则替换成 value 值，否则返回 null, 类似代码如下:

1
2
3
4
if (map.containsKey(key)) {
    return map.put(key, value);
} else
    return null;

示例代码如下：

1
2
3
4
5
6
7
8
9
// 输出旧的值， a
System.out.println(map.replace(1, "aa"));
// 替换成功，输出新的值， aa
System.out.println(map.get(1));

// 不存在key为4， 输出 null
System.out.println(map.replace(4, "d"));
// 不存在key为4， 输出 null
System.out.println(map.get(4));

computeIfAbsent 方法

如果指定的 key 不存在，则通过指定的 K -> V 计算出新的值设置为 key 的值，类似代码如下：

1
2
3
4
5
if (map.get(key) == null) {
    V newValue = mappingFunction.apply(key);
    if (newValue != null)
        map.put(key, newValue);
}

示例代码如下：

1
2
3
4
5
6
7
map.computeIfAbsent(1, key -> key + " computed");
// 存在key为1，则不进行计算，输出值 a
System.out.println(map.get(1));

map.computeIfAbsent(4, key -> key + " computed");
// 不存在key为4，则进行计算，输出值 4 computed
System.out.println(map.get(4));

computeIfPresent 方法

如果指定的 key 存在，则根据旧的 key 和 value 计算新的值 newValue, 如果 newValue 不为 null，则设置 key 新的值为 newValue, 如果 newValue 为 null, 则删除该 key 的值，类似代码如下：

1
2
3
4
5
6
7
8
if (map.get(key) != null) {
    V oldValue = map.get(key);
    V newValue = remappingFunction.apply(key, oldValue);
    if (newValue != null)
        map.put(key, newValue);
    else
        map.remove(key);
}

示例代码如下：

1
2
3
4
5
6
7
map.computeIfPresent(1, (key, value) -> (key + 1) + value);
// 存在key为1， 则根据旧的key和value计算新的值，输出 2a
System.out.println(map.get(1));

map.computeIfPresent(2, (key, value) -> null);
// 存在key为2， 根据旧的key和value计算得到null，删除该值，输出 null
System.out.println(map.get(2));

compute 方法

compute 方法是 computeIfAbsent 与 computeIfPresent 的综合体。

merge 方法

构造

1
merge(K key, V value, BiFunction<? super V, ? super V, ? extends V> remappingFunction)

如果指定的 key 不存在，则设置指定的 value 值，否则根据 key 的旧的值 oldvalue，value 计算出新的值 newValue, 如果 newValue 为 null, 则删除该 key，否则设置 key 的新值 newValue。类似如下代码：

1
2
3
4
5
6
7
V oldValue = map.get(key);
V newValue = (oldValue == null) ? value :
        remappingFunction.apply(oldValue, value);
if (newValue == null)
    map.remove(key);
else
    map.put(key, newValue);

示例代码如下：

1
2
3
4
5
6
// 存在key为1， 输出 a merge
System.out.println(map.merge(1, " merge", (oldValue, newValue) -> oldValue + newValue));
// 新值为null，删除key，输出 null
System.out.println(map.merge(1, " merge", (oldValue, newValue) -> null));
// 输出 " merge"
System.out.println(map.merge(4, " merge", (oldValue, newValue) -> oldValue + newValue));