如何使用 java 流对子列表中的元素进行分组和计数

Question

我想使用 java 流对子列表中的元素进行分组和计数。

例如，我有一个 AnswerWithOneCorrectOption 类型的答案，如下所示：

class AnswerWithOneCorrectOption {
     Long choiceId;
}

此答案类型只有一个正确选项，存储在“AnswerWithOneCorrectOption.id”中。我正在通过 AnswerWithOneCorrectOption 的列表进行流式传输，根据 id 进行分组并使用以下方法进行计数：

private Map<Long, Long> countChoicesAndGroup(List<AnswerWithOneCorrectOption> answers){

Map<Long, Long> map = answers.parallelStream()
             .collect(Collectors.groupingBy(AnswerWithOneCorrectOption::getChoiceId, 
 Collectors.counting())); 

 return map;
}

假设我有另一个可以有多个正确选项的答案类型。我将这些选项保存在List<Long> choiceIds 中。

class AnswerWithMultipleCorrectOptions {
     List<Long> choiceIds;
}

如何按List<Long> choiceIds 中的choiceId 分组并计数？

Answer 1

如果用户只选择了一个选项，它将被保存在 answer.id 中。如果他选择了多个答案，我会将其添加到列表 answer.ids 中。

最好只使用Answer 和List<Long> ids。如果用户只选择一个选项，您将只有一个元素的列表。它允许您按答案分组（不要忘记equals/hashcode）两种情况：

Map<Answer, Long> collect = answers.stream()
        .collect(groupingBy(Function.identity(), counting()));

但如果您想按List<Long> 分组，也可以使用相同的方式：

Map<List<Long>, Long> collect = answers.stream()
            .collect(groupingBy(Answer::choiceIds, counting()));

更新：按子列表中的元素分组，您可以在之前使用flatMap：

Map<Long, Long> map = answers.stream()
        .flatMap(answer -> answer.getIds().stream())
        .collect(groupingBy(Function.identity(), counting()));