使用流將列表的收集與單個值組合

Question

我有以下數據：

 CustomClass first = new CustomClass(1, Arrays.asList(1, 2, 3));
 CustomClass second  = new CustomClass(5, Arrays.asList(6,7,8));
 List<CustomClass> list = Arrays.asList(first, second);

現在我想將這些數據收集到一個列表中，該列表包含CustomClass中的值和列表中的值。

我想得到這個： 1, 1, 2, 3, 5, 6, 7, 8

這是我實施的：

list.stream().forEach(item -> {
        result.add(item.value);
        item.values.stream().forEach(seconditem -> result.add(seconditem));
    });

我知道有一個收集方法的流，但我不明白如何使用它來收集同一個類的一個值和值列表。

Answer 1

您可以使用Stream.concat組合單個值的Stream.of和其他值的流：

List<Integer> result = list.stream()
        .flatMap(x -> Stream.concat(Stream.of(x.value), x.values.stream()))
        .collect(Collectors.toList());