.NET 4.0中有一种很好的新方法，可以通过枚举以流方式获取目录中的文件。
问题在于，如果想要枚举所有文件，则事先可能不知道哪些文件或文件夹受到保护而会引发UnauthorizedAccessException。
要重现此问题，只需运行此片段：
foreach (var file in Directory.EnumerateFiles(@"c:\", "*", SearchOption.AllDirectories))
{
// whatever
}
在这个.NET方法出现之前，可以通过对字符串数组返回方法实现递归迭代器来实现大致相同的效果。但是它不像新的.NET方法那样懒惰。
那么该怎么做呢？UnauthorizedAccessException可以被抑制吗？还是在使用此方法时必然会发生？
在我看来，这个方法应该有一个重载，接受一个操作来处理任何异常。
我无法让以上内容起作用，但这是我的实现方法，我已在“Win7”框中的 c:\users 上进行了测试，因为它拥有所有这些“恶意”目录：
SafeWalk.EnumerateFiles(@"C:\users", "*.jpg", SearchOption.AllDirectories).Take(10)
类：
public static class SafeWalk
{
public static IEnumerable<string> EnumerateFiles(string path, string searchPattern, SearchOption searchOpt)
{
try
{
var dirFiles = Enumerable.Empty<string>();
if(searchOpt == SearchOption.AllDirectories)
{
dirFiles = Directory.EnumerateDirectories(path)
.SelectMany(x => EnumerateFiles(x, searchPattern, searchOpt));
}
return dirFiles.Concat(Directory.EnumerateFiles(path, searchPattern));
}
catch(UnauthorizedAccessException ex)
{
return Enumerable.Empty<string>();
}
}
}
/// <summary>
/// A safe way to get all the files in a directory and sub directory without crashing on UnauthorizedException or PathTooLongException
/// </summary>
/// <param name="rootPath">Starting directory</param>
/// <param name="patternMatch">Filename pattern match</param>
/// <param name="searchOption">Search subdirectories or only top level directory for files</param>
/// <returns>List of files</returns>
public static IEnumerable<string> GetDirectoryFiles(string rootPath, string patternMatch, SearchOption searchOption)
{
var foundFiles = Enumerable.Empty<string>();
if (searchOption == SearchOption.AllDirectories)
{
try
{
IEnumerable<string> subDirs = Directory.EnumerateDirectories(rootPath);
foreach (string dir in subDirs)
{
foundFiles = foundFiles.Concat(GetDirectoryFiles(dir, patternMatch, searchOption)); // Add files in subdirectories recursively to the list
}
}
catch (UnauthorizedAccessException) { }
catch (PathTooLongException) {}
}
try
{
foundFiles = foundFiles.Concat(Directory.EnumerateFiles(rootPath, patternMatch)); // Add files from the current directory
}
catch (UnauthorizedAccessException) { }
return foundFiles;
}
我理解是MoveNext抛出了异常。
我尝试编写一个方法，安全地遍历序列并尝试忽略MoveNext异常。但是我不确定MoveNext在抛出异常时是否会推进位置，所以这也可能是无限循环。这也是个坏主意，因为我们会依赖于实现细节。
但这只是非常有趣！
public static IEnumerable<T> SafeWalk<T> (this IEnumerable<T> source)
{
var enumerator = source.GetEnumerator();
bool? hasCurrent = null;
do {
try {
hasCurrent = enumerator.MoveNext();
} catch {
hasCurrent = null; // we're not sure
}
if (hasCurrent ?? false) // if not sure, do not return value
yield return enumerator.Current;
} while (hasCurrent ?? true); // if not sure, continue walking
}
foreach (var file in Directory.EnumerateFiles("c:\\", "*", SearchOption.AllDirectories)
.SafeWalk())
{
// ...
}
只有在以下关于框架实现此迭代器的条件为真时，此方法才能正常工作（参见 Reflector 中的 FileSystemEnumerableIterator<TSource>）：
MoveNext 在失败时会移动其位置；MoveNext 在最后一个元素上失败时，后续调用将返回 false 而不是抛出异常；即使它可以工作，请永远不要在生产环境中使用！
但我真的很想知道它是否可以。
static IEnumerable<string> FindFiles(string path, string filter = "*", bool recursive = false)
{
IEnumerator<string> fEnum;
try
{
fEnum = Directory.EnumerateFiles(path, filter, recursive ? SearchOption.AllDirectories : SearchOption.TopDirectoryOnly).GetEnumerator();
}
catch (UnauthorizedAccessException) { yield break; }
while (true)
{
try { if (!fEnum.MoveNext()) break; }
catch (UnauthorizedAccessException) { continue; }
yield return fEnum.Current;
}
}
Dan Bechard在评论中提到：
不幸的是，当它抛出异常时，MoveNext()不会推进其位置。
这个问题可能已经在较新版本的.NET或Windows 10构建中修复了？我在Windows 10上的.NET 5.0中没有遇到这个问题。通过搜索我的整个系统驱动器进行测试。
在VB.NET中：
Public Iterator Function FindFiles(path As String, Optional filter As String = "*", Optional recursive As Boolean = False) As IEnumerable(Of String)
Dim fEnum As IEnumerator(Of String)
Dim searchDepth = If(recursive, SearchOption.AllDirectories, SearchOption.TopDirectoryOnly)
Try
fEnum = Directory.EnumerateFiles(path, filter, searchDepth).GetEnumerator()
Catch uae As UnauthorizedAccessException
Return
End Try
Do While True
Try
If Not fEnum.MoveNext() Then Exit Do
Yield fEnum.Current
Catch uae As UnauthorizedAccessException
End Try
Loop
End Function
我迟到了，但我建议使用观察者模式：
public class FileUtil
{
private static void FindFiles_(string path, string pattern,
SearchOption option, IObserver<string> obs, CancellationToken token)
{
try
{
foreach (var file in Directory.EnumerateFiles(path, pattern,
SearchOption.TopDirectoryOnly))
{
if (token.IsCancellationRequested) break;
obs.OnNext(file);
}
if (option != SearchOption.AllDirectories) return;
foreach (var dir in Directory.EnumerateDirectories(path, "*",
SearchOption.TopDirectoryOnly))
{
FindFiles_(dir, pattern, option, obs, token);
}
}
catch (UnauthorizedAccessException) { }
catch (PathTooLongException) { }
catch (IOException) { }
catch (Exception err) { obs.OnError(err); }
}
public static IObservable<string> GetFiles(string root, string pattern,
SearchOption option)
{
return Observable.Create<string>(
(obs, token) =>
Task.Factory.StartNew(
() =>
{
FindFiles_(root, pattern, option, obs, token);
obs.OnCompleted();
},
token));
}
}
我自己实现了一个类来解决这个问题，因为之前的答案似乎不能满足我的需求。它会跳过所有无法访问的文件和文件夹，并返回所有可以访问的文件。
public static class SafeWalk
{
public static IEnumerable<string> EnumerateFiles(string path, string searchPattern, SearchOption searchOpt)
{
if (searchOpt == SearchOption.TopDirectoryOnly)
{
return Directory.EnumerateFiles(path, searchPattern, SearchOption.TopDirectoryOnly);
}
List<string> folders = new List<string>() { path };
int folCount = 1;
List<string> files = new List<string>() { };
for (int i = 0; i < folCount; i++)
{
try
{
foreach (var newDir in Directory.EnumerateDirectories(folders[i], "*", SearchOption.TopDirectoryOnly))
{
folders.Add(newDir);
folCount++;
try
{
foreach (var file in Directory.EnumerateFiles(newDir, searchPattern))
{
files.Add(file);
}
} catch (UnauthorizedAccessException)
{
// Failed to read a File, skipping it.
}
}
}
catch (UnauthorizedAccessException)
{
// Failed to read a Folder, skipping it.
continue;
}
}
return files;
}
}
可以像常规的EnumerateFiles函数一样使用，只需使用SafeWalk.EnumerateFiles(...)而不是Dictionary.EnumerateFiles(...)即可。
Dump()方法，它只是遍历字符串枚举。问题在于Directory.EnumerateFiles方法本身。我认为没有办法解决这个问题。你必须使用SearchOption.TopDirectoryOnly并自己处理递归。 - Mormegil