Delphi下递归查找目录的通用方法

来源:互联网 发布:淘宝网袜子批发 编辑:程序博客网 时间:2024/04/28 07:28

最近的项目中用到了Windows下的目录搜索,特整理了一段通用算法如下:

//定义搜索到文件时的回调函数
//如果是非对象方法,请去掉of object部分
//aFile: 搜索到的文件
//willStop: 外部变量,用于决定回调操作后,是否终止搜索。
TFindFile = procedure(aFile: string; var willStop: boolean) of object;

//通用的目录搜索算法
//aDir: 要搜索的目录
//onFind: 搜索到文件时的回调函数
procedure doFindFile(aDir: string; onFind: TFindFile);
var strDir, strfile: string;
    ff: _WIN32_FIND_DATAA;
    hf: THandle;
    blStop: boolean;
begin
  //记录当前的目录
  strDir := getCurrentDir;
  //设置当前目录为要搜索的目录
  setcurrentDir(aDir);
  try
    //开始搜索
    hf := Windows.FindFirstFile('*.*', ff);
    if hf > 0 then begin
      repeat
        strFile := ff.cFileName;
        //如果是目录,则递归调用
        if (ff.dwFileAttributes and FILE_ATTRIBUTE_DIRECTORY = FILE_ATTRIBUTE_DIRECTORY) then begin
          if (strFile <> '') and (pos('.', strFile) = 0) then begin
            doFindFile(aDir + strFile + '/', onFind);
          end;
        end else begin
          blStop := False;
          //如果是文件,交由回调函数处理
          onFind(aDir + strFile, blStop);
          //如果回调函数要求终止搜索,则退出当前的递归过程
          if blStop then begin
            Windows.FindClose(hf);
            Exit;
          end;
        end;
      until (not Windows.FindNextFile(hf, ff));
      //终止查找
      Windows.FindClose(hf);
    end;
  finally
    //恢复搜索前的目录
    setCurrentDir(strDir);
  end;
end;

算法的关键在于通过定义具有通用参数的回调函数,来将目录搜索中的关键操作独立出来,以实现算法的通用性。
下面举个例子说明其调用过程:

procedure TForm1.findTextFile(aFile: string; var willStop: boolean);
begin
  if LowerCase(ExtractFileExt(aFile)) = '.txt' then
    willStop := Application.MessageBox(pchar('找到了文本文件 ' + aFile + '! 要停止搜索吗?'), '提示', MB_YESNO) = mrYes;
end;

procedure TForm1.Button1Click(Sender: TObject);
begin
  //Edit1.Text 定义了要搜索的目录
  if Copy(Edit1.Text, Length(Edit1.Text), 1) <> '/' then Edit1.Text := Edit1.Text + '/';
  //调用通用搜索算法,查找文本文件
  doFindFile(Edit1.Text, findTextFile);
end;

原创粉丝点击