可控制遍历深度的文件搜索类

最新推荐文章于 2020-11-24 15:34:50 发布

熬夜程序猴

最新推荐文章于 2020-11-24 15:34:50 发布

阅读量1k

点赞数 1

分类专栏： C/C++ 编程技巧[原创] 文章标签： search c 搜索遍历 win32

本文链接：https://blog.csdn.net/XinChiMaker/article/details/10433597

版权

编程技巧[原创] 同时被 2 个专栏收录

33 篇文章 1 订阅

订阅专栏

C/C++

17 篇文章 0 订阅

订阅专栏

.H 文件

//---------------------------------------------------------------------------

#ifndef SearchFilesUnitH
#define SearchFilesUnitH
//---------------------------------------------------------------------------

#include <windows.h>
#include <stdio.h>

typedef void(__stdcall *PSEARCHCALLBACK)(LPCTSTR);
typedef void(__stdcall *PDIRCOMPLETEDCALLBACK)(LPCTSTR);

class CSearchFile
{
public:
    // 无深度初始化构造函数，此时扫描深度不受限制
    CSearchFile()
    {
        m_depthLevel = -1;
    }
    
    // 扫描深度初始化构造函数
    CSearchFile(int depthLevel)
    {
        m_depthLevel = depthLevel;
    }
private:
    int m_depthLevel;

private:
    BOOL DirectoryExist(LPCTSTR lpszDir)
    {
        DWORD dwAttribute = ::GetFileAttributes(lpszDir);
        if (dwAttribute != 0xFFFFFFFF)
        {
            return ((dwAttribute & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY);
        }
        return FALSE;
    }
public:
    BOOL SearchFile(
            LPCTSTR root                                    /* 搜索文件的根目录 */
            , PSEARCHCALLBACK pFoundCallback                /* 文件搜到后的 Callback
                                                            在此Callback中可以实现
                                                            文件类型的过滤，文件删除，复制等等
                                                            与文件相关的操作
                                                            */
            , PDIRCOMPLETEDCALLBACK pDirCompletedCallback   /* 完成一个子目录后的 Callback
                                                            在此Callback中可以配合上个Callback
                                                            的文件删除功能，删除文件夹
                                                            */
            , int depthLevel = 0
            );
};
#endif

.cpp 文件

//---------------------------------------------------------------------------
#include "SearchFilesUnit.h"
//---------------------------------------------------------------------------


BOOL CSearchFile::SearchFile(
                        LPCTSTR root                                    /* 搜索文件的根目录 */
                        , PSEARCHCALLBACK pFoundCallback                /* 文件搜到后的 Callback
                                                                        在此Callback中可以实现
                                                                        文件类型的过滤，文件删除，复制等等
                                                                        与文件相关的操作
                                                                        */
                        , PDIRCOMPLETEDCALLBACK pDirCompletedCallback   /* 完成一个子目录后的 Callback
                                                                        在此Callback中可以配合上个Callback
                                                                        的文件删除功能，删除文件夹
                                                                        */
                        , int depthLevel
                        )
{
    char szPath[MAX_PATH] = {0};
    WIN32_FIND_DATA stWFD;
    memset(&stWFD, 0, sizeof(WIN32_FIND_DATA));
    sprintf(szPath, "%s\\*", root);
    
    // 如果有扫描深度控制
    if (m_depthLevel > 0)
    {
        if (depthLevel++ > m_depthLevel)
        {
            depthLevel = 0;
            return FALSE;
        }
    }

    HANDLE file = ::FindFirstFile(szPath, &stWFD);

    if (INVALID_HANDLE_VALUE == file) return FALSE;

    do
    {
        if (strcmp(stWFD.cFileName, ".") && strcmp(stWFD.cFileName, ".."))
        {
            char szFoundFile[MAX_PATH] = { 0 };
            sprintf(szFoundFile, "%s\\%s", root, stWFD.cFileName);


            // 判断是目录还是文件
            if (DirectoryExist(szFoundFile))
            {
                SearchFile(szFoundFile, pFoundCallback, pDirCompletedCallback, depthLevel);
            }
            if (NULL != pFoundCallback)
            {
                pCallback(szFoundFile);
            }
        }

    }while (::FindNextFile(file, &stWFD));
        
    ::FindClose(file);
        
    if (NULL != pDirCompletedCallback)
    {
        pDirCompletedCallback(root);
    }
    return TRUE;
}