使用 Java Streams 组装列表

Question

我有一个名为“文件”的实体，具有以下属性：

public class File {
    private int id;
    private String proposal;
    private String hash;
    private String path;
    private LocalDateTime createdAt;
    private LocalDateTime finishedAt;
    private int size;
    private boolean processed;
}

我需要组织一个队列，将哪些文件上传到特定服务。对于每次提交，我可以发送尽可能多的文件，只要它们的大小总和不超过 100mb。此外，我必须尊重每个文件允许的最大日期（在这种情况下涉及finishedAt 属性）。

我已经设法对我的列表进行排序。换句话说，我能够确定首先上传哪些文件：

files.sort(Comparator.comparing(File::getFinishedAt));
files.stream().forEach(file -> System.out.println(file.getId()));

现在我想组装一个数组，其中包含每次提交时将排序的文件列表。类似的东西：

[[1, 4, 7], [8, 2], [6, 3], [5]]

上述数组的每个“子集”都与上传有关。上面是我的文件的 ID。所以我希望我的程序返回一个数组数组，其中每个集合都包含尽可能多的文件（只要它们的总和不超过 100mb）。由于每条记录都有截止日期 (finishedAt)，因此它们也需要按顺序排列。

最合适的方法是什么？我试图理解 Java 中的 Streams，但我能做的最多就是根据日期对列表进行排序。使用 Streams 我可以达到我展示的结果吗？实现这些目标最合适的方法是什么？

非常感谢！

Answer 1

Stream API 有其自身的局限性，可能不适用于给定任务中发生的有状态操作：此处应收集子列表（或需要增加适当的索引）直到满足大小限制，并且 那么计算应该从最后处理的索引计算。也就是说，有两个状态参数：总和和定义子列表的第一/最后索引对。

因此，可能会建议使用旧的基于循环的良好解决方案。

更新

添加了返回 int[][] 数组和 ID。

static int[][] groupFiles(List<File> files) {
    files.sort(Comparator.comparing(File::getFinishedAt));
    
    List<int[]> result = new ArrayList<>();
    
    for (int i = 0, n = files.size(); i < n; i++) {
        int sum = 0;
        int j = i;
        for (; j < n && sum + files.get(j).size <= 100; sum += files.get(j++).size);
    
        System.out.println("total size=" + sum + "; sublist: [" + i + ", " + j + "]");
        
        List<File> sublist = files.subList(i, j);
        
        sublist.forEach(s -> System.out.println("\t" + s));
        i = --j;
        
        sendFiles(sublist);
        result.add(sublist.stream().mapToInt(File::getId).toArray());
    }
    return result.toArray(new int[0][]);
}

测试：

List<File> files = Arrays.asList(
    new File(1, "f1", LocalDateTime.of(2021, 12, 20, 20, 00), 55),
    new File(2, "f2", LocalDateTime.of(2021, 12, 20, 19, 45), 20),
    new File(3, "f3", LocalDateTime.of(2021, 12, 20, 19, 30), 40),
    new File(4, "f4", LocalDateTime.of(2021, 12, 20, 19, 50), 45),
    new File(5, "f5", LocalDateTime.of(2021, 12, 20, 20, 10), 35)
);
        
System.out.println(Arrays.deepToString(groupFiles(files)));

输出：

total size=60; sublist: [0, 2]
    File: id=3; path=f3; finishedAt=2021-12-20T19:30; size=40
    File: id=2; path=f2; finishedAt=2021-12-20T19:45; size=20
total size=100; sublist: [2, 4]
    File: id=4; path=f4; finishedAt=2021-12-20T19:50; size=45
    File: id=1; path=f1; finishedAt=2021-12-20T20:00; size=55
total size=35; sublist: [4, 5]
    File: id=5; path=f5; finishedAt=2021-12-20T20:10; size=35
[[3, 2], [4, 1], [5]]