HashMap computeIfPresent(key, BiFunction) 方法在Java中的使用示例

在Java开发中，我们经常会使用到HashMap这个类，它提供了一个非常方便的方法，那就是computeIfPresent(key, BiFunction)方法。该方法可以用来检索key对应的value值，并在value值存在的情况下，对其进行操作并返回新的value值。

什么是BiFunction

BiFunction是一个函数式接口，其定义如下：

@FunctionalInterface
public interface BiFunction<T, U, R> {
    R apply(T t, U u);
}

定义了两个参数T和U以及返回值R，apply方法将接收两个参数T和U，并返回一个参数R。在HashMap的computeIfPresent方法中，BiFunction被用作操作已存在的value值并返回新的value值。

HashMap computeIfPresent方法的签名

HashMap computeIfPresent方法的签名如下：

default V computeIfPresent(K key, BiFunction<? super K, ? super V, ? extends V> remappingFunction)

它接收两个参数，一个是key，另一个是用于操作value的BiFunction函数式接口。如果key不存在，则返回null。如果key存在，则使用BiFunction对其value进行操作并返回新的value值。

下面是一个简单的示例，帮助我们更好地理解computeIfPresent方法的实现。

Map<String, Integer> map = new HashMap<>();

// 添加一些数据到map中
map.put("key1", 1);
map.put("key2", 2);
map.put("key3", 3);

// 在key1对应的value存在的情况下，将其值加10
int result = map.computeIfPresent("key1", (key, value) -> value + 10);

// 输出加10后的value
System.out.println(result);    // 输出11

在上面的例子中，map中的key为“key1”的value存在，所以BiFunction会将其value加10，并返回新的value值11。

HashMap computeIfPresent方法的使用场景

HashMap computeIfPresent方法的使用场景非常广泛，尤其在多线程并发操作的情况下非常有用。下面再看一个更加复杂的例子：

Map<String, List<String>> map = new HashMap<>();

// 给map添加一些数据
List<String> list1 = new ArrayList<>();
list1.add("value1");
list1.add("value2");

map.put("key1", list1);

// 给map添加一些数据
List<String> list2 = new ArrayList<>();
list2.add("value3");

map.put("key2", list2);

// 在key1对应的value存在的情况下，将其值加10
List<String> result = map.computeIfPresent("key1", (key, value) -> {
    // 在value中添加一些新的元素
    value.add("value3");
    value.add("value4");
    value.add("value5");
    return value;
});

// 输出操作后的value
System.out.println(result);    // 输出[value1, value2, value3, value4, value5]

在上面的这个例子中，我们给key1对应的value中添加了三个新的元素。如果key1在map中不存在，computeIfPresent则会返回null。