ACE目录遍历功能

freeeyes

有时候需要扫描当前文件夹，返回给远端。
于是写了一个小代码，发现ace对目录操作并不完整，目前到6.20依旧只支持windows，其实linux下也差不多，几十行的代码量。整合进ace应该不难吧。
由于需要传输到远端，所以我定义了一个结构，记录fileinfo
这里包括如下东西。

1. //文件列表结构
2. struct _FileInfo
3. {
4.         char   m_szFileName[500];
5.         int    m_nFileType;
6.         int    m_nFileSize;
8.         _FileInfo()
9.         {
10.                 m_szFileName[0] = '\0';
11.                 m_nFileType     = 0;
12.                 m_nFileSize     = 0;
13.         }
15.         void print()
16.         {
17.                 if(m_nFileType == IS_FILE)
18.                 {
19.                         printf("[GetDir]filename=%s, filetype=file, filesize=%d.\n", m_szFileName, m_nFileSize);
20.                 }
21.                 else
22.                 {
23.                         printf("[GetDir]filename=%s, filetype=direct, filesize=%d.\n", m_szFileName, m_nFileSize);
24.                 }
25.         }
26. };

复制代码

文件名，文件大小，文件类型。
获得当前路径的文件和文件夹显示并打印出来。
添加了Linux的支持，手写了一个支持跨平台的操作。
一共159行代码，还带换行符，代码量非常少。

1. // ACE_Dirtest.cpp : 定义控制台应用程序的入口点。
2. //
3. #ifdef WIN32
4. #include <stdio.h>
5. #include <ace/Dirent.h>
6. #include <vector>
7. #include <ace/OS.h>
8. #include <sys/stat.h>
9. #else
10. #include <ace/OS_main.h>
11. #include <ace/OS_NS_stdio.h>
12. #include <sys/stat.h>
13. #include <stdio.h>  
14. #include <string.h>
15. #include <stdlib.h>  
16. #include <dirent.h>  
17. #include <sys/stat.h>  
18. #include <unistd.h>  
19. #include <vector>
20. #include <sys/types.h>
21. #endif
23. using namespace std;
25. #define DIR_OK             0
26. #define DIR_ERR_ISFILE     1
27. #define DIR_ERR_ISNOEXIST  2
29. #define IS_FILE            0
30. #define IS_DIRRNT          1
32. //文件列表结构
33. struct _FileInfo
34. {
35.         char   m_szFileName[500];
36.         int    m_nFileType;
37.         int    m_nFileSize;
39.         _FileInfo()
40.         {
41.                 m_szFileName[0] = '\0';
42.                 m_nFileType     = 0;
43.                 m_nFileSize     = 0;
44.         }
46.         void print()
47.         {
48.                 if(m_nFileType == IS_FILE)
49.                 {
50.                         printf("[GetDir]filename=%s, filetype=file, filesize=%d.\n", m_szFileName, m_nFileSize);
51.                 }
52.                 else
53.                 {
54.                         printf("[GetDir]filename=%s, filetype=direct, filesize=%d.\n", m_szFileName, m_nFileSize);
55.                 }
56.         }
57. };
59. typedef vector<_FileInfo> vevFileInfo;
61. int GetDir( const char* pPath, vevFileInfo& objvevFileInfo )
62. {
63. #ifdef WIN32       
64.         objvevFileInfo.clear();
66.         //查看当前路径是否为一个目录
67.         struct ACE_DIR* objopendir = ACE_OS::opendir(pPath);
68.         if(!objopendir)
69.         {
70.                 //当前是一个文件不是目录
71.                 return DIR_ERR_ISFILE;
72.         }
73.         else
74.         {
75.                 //是目录，打开目录
76.                 struct ACE_DIRENT * objreaddir = ACE_OS::readdir(objopendir);
77.                 //略过.和..
78.                 while(objreaddir && objreaddir->d_name[0] == '.')
79.                 {
80.                         objreaddir = ACE_OS::readdir(objopendir);
81.                 }
83.                 while(objreaddir)     
84.                 {
85.                         char szFilePath[500] = {'\0'};
86.                         sprintf_s(szFilePath, 500, "%s%s", pPath, objreaddir->d_name);
87.                         _FileInfo objFileInfo;
88.                         sprintf_s(objFileInfo.m_szFileName, 500, "%s", objreaddir->d_name);
90.                         struct ACE_DIR* objIsdir = ACE_OS::opendir(szFilePath);
91.                         if(objIsdir == NULL)
92.                         {
93.                                 objFileInfo.m_nFileType = IS_FILE;
94.                                 objFileInfo.m_nFileSize = (int)ACE_OS::filesize(szFilePath);
95.                         }
96.                         else
97.                         {
98.                                 objFileInfo.m_nFileType = IS_DIRRNT;
99.                                 objFileInfo.m_nFileSize = 0;
100.                         }
102.                         objvevFileInfo.push_back(objFileInfo);
104.                         //printf("[GetDir]filename=%s.\n", objreaddir->d_name);
105.                         objreaddir = ACE_OS::readdir_emulation(objopendir);
106.                 }
107.                 ACE_OS::closedir_emulation(objopendir);
108.         }
109. #else
110.         objvevFileInfo.clear();
111.        
112.         struct dirent *objreaddir = NULL;  
113.         DIR *pDir = opendir(pPath);
114.         while((objreaddir=readdir(pDir))!=NULL)  
115.   {
116.                  if(strcmp(objreaddir->d_name,".")==0 || strcmp(objreaddir->d_name,"..")==0)
117.                  {  
118.                                 continue;
119.                  }
120.                  
121.                  char szFilePath[500] = {'\0'};
122.                  sprintf(szFilePath, "%s%s", pPath, objreaddir->d_name);                 
123.          
124.            _FileInfo objFileInfo;       
125.            if(objreaddir->d_type & DT_DIR)
126.            {
127.                            //如果是目录
128.                                 sprintf(objFileInfo.m_szFileName, "%s", objreaddir->d_name);
129.                                 objFileInfo.m_nFileType = IS_DIRRNT;
130.                                 objFileInfo.m_nFileSize = 0;                       
131.            }
132.            else
133.            {
134.                      //如果是文件
135.                                 sprintf(objFileInfo.m_szFileName, "%s", objreaddir->d_name);
136.                                 objFileInfo.m_nFileType = IS_FILE;
137.                                 objFileInfo.m_nFileSize = (int)ACE_OS::filesize(szFilePath);                                     
138.            }
139.            objvevFileInfo.push_back(objFileInfo);
140.   }
142. #endif
143.         return DIR_OK;
144. }
147. int main(int argc, char* argv[])
148. {
149.         vevFileInfo objfilegroup;
150.         GetDir("./", objfilegroup);
152.         for(int i = 0; i < (int)objfilegroup.size(); i++)
153.         {
154.                 objfilegroup[i].print();
155.         }
157.         getchar();
158.         return 0;
159. }

复制代码

windows 7下测试通过，Linux下测试通过。
记录一下。

winston

可以发给ACE邮件组的人

魔鬼

这个用STL就很方便了，没有必要在ACE中做一个这个功能。