用LINQ遍历目录

今天看了一天的LINQ，给自己布置一个问题：用LINQ遍历一个指定目录并输出为XML

写出的代码如下：

public static XElement ToXml(this DirectoryInfo dir)

{

XElement root = new XElement("Dir",

new XAttribute("Name", dir.Name),

from subDir in dir.GetDirectories()

select subDir.ToXml(),

from file in dir.GetFiles()

select new XElement("File", file.Name));

return root;

}

 

使用的时候，你可以这样：

DirectoryInfo dir = new DirectoryInfo(path);

XElement dirXml = dir.ToXml();

OK，搞定。

可以看出，LINQ非常强大，既然这么强大，为什么不更进一步呢？

我要遍历一下目录里的制定格式的文件应该怎么做呢？

比如说我要找出目录里的所有的mp3

于是可以写出这样的代码

public static XElement ToXml(this DirectoryInfo dir, string searchPattern)

{

XElement root = new XElement("Dir",

new XAttribute("Name", dir.Name),

from subDir in dir.GetDirectories()

select subDir.ToXml(searchPattern),

from file in dir.GetFiles(searchPattern)

select new XElement("File", file.Name));

return root;

}

使用的时候是这样

DirectoryInfo dir = new DirectoryInfo(path);

XElement dirXml = dir.ToXml("*.mp3");

轻松搞定

更进一步，如果我要找多个格式的文件呢？比如说要找出目录里面的图片文件（jpg, bmp, gif…）

Ok，没问题，可以写成这样：

public static XElement ToXml(this DirectoryInfo dir, string[] searchPatterns)

{

XElement root = new XElement("Dir",

new XAttribute("Name", dir.Name),

from subDir in dir.GetDirectories()

select DirToXml(subDir),

from pattern in searchPatterns

from file in dir.GetFiles(pattern)

select new XElement("File", file.Name));

return root;

}

使用的时候可以这样：

DirectoryInfo dir = new DirectoryInfo(path);

XElement dirXml = dir.ToXml(new[] { "*.mp3", "*.txt" });

Ok，又搞定了。

为什么我要写三个函数呢？我把三个合并成一个可以吗？试试看吧

于是，代码就成了这样：

public static XElement ToXml(this DirectoryInfo dir, params string[] searchPatterns)

{

XElement root = new XElement("Dir",

new XAttribute("Name", dir.Name),

from subDir in dir.GetDirectories()

select subDir.ToXml(searchPatterns),

searchPatterns.Length > 0 ?

(from pattern in searchPatterns.Distinct() 

from file in dir.GetFiles(pattern)

select new XElement("File", file.Name))

:

    (from file in dir.GetFiles()

select new XElement("File", file.Name)));

 

return root;

}

测试一下，perfect！

美中不足的是，在代码中的一个（searchPatterns.Length > 0 ?）就像一个刀疤，有谁有什么好些的主意吗？

转载地址：http://www.cnblogs.com/alexch/archive/2008/10/20/1315438.html