源代码修改-TravelAllFiles

using System;
using System.Collections.Generic;
using System.IO;
using System.Text;

namespace 源代码修改
{
    //  遍历程序当前目录下所有.cs文件
    public class TravelAllFiles
    {
        private DirectoryInfo rootDirectory = new DirectoryInfo(Directory.GetCurrentDirectory());
        private string extension = ".cs";
        private IOperateSource operate ;

        public TravelAllFiles()
        {
        }

        private void Go(DirectoryInfo currentDirectory)
        {
            foreach (FileInfo file in currentDirectory.GetFiles())
            {
                if (file.Extension == extension)
                {
                    //  处理文件
                    operate.Go(file);
                }
            }

            foreach (DirectoryInfo directory in currentDirectory.GetDirectories())
            {
                Go(directory);
            }
        }

        public void Go(IOperateSource operate)
        {
            this.operate = operate;
            Go(rootDirectory);
        }
    }
}