使用Java Stream高效处理Map：按值排序并提取键名教程

使用Java Stream对Map进行值排序并提取键名

在Java开发中，我们经常会遇到需要对Map中的数据进行操作的场景，例如根据Map的值进行排序，然后提取对应的键。一个常见的挑战是，如何在排序过程中保留键与值的关联，以便在排序完成后能够获取到排序后的键名列表。

考虑一个场景：我们有一个存储城市及其人口数量的Map，现在需要根据人口数量从高到低对城市进行排序，并最终只输出城市名称。

初始的尝试可能如下：

import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

Map cities = new HashMap<>();
cities.put("Minsk", 1999234);
cities.put("Mogilev", 1599234);
cities.put("Vitebsk", 3999231);
cities.put("Brest", 4999234);

// 错误的尝试：先映射到值，再排序，导致键信息丢失
List collect = cities.entrySet().stream()
    .map(o -> o.getValue()) // 此时已经失去了城市名称（键）的信息
    .sorted(Comparator.reverseOrder())
    .collect(Collectors.toList());

System.out.println("仅按人口排序（丢失城市名）: " + collect);
// 预期输出：[4999234, 3999231, 1999234, 1599234]
// 但无法获取对应的城市名

上述代码的问题在于，在执行map(o -> o.getValue())操作时，Stream流中的元素已经从Map.Entry变为了Integer类型（即人口数量），此时城市名称（键）的信息已经丢失。后续的排序操作只能针对人口数量进行，而无法再关联回城市名称。

立即学习“Java免费学习笔记（深入）”；

正确的做法是，在排序之前不要丢失键与值的关联。我们可以直接对Map.Entry对象进行排序，然后根据排序后的Map.Entry提取键。Map.Entry接口提供了一个非常方便的静态方法comparingByValue()，结合Comparator.reverseOrder()可以轻松实现按值降序排序。

import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

Map cities = new HashMap<>();
cities.put("Minsk", 1999234);
cities.put("Mogilev", 1599234);
cities.put("Vitebsk", 3999231);
cities.put("Brest", 4999234);

// 正确的方法：先对Map.Entry进行排序，再提取键
List sortedCityNames = cities.entrySet().stream()
    .sorted(Map.Entry.comparingByValue(Comparator.reverseOrder())) // 按值降序排序Map.Entry
    .map(Map.Entry::getKey) // 提取排序后的Map.Entry的键
    .toList(); // 或者 .collect(Collectors.toList());

System.out.println("按人口降序排序的城市名: " + sortedCityNames);
// 预期输出：[Brest, Vitebsk, Minsk, Mogilev]

这段代码首先通过cities.entrySet().stream()获取一个包含Map.Entry对象的Stream。接着，sorted(Map.Entry.comparingByValue(Comparator.reverseOrder()))对这些Map.Entry对象进行排序，排序的依据是它们的值（人口数量），并且是降序排列。最后，map(Map.Entry::getKey)将排序后的Map.Entry对象转换成它们的键（城市名称），最终收集为一个List。

Fireflies.ai

自动化会议记录和笔记工具，可以帮助你的团队记录、转录、搜索和分析语音对话。

下载

优化数据模型：使用自定义类或记录 (Record)

尽管直接操作Map.Entry可以解决问题，但在实际项目中，将数据建模为独立的类或Java 14+引入的record通常是更好的实践。这种方式提供了更强的类型安全性、更好的可读性，并且能够更好地封装相关数据。

例如，我们可以定义一个City记录来表示城市及其人口：

// Java 14+ record
record City(String name, int population) {}

// 或者传统的类
/*
class City {
    private String name;
    private int population;

    public City(String name, int population) {
        this.name = name;
        this.population = population;
    }

    public String getName() { return name; }
    public int getPopulation() { return population; }

    // 建议重写equals, hashCode, toString
}
*/

有了City记录后，我们可以将城市数据存储在一个List中，而不是Map。这样，排序和提取操作会变得更加直观和类型安全。

import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

// 定义City记录
record City(String name, int population) {}

List citiesList = List.of(
    new City("Minsk", 1999234),
    new City("Mogilev", 1599234),
    new City("Vitebsk", 3999231),
    new City("Brest", 4999234)
);

// 使用自定义类进行排序和提取
List sortedCityNamesFromList = citiesList.stream()
    .sorted(Comparator.comparingInt(City::population).reversed()) // 按人口降序排序City对象
    .map(City::name) // 提取排序后的City对象的名称
    .toList(); // 或者 .collect(Collectors.toList());

System.out.println("使用City记录按人口降序排序的城市名: " + sortedCityNamesFromList);
// 预期输出：[Brest, Vitebsk, Minsk, Mogilev]

在这个例子中，Comparator.comparingInt(City::population)创建了一个Comparator，用于比较City对象的population属性。.reversed()方法则将排序顺序反转为降序。最后，map(City::name)提取了每个City对象的名称。这种方式不仅清晰地表达了业务逻辑，也使得代码更易于维护和扩展。

注意事项与最佳实践

1. 操作顺序至关重要：在使用Stream API时，操作的顺序会直接影响结果。在需要根据某个属性排序并提取另一个属性时，务必确保排序操作在映射（map）操作之前，以避免丢失关键信息。
2. 选择合适的比较器：
   • 对于Map.Entry，使用Map.Entry.comparingByValue()或Map.Entry.comparingByKey()非常方便。
   • 对于自定义对象，Comparator.comparing()（或comparingInt()、comparingLong()等基本类型优化版本）结合方法引用是首选。
   • 对于降序排序，在比较器之后链式调用.reversed()方法即可。
3. 数据模型优化：当数据具有多个相关属性时，优先考虑使用自定义类或Java 14+的record来封装数据，而不是仅仅依赖于Map。这会提高代码的类型安全性、可读性和可维护性，特别是在处理复杂业务逻辑时。
4. 不可变性：在Stream操作中，尽量使用不可变的数据结构和操作，这有助于避免副作用，并使代码更易于理解和调试。record类型天生就是不可变的，非常适合这种场景。

通过掌握这些技巧，您可以更高效、更优雅地利用Java Stream API处理各种数据排序和转换任务。