通过lambda从列表构建地图

Question

我有一个Group模型，其中包含一个内部字段List

如何通过lambda和流将以下代码更改为一行

 Map<String, String> entityGroup = new HashMap<>();
 groups.forEach(g -> g.getEntities()
                .forEach(e -> entityGroup.put(e.getKey(), g.getKey()))
 );

内部列表中的每个实体都应该是映射中的键，值应该是组本身

谢谢




Answer 1

这是一个可运行的示例，不需要您像上面的解决方案那样显式创建新条目：

import java.util.*;
import java.util.stream.*;

public class MyClass {
    public static void main(String args[]) {
        List<B> b1 = List.of(new B("a", "1"), new B("b", "2"));
        List<B> b2 = List.of(new B("x", "3"), new B("y", "4"));
        A a1 = new A(b1);
        A a2 = new A(b2);
        List<A> allAs = List.of(a1, a2);
        Map<String, String> map =
               allAs.stream().flatMap(a -> a.vals.stream())
               .collect(Collectors.toMap(B::getKey, B::getValue));
        System.out.println(map); // {a=1, b=2, x=3, y=4}
    }
}

class A {
    List<B> vals;
    public A(List<B> vals) { this.vals = vals; }
}

class B {
    String k, v;
    public B(String k, String v) { this.k = k; this.v = v; }
    public String getKey() { return this.k; }
    public String getValue() { return this.v; }
}

Answer 2

匿名用户

要使用streams实现这一点，您需要创建一个map-entries流，然后将它收集到一个< code>Map中。

我在我的示例中使用了 Map.entry，它在 Java 9 及更高版本中可用。如果你使用的是Java 8，你可以使用AbstractMap.SimpleEntry或其他类型的Pair类。

大概是这样的:

java prettyprint-override">Map<String, String> entityGroup = groups.stream()
    .flatMap(group -> group.getEntities().stream()
        .map(entity -> Map.entry(entity.getKey(), group.getKey())))
    .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));