如何分组到数组映射中？

Question

流上的groupingBy操作是否可以生成一个映射，其中值是数组而不是列表或其他集合类型？

例如：我有一个类Thing。事物有所有者，所以Thing有一个getOwnerId方法。在一个事物流中，我想按所有者ID对事物进行分组，以便具有相同所有者ID的事物最终在一个数组中。换句话说，我想要一个像下面这样的映射，其中键是所有者ID，值是属于该所有者的事物数组。

    Map<String, Thing[]> mapOfArrays;

在我的例子中，因为我需要将映射值传递给一个需要数组的库方法，所以收集到一个< code >映射中是最方便的

将整个流收集到一个数组中很容易（甚至不需要显式收集器）：

    Thing[] arrayOfThings = Stream.of(new Thing("owner1"), new Thing("owner2"), new Thing("owner1"))
            .toArray(Thing[]::new);

[属于所有者1，属于所有者2，属于所有者1]

通过所有者ID进行摸索也很容易。例如，要分组到列表中：

    Map<String, List<Thing>> mapOfLists = Stream.of(new Thing("owner1"), new Thing("owner2"), new Thing("owner1"))
            .collect(Collectors.groupingBy(Thing::getOwnerId));

{owner1=[屬於 owner1， 屬於 owner1]， owner2=[屬於 owner2]}

只有这个例子给了我一个列表图。有2-arg和3-arg groupingBy方法可以提供其他集合类型(如集合)的映射。我想，如果我可以将收集到数组中的收集器(类似于上面第一个代码片段中收集到数组中的收集器)传递给two-arg Collectors

为了一个完整的示例，以下是我在上面片段中使用的类：

public class Thing {

    private String ownerId;

    public Thing(String ownerId) {
        this.ownerId = ownerId;
    }

    public String getOwnerId() {
        return ownerId;
    }

    @Override
    public String toString() {
        return "Belongs to " + ownerId;
    }

}




Answer 1

可能很明显，但您可以通过以下方式完成：

Stream.of(new Thing("owner1"), new Thing("owner2"), new Thing("owner1"))
            .collect(Collectors.toMap(
                    Thing::getOwnerId,
                    x -> new Thing[]{x},
                    (left, right) -> {
                        Thing[] newA = new Thing[left.length + right.length];
                        System.arraycopy(left, 0, newA, 0, left.length);
                        System.arraycopy(right, 0, newA, left.length, right.length);
                        return newA;
                    }
            ))

Answer 2

您可以先分组，然后使用后续的toMap：

Map<String, Thing[]> result = source.stream()
                .collect(groupingBy(Thing::getOwnerId))
                .entrySet()
                .stream()
                .collect(toMap(Map.Entry::getKey,
                        e -> e.getValue().toArray(new Thing[0])));

Answer 3

匿名用户

使用Thomas Pliakas的这个答案中的收集器：

    Map<String, Thing[]> mapOfArrays = Stream.of(new Thing("owner1"), new Thing("owner2"), new Thing("owner1"))
            .collect(Collectors.groupingBy(Thing::getOwnerId,
                    Collectors.collectingAndThen(Collectors.toList(),
                            tl -> tl.toArray(new Thing[0]))));

想法是首先收集到一个列表中(这是一个明显的想法，因为数组有固定的大小)，然后在返回到collector分组之前转换为一个数组。< code >收集然后可以通过其所谓的整理器来完成。

打印结果以供检查：

    mapOfArrays.forEach((k, v) -> System.out.println(k + '=' + Arrays.toString(v)));

owner1=[Belongs to owner1, Belongs to owner1]
owner2=[Belongs to owner2]

编辑:感谢Aomine的链接:使用< code>new Thing[0]作为< code>toArray的参数是受古人智慧阵列的启发。似乎在英特尔CPU上使用< code>new Thing[0]比使用< code>new Thing[tl.size()]要快。我很惊讶。