使用Java Stream对Map进行值排序并提取键名
在Java开发中，我们经常会遇到需要对Map中的数据进行操作的场景，例如根据Map的值进行排序，然后提取对应的键。一个常见的挑战是，如何在排序过程中保留键与值的关联，以便在排序完成后能够获取到排序后的键名列表。
考虑一个场景：我们有一个存储城市及其人口数量的Map<String, Integer>，现在需要根据人口数量从高到低对城市进行排序，并最终只输出城市名称。
初始的尝试可能如下：
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
Map<String, Integer> cities = new HashMap<>();
cities.put("Minsk", 1999234);
cities.put("Mogilev", 1599234);
cities.put("Vitebsk", 3999231);
cities.put("Brest", 4999234);
// 错误的尝试：先映射到值，再排序，导致键信息丢失
List<Integer> 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<String, Integer>变为了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<String, Integer> cities = new HashMap<>();
cities.put("Minsk", 1999234);
cities.put("Mogilev", 1599234);
cities.put("Vitebsk", 3999231);
cities.put("Brest", 4999234);
// 正确的方法：先对Map.Entry进行排序，再提取键
List<String> 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<String>。
优化数据模型：使用自定义类或记录 (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<City>中，而不是Map。这样，排序和提取操作会变得更加直观和类型安全。
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;
// 定义City记录
record City(String name, int population) {}
List<City> citiesList = List.of(
new City("Minsk", 1999234),
new City("Mogilev", 1599234),
new City("Vitebsk", 3999231),
new City("Brest", 4999234)
);
// 使用自定义类进行排序和提取
List<String> 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对象的名称。这种方式不仅清晰地表达了业务逻辑，也使得代码更易于维护和扩展。
注意事项与最佳实践
- 操作顺序至关重要：在使用Stream API时，操作的顺序会直接影响结果。在需要根据某个属性排序并提取另一个属性时，务必确保排序操作在映射（map）操作之前，以避免丢失关键信息。
-
选择合适的比较器：
- 对于Map.Entry，使用Map.Entry.comparingByValue()或Map.Entry.comparingByKey()非常方便。
- 对于自定义对象，Comparator.comparing()（或comparingInt()、comparingLong()等基本类型优化版本）结合方法引用是首选。
- 对于降序排序，在比较器之后链式调用.reversed()方法即可。
- 数据模型优化：当数据具有多个相关属性时，优先考虑使用自定义类或Java 14+的record来封装数据，而不是仅仅依赖于Map。这会提高代码的类型安全性、可读性和可维护性，特别是在处理复杂业务逻辑时。
- 不可变性：在Stream操作中，尽量使用不可变的数据结构和操作，这有助于避免副作用，并使代码更易于理解和调试。record类型天生就是不可变的，非常适合这种场景。
通过掌握这些技巧，您可以更高效、更优雅地利用Java Stream API处理各种数据排序和转换任务。
