如何用Java Collectors实现单对象多字段映射为多键Map？

你的需求确实不是普通的Collectors.groupingBy能直接解决的——因为普通分组是按单个字段把元素映射到一个键，而这里每个Person需要同时贡献两个键（firstName和lastName），每个键都要关联当前Person。下面提供两种优雅的实现方式，都基于Stream和Collectors：

方法一：用flatMap拆分键值对（最简洁） #

核心思路是先把每个Person拆成两个键值对（firstName -> Person 和 lastName -> Person），再对这些键值对进行分组。这种方式不需要自定义Collector，代码直观易懂：

import java.util.AbstractMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;

// 测试数据初始化（你的代码不变）
Person person1 = new Person();
person1.setFirstName("john");
person1.setLastName("doe");
person1.setUserId("user1");
Person person2 = new Person();
person2.setFirstName("doe");
person2.setLastName("frank");
person2.setUserId("user2");
Person person3 = new Person();
person3.setFirstName("john");
person3.setLastName("wayne");
person3.setUserId("user3");
List<Person> personList = List.of(person1, person2, person3);

// 核心实现
Map<String, List<Person>> mapPersons = personList.stream()
    .flatMap(person -> Stream.of(
        new AbstractMap.SimpleEntry<>(person.getFirstName(), person),
        new AbstractMap.SimpleEntry<>(person.getLastName(), person)
    ))
    .collect(Collectors.groupingBy(
        Map.Entry::getKey,
        Collectors.mapping(Map.Entry::getValue, Collectors.toList())
    ));

// 打印验证（和预期输出一致）
mapPersons.forEach((key, value) -> System.out.print(key + "=" + value + ", "));

代码解释： #

1. flatMap：把每个Person转换成包含两个Entry的流，这样每个Person会被"展开"成两个键值对。
2. Collectors.groupingBy：按Entry的key（即姓名）分组，再用Collectors.mapping把Entry的value（即Person对象）收集成列表。

方法二：自定义Collector（更灵活） #

如果你需要更精细的控制（比如并行流优化、自定义中间逻辑），可以自定义一个Collector。这个Collector的逻辑和你原来的循环非常相似，只是封装成了可复用的Collector：

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collector;

// 自定义Collector
Collector<Person, ?, Map<String, List<Person>>> nameGroupingCollector = Collector.of(
    // Supplier: 初始化存储结果的HashMap
    HashMap::new,
    // Accumulator: 处理每个Person，把firstName和lastName分别加入Map
    (map, person) -> {
        // 如果键不存在则创建新列表，再添加当前Person
        map.computeIfAbsent(person.getFirstName(), k -> new ArrayList<>()).add(person);
        map.computeIfAbsent(person.getLastName(), k -> new ArrayList<>()).add(person);
    },
    // Combiner: 合并两个Map（用于并行流场景）
    (map1, map2) -> {
        map2.forEach((key, persons) -> 
            map1.merge(key, persons, (existingList, newList) -> {
                existingList.addAll(newList);
                return existingList;
            })
        );
        return map1;
    }
);

// 使用自定义Collector
Map<String, List<Person>> mapPersons = personList.stream().collect(nameGroupingCollector);

代码解释： #

• Supplier：创建一个空的HashMap作为结果容器。
• Accumulator：对每个Person，用computeIfAbsent简化原来的if-else判断，自动处理列表的创建和元素添加。
• Combiner：当使用并行流时，负责把多个子Map合并成最终的结果Map，用merge方法合并相同键的列表。

额外优化：简化你的原始循环代码 #

虽然你想要用Collectors，但顺便提一下：原始循环可以用computeIfAbsent大幅简化，不需要嵌套的if-else：

Map<String, List<Person>> mapPersons = new HashMap<>();
for (Person p : personList) {
    mapPersons.computeIfAbsent(p.getFirstName(), k -> new ArrayList<>()).add(p);
    mapPersons.computeIfAbsent(p.getLastName(), k -> new ArrayList<>()).add(p);
}

以上两种Collectors实现都能得到你预期的输出，其中方法一更适合大多数场景，代码简洁易维护。

内容的提问来源于stack exchange，提问作者krakig