如何用copendir实现Linux目录的筛选遍历

本文介绍如何在Linux系统中使用C语言高效筛选遍历目录。 opendir 和 readdir 函数是目录遍历的常用工具，但若需筛选特定类型文件，则需额外处理。以下代码示例演示如何实现这一功能：

代码示例:

#include 
#include 
#include 
#include 
#include 

// 函数：检查文件类型是否匹配
int checkFileType(const char *path, const char *type) {
    if (strcmp(type, "all") == 0) return 1; // 不筛选

    char ext[10];
    const char *dot = strrchr(path, '.');
    if (dot && dot > path) {
        size_t len = dot - path;
        if (len >= sizeof(ext)) len = sizeof(ext) - 1;
        strncpy(ext, dot + 1, len);
        ext[len] = '\0';
    } else {
        strcpy(ext, "");
    }
    return strcmp(ext, type) == 0;
}

// 函数：递归遍历目录并筛选
void traverseDir(const char *path, const char *filter) {
    DIR *dir = opendir(path);
    if (dir == NULL) {
        perror("opendir");
        return;
    }

    struct dirent *entry;
    while ((entry = readdir(dir)) != NULL) {
        if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) continue;

        char fullPath[PATH_MAX];
        snprintf(fullPath, sizeof(fullPath), "%s/%s", path, entry->d_name);

        struct stat st;
        if (stat(fullPath, &st) == -1) {
            perror("stat");
            continue;
        }

        if (S_ISDIR(st.st_mode)) {
            traverseDir(fullPath, filter);
        } else if (S_ISREG(st.st_mode)) {
            if (checkFileType(fullPath, filter)) {
                printf("文件: %s\n", fullPath);
            }
        }
    }
    closedir(dir);
}

int main(int argc, char *argv[]) {
    if (argc != 3) {
        fprintf(stderr, "用法: %s <目录路径> <过滤器>\n", argv[0]);
        fprintf(stderr, "过滤器示例:\n");
        fprintf(stderr, "  all       - 所有文件\n");
        fprintf(stderr, "  txt       - .txt文件\n");
        fprintf(stderr, "  jpg,png   - .jpg或.png文件\n");
        return EXIT_FAILURE;
    }

    const char *directory = argv[1];
    const char *filter = argv[2];
    traverseDir(directory, filter);
    return EXIT_SUCCESS;
}

代码说明:

• checkFileType 函数：检查文件扩展名是否与过滤器匹配，支持 "all" (不筛选)，以及多个扩展名用逗号分隔的情况。
• traverseDir 函数：递归遍历目录，使用 stat 获取文件信息，判断文件类型并调用 checkFileType 进行筛选。
• main 函数：处理命令行参数，调用 traverseDir 开始遍历。

编译和运行:

VISBOOM

AI虚拟试衣间，时尚照相馆。

下载

1. 将代码保存为 dir_traverse.c。
2. 编译：gcc -o dir_traverse dir_traverse.c
3. 运行：./dir_traverse /path/to/directory txt (将 /path/to/directory 替换为实际路径，txt 为过滤器)

改进方向:

• 更强大的过滤器：支持更复杂的模式匹配 (例如正则表达式)。
• 错误处理：更完善的错误处理机制。
• 性能优化：对于大型目录，考虑多线程或异步IO提高效率。

这个改进后的示例提供了更清晰的代码结构和更强大的筛选功能，方便用户根据需求进行调整和扩展。 Python实现方式类似，但可以使用 os.walk 和 fnmatch 模块简化代码。