JavaMap单行操作技巧

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《Java Map 单行操作：快速创建与更新条目》，聊聊，我们一起来看看吧！

本文探讨了如何在Java中实现类似C++ `dict[key]+=1` 的Map单行条目创建与更新操作。针对传统 `put` 方法的局限性，重点介绍了 `Map` 接口提供的 `merge` 方法，并通过示例代码详细阐述其工作原理、参数使用及应用场景，帮助开发者实现更简洁、高效的Map数据处理。同时，也简要提及了其他相关辅助方法，以提供更全面的解决方案。

在C++等语言中，对 map 或 dictionary 进行条目更新或创建时，通常可以通过 dict[key] += 1 这样的简洁语法实现。这种方式的优势在于，它无需显式检查 key 是否存在，即可实现值的累加或初始化。然而，在Java的 Map 接口中，直接使用 put 方法结合 get 进行此类操作往往需要额外的条件判断，例如：

Map<String, Integer> wordCounts = new HashMap<>();
String key = "example";

// 传统方式：需要检查键是否存在
if (wordCounts.containsKey(key)) {
    wordCounts.put(key, wordCounts.get(key) + 1);
} else {
    wordCounts.put(key, 1);
}

这种模式在代码中会显得冗长，尤其是在需要频繁进行此类操作时。为了解决这一问题，Java 8及更高版本为 Map 接口引入了更强大、更简洁的方法，其中最常用且最适合上述场景的是 merge() 方法。

使用 Map.merge() 实现单行条目创建与更新

Map.merge() 方法旨在提供一种原子性操作，用于在给定键存在时合并其值，或在键不存在时插入一个新值。其方法签名如下：

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

该方法接受三个参数：

1. key：要操作的键。
2. value：如果键不存在，则将其作为新值插入。如果键已存在，此值将作为 remappingFunction 的第二个参数。
3. remappingFunction：一个 BiFunction 函数，用于定义当键已存在时如何合并旧值和新值。它接受旧值和传入的新值作为参数，并返回合并后的新值。

让我们以上述单词计数为例，使用 merge() 方法实现同样的功能：

import java.util.HashMap;
import java.util.Map;
import java.util.function.BiFunction;

public class MapMergeExample {
    public static void main(String[] args) {
        Map<String, Integer> wordCounts = new HashMap<>();

        String key1 = "apple";
        String key2 = "banana";
        String key3 = "apple"; // 再次出现

        // 第一次遇到 "apple"，键不存在，插入值 1
        wordCounts.merge(key1, 1, (oldValue, newValue) -> oldValue + newValue);
        System.out.println("After first 'apple': " + wordCounts); // {apple=1}

        // 第一次遇到 "banana"，键不存在，插入值 1
        wordCounts.merge(key2, 1, (oldValue, newValue) -> oldValue + newValue);
        System.out.println("After 'banana': " + wordCounts); // {apple=1, banana=1}

        // 再次遇到 "apple"，键已存在，执行 remappingFunction (1 + 1 = 2)
        wordCounts.merge(key3, 1, (oldValue, newValue) -> oldValue + newValue);
        System.out.println("After second 'apple': " + wordCounts); // {apple=2, banana=1}

        // 更简洁的单行写法
        wordCounts.merge("orange", 1, Integer::sum);
        System.out.println("After 'orange': " + wordCounts); // {apple=2, banana=1, orange=1}
    }
}

在上述示例中：

• 当 key (apple 或 banana) 第一次出现时，它不存在于 wordCounts 中。merge 方法会直接将 value (即 1) 插入到 Map 中。
• 当 key (apple) 再次出现时，它已存在。此时，merge 方法会调用 remappingFunction。oldValue 是 apple 当前的值 (1)，newValue 是我们传入的 1。remappingFunction ((oldValue, newValue) -> oldValue + newValue) 将两者相加，返回 2，然后 Map 中 apple 的值被更新为 2。
• Integer::sum 是一个方法引用，等同于 (oldValue, newValue) -> oldValue + newValue，提供了更简洁的写法。

其他相关方法

除了 merge() 之外，Java Map 还提供了一些其他用于处理键值对的便捷方法，它们在特定场景下也非常有用：

• computeIfAbsent(K key, Function mappingFunction): 如果指定的键不存在或其值为 null，则尝试使用 mappingFunction 计算其值并将其放入此映射中。这个方法非常适合用于缓存或懒加载场景，确保某个键只被计算一次。

  Map<String, String> cache = new HashMap<>();
  String result = cache.computeIfAbsent("dataKey", k -> fetchDataFromDatabase(k));
  // 如果 "dataKey" 不存在，则调用 fetchDataFromDatabase 并存储结果。
  // 如果存在，则直接返回现有值。

• computeIfPresent(K key, BiFunction remappingFunction): 如果指定的键存在且其值不为 null，则尝试使用 remappingFunction 计算其新值。如果函数返回 null，则删除该条目。

  Map<String, Integer> scores = new HashMap<>();
  scores.put("Alice", 100);
  scores.computeIfPresent("Alice", (k, v) -> v + 10); // Alice分数变为110
  scores.computeIfPresent("Bob", (k, v) -> v + 10);   // Bob不存在，不做任何操作

• putIfAbsent(K key, V value): 如果指定的键尚未与值关联（或映射到 null），则将其与给定值关联并返回 null。否则，返回当前关联的值。这是最简单的条件插入方法。

  Map<String, String> settings = new HashMap<>();
  settings.putIfAbsent("theme", "dark"); // 如果theme不存在，则设置为"dark"
  settings.putIfAbsent("theme", "light"); // theme已存在，不做任何改变

注意事项与总结

• 原子性：Map.merge() 和 computeIfAbsent() 等方法在 ConcurrentHashMap 中是原子操作，这对于多线程环境下的并发编程至关重要。在普通 HashMap 中，它们也是单次操作，避免了手动 get 和 put 可能导致的竞态条件。
• 可读性与简洁性：使用这些方法可以显著减少代码量，提高代码的可读性和维护性，避免了繁琐的 if-else 逻辑。
• 空值处理：merge() 方法中，如果 remappingFunction 返回 null，则对应的键值对将被从 Map 中移除。如果 value 参数为 null 且键不存在，则会抛出 NullPointerException，除非 remappingFunction 能够处理 null。
• 性能：这些方法通常比手动 containsKey 结合 get/put 更高效，因为它们避免了两次查找操作。

综上所述，Java 8引入的 Map.merge() 方法为开发者提供了一种优雅、高效且原子性的方式来处理Map中条目的创建与更新，尤其是在需要根据现有值进行计算或累加的场景中。结合 computeIfAbsent() 等其他辅助方法，开发者可以更灵活、更简洁地编写Map操作逻辑。

好了，本文到此结束，带大家了解了《JavaMap单行操作技巧》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多文章知识！