java stream 多字段分组

### 使用Java Stream API 实现多字段分组 在Java中，`Stream API` 提供了一种强大的方式来处理集合数据。然而，与Scala中的 `groupBy` 方法不同的是，Java 的 `Collectors.groupingBy()` 只支持单字段分组。为了实现基于多个字段的分组操作，可以组合使用自定义键对象或者通过嵌套分组的方式完成。 以下是具体方法： #### 自定义复合键类 可以通过创建一个包含所需字段的复合键类来进行分组。这个类需要重写 `equals()` 和 `hashCode()` 方法以确保正确比较和散列行为。 ```java import java.util.*; import java.util.stream.Collectors; class MyClass { private final String a; private final int b; private final double c; public MyClass(String a, int b, double c) { this.a = a; this.b = b; this.c = c; } public String getA() { return a; } public int getB() { return b; } @Override public String toString() { return "MyClass{" + "a='" + a + '\'' + ", b=" + b + ", c=" + c + '}'; } } // 复合键类 class CompositeKey { private final String a; private final int b; public CompositeKey(String a, int b) { this.a = a; this.b = b; } @Override public boolean equals(Object o) { if (this == o) return true; if (!(o instanceof CompositeKey)) return false; CompositeKey that = (CompositeKey) o; return b == that.b && Objects.equals(a, that.a); } @Override public int hashCode() { return Objects.hash(a, b); } } ``` 接着，在流上应用此复合键进行分组： ```java List<MyClass> data = Arrays.asList( new MyClass("key1", 10, 1.1), new MyClass("key2", 20, 2.2), new MyClass("key1", 10, 3.3), new MyClass("key2", 20, 4.4) ); Map<CompositeKey, List<MyClass>> groupedData = data.stream() .collect(Collectors.groupingBy(item -> new CompositeKey(item.getA(), item.getB()))); groupedData.forEach((key, value) -> System.out.println(key + ": " + value)); ``` 上述代码实现了按照两个字段 (`a`, `b`) 进行分组的功能[^1]。 #### 嵌套分组 另一种方法是利用嵌套调用 `Collectors.groupingBy()` 来逐层分组。这种方法适用于更复杂的场景，但可能不如单一层次结构直观。 ```java Map<String, Map<Integer, List<MyClass>>> nestedGroupedData = data.stream() .collect(Collectors.groupingBy(MyClass::getA, Collectors.groupingBy(MyClass::getB))); nestedGroupedData.forEach((outerKey, innerMap) -> innerMap.forEach((innerKey, values) -> System.out.println(outerKey + ":" + innerKey + "->" + values))); ``` 这种方式会先按第一个字段分组，再对每个子集按第二个字段进一步分组[^2]。 --- ### 总结 以上两种方法都可以用于解决多字段分组问题。如果希望保持简单明了的数据结构，则推荐使用 **自定义复合键**；而当需求涉及多层次逻辑时，可以选择 **嵌套分组** 方式。