使用 Java Lambda（如 Map<key1, Map<key2, List<Obj>>>）时如何在 group-2 中使用 group-1 结果 key1

Question

Java Lambda 如何使用 group-1 生成 group-2。

这个问题是我解决的问题的更新： How to covert List to Map<T, <K, List<Person>>> use java lambda?

你可以看到最后的代码。 function key2 想使用 function key1 的结果，怎么办？

class Person{
   int age;
   int cityCode;
   String name;
}

method:

// Map<age, Map<cityCode, List<Person>>>
public Map<Integer, Map<Integer, List<Person>>> covertListToMap(List<Person> list){

      // TODO: How to make List<Person> to Map<age, Map<cityCode, List<Person>>>

}

Function<Person, Integer> key1 = (Person p) -> {
            // do many sth then get a key1Result:
            int key1Result = p.getAge() * new Random(10).nextInt();
            return key1Result;
        };


Function<Person, Integer> key2 = (Person p) -> {
            //  Question: how to get and use Function key1 key1Result:

            int key1Result = 0;

            int result = p.getCityCode() + key1Result;
            return result;
        };

Map<Integer, Map<Integer, List<Person>>> collect = list.stream().collect(groupingBy(key1, groupingBy(key2)));

我写错了样例来说明我的意思：

Map<Integer, Map<Integer, List<Person>>> collect = list.stream().collect(groupingBy((Person p) -> {
            int key1Result = p.getAge() * new Random(10).nextInt();
            return key1Result;  // perhaps, now, key1Result is 10
        }, groupingBy((Person p) -> {

            // now, I want to get the key1Result which is 10, How to do it?  
            int key1Result = 0;

            int result = p.getCityCode() + key1Result;
            return result;
        })));

Answer 1

您可以使用函数apply(T t)。根据javadoc：

将此函数应用于给定参数。

在你的情况下，这意味着：

int key1Result = key1.apply(p)

编辑：当你修改你的问题时，我添加了一个答案，如果我理解你想要做什么应该是正确的：

AtomicInteger atomicInteger = new AtomicInteger(0);
Map<Integer, Map<Integer, List<Person>>> collect = personList.stream().collect(Collectors
        .groupingBy(p -> {
                int age = p.getAge() * new Random(10).nextInt();
                atomicInteger.set(age);
                return age;
            }, Collectors.groupingBy(p2 -> p2.getCityCode() + atomicInteger.get())));

如果您想了解更多关于AtomicInteger 的信息，请参阅atomic。基本上它是一个包装器，它将在线程之间保持其变量。

Answer 2

首先，您使用修改后的密钥收集所需的地图

Map<Integer, Map<Integer, List<Person>>> collect2 = personList.stream().collect(Collectors
        .groupingBy(p -> p.getAge() * new Random(10).nextInt(),
                Collectors.groupingBy(Person::getCityCode)));

然后改变地图。这种情况下的问题是您使用的是嵌套地图。并使用您想要change the key of inner Map 的外部地图的key。所以除了为每个外部 Key 创建一个新的内部 Map 之外别无他法。

在这里，我正在迭代外部映射的每个条目，并使用已经存在的内部映射创建一个新的正确（必需）内部映射。

collect2.entrySet().stream().forEach(e -> {
    Integer key = e.getKey();
    Map<Integer, List<Person>> oldValueMap = e.getValue();
    Map<Integer, List<Person>> newValueMap = new HashMap<>();
    oldValueMap.keySet().forEach(k -> {
        newValueMap.put(k + key, oldValueMap.get(k));
    });
    e.setValue(newValueMap);
});