温馨提示×

readdir如何实现多线程目录扫描

小樊
48
2025-09-30 10:09:40
栏目: 编程语言

readdir 是一个用于读取目录内容的函数,通常在 C 语言中使用。要实现多线程目录扫描,你可以将目录分割成多个子目录,然后为每个子目录创建一个线程来执行 readdir。下面是一个简单的示例,展示了如何使用 C 语言和 POSIX 线程库(pthread)实现多线程目录扫描:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#include <pthread.h>

#define NUM_THREADS 4

typedef struct {
    char *path;
} thread_data_t;

void *scan_directory(void *arg) {
    thread_data_t *data = (thread_data_t *)arg;
    DIR *dir = opendir(data->path);
    struct dirent *entry;

    if (dir == NULL) {
        perror("opendir");
        pthread_exit(NULL);
    }

    while ((entry = readdir(dir)) != NULL) {
        if (entry->d_type == DT_DIR) {
            // 忽略当前目录和上级目录
            if (strcmp(entry->d_name, ".") != 0 && strcmp(entry->d_name, "..") != 0) {
                char sub_path[1024];
                snprintf(sub_path, sizeof(sub_path), "%s/%s", data->path, entry->d_name);
                printf("Scanning directory: %s\n", sub_path);

                // 递归扫描子目录
                pthread_t thread;
                thread_data_t sub_data = {sub_path};
                pthread_create(&thread, NULL, scan_directory, &sub_data);
                pthread_join(thread, NULL);
            }
        } else {
            printf("Found file: %s\n", entry->d_name);
        }
    }

    closedir(dir);
    pthread_exit(NULL);
}

int main(int argc, char *argv[]) {
    if (argc != 2) {
        fprintf(stderr, "Usage: %s <directory>\n", argv[0]);
        return 1;
    }

    pthread_t threads[NUM_THREADS];
    thread_data_t thread_data[NUM_THREADS];

    for (int i = 0; i < NUM_THREADS; ++i) {
        thread_data[i].path = argv[1];
        pthread_create(&threads[i], NULL, scan_directory, &thread_data[i]);
    }

    for (int i = 0; i < NUM_THREADS; ++i) {
        pthread_join(threads[i], NULL);
    }

    return 0;
}

这个示例程序接受一个目录作为命令行参数,并创建多个线程来扫描该目录及其子目录。每个线程都会执行 scan_directory 函数,该函数使用 readdir 读取目录内容。如果遇到子目录,它会创建一个新的线程来递归扫描子目录。

注意:这个示例程序没有处理线程同步和资源竞争问题。在实际应用中,你可能需要使用互斥锁(mutex)或其他同步机制来确保线程安全。

0