linux下获取所有文件夹和文件，支持nfs和xfs

 linux下获取所有文件夹和文件，支持nfs和xfs

要获取一个指定目录下的文件或文件夹，在linux下一般用dirent来做。dirent结构体中d_type表示类型，4表示目录，8表示普通文件，0表示未知设备。一般的文件系统直接用d_type判断，不会出现问题。由于项目中用到了nfs，实际的文件存放在另一台机器上，通过nfs映射到当年机器上。读文件的时候发现d_type为0，导致判断错误。因此用stat中的st_mode来判断是否为文件或文件夹。

1. [代码]File.h 

view source

01	//written by baoer1024 on 2012-3-22

02	#ifndef _FILE_H_

03	#define _FILE_H_

04

05	#include <vector>

06	#include <string>

07	#include <time.h>

08	#include <sys/stat.h>

09	using namespace std;

10

11	const int MAX_DUR = 5*60;  //时间阈值

12

13	class File

14

{

15

public:

16	//获取路径下的所有文件夹

17	vector<string> GetFolders(const string &path);

18

19	//获取路径下的所有文件，可指定文件后缀名

20	vector<string> GetFiles(const string &path, const string &postfix="");

21

22	//检查文件或目录是否准备好，一定程度上防止数据正在拷贝的时候去读写。

23	//方法是检查文件或目录的access time是否在5分钟以前

24	bool IsPrepared(const string &path);

25

26	//判断是否为文件夹

27	bool IsFolder(const string &path);

28

29

//判断是否为文件

30	bool IsFile(const string &path);

31

};

32

33

#endif

2. [代码]File.cpp     

01	//written by baoer1024 on 2012-3-22

02	#include "File.h"

03	#include <string.h>

04	#include <stdio.h>

05	#include <stdlib.h>

06	#include <dirent.h>

07

08	//获取路径下的所有文件夹

09	vector<string> File::GetFolders(const string &path)

10

{

11	vector<string> folders;

12	struct dirent* ent = NULL;

13	DIR* pDir;

14	pDir = opendir(path.c_str());

15	while(NULL != (ent = readdir(pDir)))

16

{

17	string fullpath = path + "/" + ent->d_name;

18	//if(4 == ent->d_type)  //在nfs或xfs下，有的目录d_type也是0

19	if(IsFolder(fullpath))

20

{

21	if(strcmp(ent->d_name, ".")!=0 && strcmp(ent->d_name, "..")!=0)

22

{

23	folders.push_back(ent->d_name);

24

}

25

}

26

}

27	closedir(pDir);

28	return folders;

29

}

30

31	//获取路径下的所有文件，可指定文件后缀名

32	vector<string> File::GetFiles(const string &path, const string &postfix)

33

{

34	vector<string> files;

35	struct dirent* ent = NULL;

36	DIR* pDir;

37	pDir = opendir(path.c_str());

38	while(NULL != (ent = readdir(pDir)))

39

{

40	string fullpath = path + "/" + ent->d_name;

41	//if(8 == ent->d_type)  //在nfs或xfs下，有的文件d_type也是0

42	if(IsFile(fullpath))

43

{

44	if(postfix == "" || strstr(ent->d_name, postfix.c_str())!=NULL)

45

{

46	files.push_back(ent->d_name);

47

}

48

}

49

}

50	closedir(pDir);

51	return files;

52

}

53

54	//检查文件或目录是否准备好，一定程度上防止数据正在拷贝的时候去读写。

55	//方法是检查文件或目录的access time是否在5分钟以前

56	bool File::IsPrepared(const string &path)

57

{

58	struct stat st;

59	stat(path.c_str(), &st);

60	return time(0)-st.st_ctime >= MAX_DUR;

61

}

62

63	//判断是否为文件夹，用stat的标志来判断

64	bool File::IsFolder(const string &path)

65

{

66	struct stat st;

67	int ret = stat(path.c_str(), &st);

68	return ret>=0 && S_ISDIR(st.st_mode);

69

}

70

71	//判断是否为文件，用stat的标志来判断

72	bool File::IsFile(const string &path)

73

{

74	struct stat st;

75	int ret = stat(path.c_str(), &st);

76	return ret>=0 && S_ISREG(st.st_mode);

77

}

转自： http://www.oschina.net/code/snippet_203297_9027