/// <summary>
/// 用于添加子节点
/// </summary>
/// <param name="data"></param>
/// <param name="depth"></param>
private NTree(T data, NTree <T> parent, int depth)
{
this.data = data;
this.parent = parent;
this.depth = depth;
children = new LinkedList <NTree <T> >();
}
public NTree(T data)
{
this.data = data;
//默认是根节点
this.depth = 0;
this.parent = null;
children = new LinkedList <NTree <T> >();
}
/// <summary>
/// 弹出一个子节点 并Remove
/// </summary>
/// <returns></returns>
public NTree <T> PushChild()
{
if (children.Count > 0)
{
NTree <T> n = children.First.Value;
children.RemoveFirst();
return(n);
}
return(null);
}
public static IEnumerable <string> EnumFilesYield(string path)
{
if (string.IsNullOrEmpty(path))
{
throw new ArgumentNullException(path);
}
WIN32_FIND_DATA findData;
NTree <string> parent = new NTree <string>(path);
//后序遍历树
NTree <string> child = null;
while (true)
{
string tempPath = parent.Data;
string uncPath = getUncPath(parent.Data);
var findHandle = FindFirstFile(uncPath, out findData);
if (findHandle != INVALID_HANDLE_VALUE)
{
try
{
do
{
if (findData.cFileName != "." && findData.cFileName != "..")
{
if (findData.dwFileAttributes == 16)
{
parent.AddChild(tempPath + Path.DirectorySeparatorChar + findData.cFileName);
}
else
{
yield return(tempPath + Path.DirectorySeparatorChar + findData.cFileName);
}
}
} while (FindNextFile(findHandle, out findData));
}
finally
{
FindClose(findHandle);
}
}
//判断是否有子节点
child = parent.PushChild();
if (child == null)
{
//如果没有子节点，则回溯至上层,否则跳出（表示遍历结束）
again : if (parent.Depth > 0)
{
parent = parent.Parent;
}
else
{
yield break;
}
child = parent.PushChild();
if (child == null)
{
goto again;
}
}
//作为父节点 进行循环
parent = child;
}
//log
//Marshal.GetExceptionForHR(Marshal.GetHRForLastWin32Error());
}