如何按 Java List 中对象的属性进行分组？

Question

我有一个资源对象列表。资源对象定义为：

class Resource {
    String ip;
    String subnet;
    // other properties...
}

我需要一个像地图这样的数据结构，其中键是一个子网中的随机资源，值是其他子网中所有资源的列表。

例如，如果S 表示子网，R 表示资源，则列表有：

S1R1, S1R2, S2R1, S2R2, S3R1 and S3R2

预期输出是：

S1R1 -> S2R1, S2R2, S3R1, S3R2
S2R2 -> S1R1, S1R2, S3R1, S3R2
S3R1 -> S1R1, S1R2, S2R1, S2R2

我尝试使用流并收集基于子网的分组，不知道如何从这里开始。

Map<String, Set<Resource>> grouped = resourceList.stream()
    .collect(Collectors.groupingBy(Resource::getSubnet, Collectors.toSet()));

Answer 1

我不确定子网是如何工作的，但你必须弄清楚下面的关系（isSubNetOf）。

BiPredicate<Resource, Resource> isSubNetOf = (parent, subnet) -> ???;

Map<Resource, Set<Resource>> resourceToSubnets = resourceList.stream()
        .collect(toMap(
                Function.identity(),
                x -> resourceList.stream().filter(v -> isSubNetOf.test(x, v)).collect(toSet())
                ));

如果您想要Map<String, Set<Resource>>，请将Function.identity() 更改为所需的Resource::getIp 或Resource::getSubnet。

Answer 2

这不适用于 groupingBy 收集器，因为它通过将属于一个组的元素传递给每个组的下游收集器来工作，而您不能告诉它传递其他的。

实现您所描述的一种方法是

TreeMap<Resource, Set<Resource>> map
                              = new TreeMap<>(Comparator.comparing(Resource::getSubnet));
for(Resource resource: resourceList)
    map.computeIfAbsent(resource, x -> new HashSet<>(resourceList)).remove(resource);

map.forEach((resource,set) -> System.out.println(resource+" -> "
    +set.stream().map(Resource::toString).collect(Collectors.joining(", "))));

鉴于列表S1R1, S1R2, S2R1, S2R2, S3R1, S3R2 将产生

S1R1 -> S3R2, S2R1, S2R2, S3R1
S2R1 -> S3R2, S1R2, S3R1, S1R1
S3R1 -> S1R2, S2R1, S2R2, S1R1

虽然Set 的元素顺序未定义，但Map 的键实际上不是随机选择的，它是每个子网的第一个遇到的资源。

您可以通过两个链式流操作来做到这一点

Map<Resource, Set<Resource>> map = resourceList.stream()
    .collect(Collectors.groupingBy(Resource::getSubnet, Collectors.toSet()))
    .values().stream()
    .collect(Collectors.toMap(
        set -> set.iterator().next(),
        set -> {
            Set<Resource> other = new HashSet<>(resourceList);
            other.removeAll(set);
            return other;
        }));

这里，从每个组中挑选的资源确实是未指定的。与基于TreeMap 的方法不同，您必须知道实际的Resource（或全部尝试）才能查找组。 TreeMap 允许使用同一子网中的任何 Resource 查找组。