使用流对内部对象的Map属性进行分组?

时间:2019-01-18 13:48:43

标签: java collections java-8 java-stream java-11

我正在学习Java 8-Java 11,并且得到了要转换为的代码。我有以下课程:

class Resource {
   List<Capability> capabilities;
}

class Capability {
   String namespace;
   Map<String, Object> attributes;
}

我有大量的资源,我想将其所有功能属性从两个不同的命名空间(“ a”,“ b”)提取到一个Map<Resource, Map<String, Object>>,我确定其中没有重复的键。 / p>

我使用map,flatMap进行了许多尝试,但是通过这些尝试,我无法保留对主要资源对象的引用。使用java9的新功能,我可以取得进步,但是我陷入了下面的代码中,在该代码中,我可以返回所有属性,但可以返回一组属性。 我还无法按功能名称空间进行过滤,也无法将它们放在地图中:

Map<Resource, Set<Object>> result = pResolved.stream()
    .collect(groupingBy(t -> t, flatMapping(
            resource -> resource.getCapabilities(null).stream(),
            flatMapping(
                    cap -> cap.getAttributes().entrySet().stream(),
                    toSet()))));

看来我走对了。

2 个答案:

答案 0 :(得分:6)

还有一种方法也仅使用方法:

Map<String, Set<Object>> result = pResolved.stream()                        
    .map(Resource::getCapabilities)                         // Stream<List<Capability>>
    .flatMap(List::stream)                                  // Stream<Capability>
    .collect(Collectors.toMap(                              // Map<String, Set<Object>>
        c -> c.getNamespace(),                              // Key: String (namespace)
        i -> new HashSet<>(i.getAttributes().values())));   // Value: Set of Map values

假设示例输入为:

Resource [capabilities=[
    Capability [namespace=a, attributes={a1=aa1, a2=aa2, a3=aa3}]]]
Resource [capabilities=[
    Capability [namespace=b, attributes={b2=bb2, b3=bb3, b1=bb1}], 
    Capability [namespace=c, attributes={c3=cc3, c1=cc1, c2=cc2}]]]

然后上面的代码将导致:

a: [aa1, aa3, aa2]
b: [bb1, bb3, bb2]
c: [cc1, cc3, cc2]

答案 1 :(得分:5)

您可以改用Collectors.toMap作为downstream

Map<Resource, Map<String, Object>> result = pResolved
        .stream()
        .collect(groupingBy(Function.identity(),
                flatMapping(resource -> resource.getCapabilities().stream(),
                        flatMapping(cap -> cap.getAttributes().entrySet().stream(),
                                toMap(Map.Entry::getKey, Map.Entry::getValue)))));