列表映射的Java流映射

Question

我有一个实体，看起来像这样：

public class Snippet {
    private Integer docId;
    private Integer page;
    private Payload payload;
}

输入数据是一个列表

我需要创建一个索引，允许我们迭代docid和页面，并获取相关的Snippet对象。

这样的数据结构：

地图

我可以使用Java流来获取地图

        List<Snippet> input = ....;
        input.stream()
        .collect(Collectors.groupingBy(
                    Snippet::getDocId, 
                    Collectors.toMap(Snippet::getPage, Function.identity())
                 )
        );

如何收集以获取列表作为最终地图值？




Answer 1

以下是另一个供您当前或未来考虑的备选方案。它使用地图界面的computeIfAbsent方法。

• 创建最终的内容映射
• 反复浏览代码段列表
• 如果DocId不存在，则使用DocId键创建条目。然后尝试存储页面和片段
• 如果页面不存在，则使用页面作为键创建条目。然后将代码段存储在列表中

下面是一个使用记录代替类的示例。

record PayLoad(String getValue) {
    @Override
    public String toString() {
        return "[" + getValue() + "]";
    }
}

record Snippet(int getDocId, int getPage,
        PayLoad getPayLoad) {
    @Override
    public String toString() {
        return String.format("{%s, %s, %s}", getDocId,
                getPage, getPayLoad);
    }
}

创建一些数据

List<Snippet> input =
        List.of(new Snippet(1, 1, new PayLoad("A")),
                new Snippet(1, 2, new PayLoad("B")),
                new Snippet(1, 3, new PayLoad("C")),
                new Snippet(2, 1, new PayLoad("D")),
                new Snippet(2, 2, new PayLoad("E")),
                new Snippet(2, 3, new PayLoad("F")));

存储代码片段

Map<Integer, Map<Integer, List<Snippet>>> map =
        new HashMap<>();

for (Snippet s : input) {
    map.computeIfAbsent(s.getDocId(), v -> new HashMap<>())
            .computeIfAbsent(s.getPage(),
                    v -> new ArrayList<>())
            .add(s);
}

打印出来

map.forEach((k, v) -> {
    System.out.println(k);
    v.forEach((kk, vv) -> System.out
            .println("          " + kk + " -> " + vv));
});

印刷品

1
          1 -> [{1, 1, [A]}]
          2 -> [{1, 2, [B]}]
          3 -> [{1, 3, [C]}]
2
          1 -> [{2, 1, [D]}]
          2 -> [{2, 2, [E]}]
          3 -> [{2, 3, [F]}]

Answer 2

如您所见，收集器。分组依据（…） 获取收集器作为第二个参数。因此，只需使用收集器即可。分组依据（…）作为收集器的第二个参数。分组依据（…）

input.stream().collect(Collectors.groupingBy(
                        Snippet::getDocId,
                        Collectors.groupingBy(Snippet::getPage,
                                Collectors.toList())));