c# - 根据运行总计选择条目

Question

我想从文件列表中选择只有这么多文件，它们的总大小不超过阈值（即目标驱动器上的可用空间量）。

我知道我可以通过在循环中添加文件大小直到达到阈值然后使用该数字从列表中选择文件来做到这一点。但是，是否可以使用 LINQ 查询来代替？

Answer 1

这可以工作（files是一个List<FileInfo>）：

var availableSpace = DriveInfo.GetDrives()
    .First(d => d.Name == @"C:\").AvailableFreeSpace;
long usedSpace = 0;
var availableFiles = files
    .TakeWhile(f => (usedSpace += f.Length) < availableSpace);
foreach (FileInfo file in availableFiles)
{
    Console.WriteLine(file.Name);
}

Answer 2

您可以通过使用闭包来实现：

var directory = new DirectoryInfo(@"c:\temp");
var files = directory .GetFiles();

long maxTotalSize = 2000000;

long aggregatedSize = 0;
var result = files.TakeWhile(fileInfo => 
{
    aggregatedSize += fileInfo.Length;
    return aggregatedSize <= maxTotalSize;
});

但是有一个警告，因为变量aggregatedSize可能会在您离开定义它的范围后被修改。

您可以将其包装在扩展方法中 - 这将消除关闭：

public static IEnumerable<FileInfo> GetWithMaxAggregatedSize(this IEnumerable<FileInfo> files, long maxTotalSize)
{
    long aggregatedSize = 0;
    return files.TakeWhile(fileInfo => 
    {
        aggregatedSize += fileInfo.Length;
        return aggregatedSize <= maxTotalSize;
    });
}

你最终使用这样的方法：

var directory = new DirectoryInfo(@"c:\temp");
var files = directory.GetFiles().GetWithMaxAggregatedSize(2000000);

编辑：我用Where-method替换了TakeWhile-method。TakeWhile一旦达到阈值，-extension 将停止，而-extensionWhere将继续。提出 - 扩展名的TakeWhile功劳归于 Tim Schmelter。