Java流如何将具有数量的对象解压缩到单个对象列表

Question

我有一个“项目”对象列表，如:

public class Item {
    private String name;
    private int qty;

    public Item() { }

    public Item(String name, int qty) {
        this.name = name;
        this.qty = qty;
    }

    public List<Item> unpack() {
        List<Item> items = new ArrayList<>();
        items.add(new Item("foo", 2));
        items.add(new Item("bar", 3));
        items.add(new Item("baz", 1));

        List<Item> unpackedItems = unpack(items);
        System.out.println(unpackedItems.size()); // it should be == 6
        return unpackedItems;
    }

    private List<Item> unpack(List<Item> items) {
        // ..
    }
}

有没有办法使用流“解包”这些对象，以便在数量大于 1 时最终重复项目列表，因此最后我将拥有数量等于 1 的对象？

Answer 1

您可以基于 qty 展平嵌套流(这基本上创建了一个 qty 大小的 int 流，然后映射到一个新的 Item，每个元素带有 qty = 1，重用名称):

private List<Item> unpack(List<Item> items) {
    return items.stream()
            .flatMap(item -> IntStream.range(0, item.qty)
                               .mapToObj(i -> new Item(item.name, 1)))
            .collect(Collectors.toList());
}