支持目录、多文件的扩展代码

 

原程序要扩展功能,支持右键菜单选择目录.为了减少改动,只得在接受到shell传递给程序的COPYDATASTRUCT后，处理路径字符串.
比如：path1*path2*path3……，其中path可能是文件路径，也可能是文件夹路径.需转换成文件路径的字符串,以防止大改代码,filepath1*filepath2*filepath3…….

 

---

 

 

[in]          pszNodeValue    路径

[in,out]    strFilePaths       指定格式字符串

[in]          seperator          分隔符(如：* 或 | )

  处理单个文件：void do_findFile(char *pszNodeValue, string &strFilePaths, const char *seperator) { strFilePaths.append(pszNodeValue); strFilePaths.append(seperator); }

 

处理文件夹：void do_findFilesFromDir(char *pszNodeValue,string &strFilePaths, const char *seperator) { char szFindPath[MAX_PATH]; strcpy(szFindPath,pszNodeValue); strcat(szFindPath,"//*.*"); WIN32_FIND_DATA wfd; HANDLE hFind; hFind = FindFirstFile(szFindPath,&wfd); if(hFind == INVALID_HANDLE_VALUE) return; while(TRUE) { if(wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { if(wfd.cFileName[0]!='.') { char szTempDir[MAX_PATH]; strcpy(szTempDir,pszNodeValue); strcat(szTempDir,"//"); strcat(szTempDir,wfd.cFileName); do_findFilesFromDir(szTempDir,strFilePaths, seperator); } } else { strFilePaths.append(pszNodeValue); strFilePaths.append("//"); strFilePaths.append(wfd.cFileName); strFilePaths.append(seperator); } if(!FindNextFile(hFind,&wfd)) break; } FindClose(hFind); return; }

---

[in,out]      strDest         生成的新字符串()

[in]           szSrc               传入的字符串(如：path1*path2*path3)

[in]           seperator       分隔符(如：* 或|)

得到了转换字符串的函数：BOOL do_CovertPaths(string &strDest, const char *szSrc, const char *seperator) { char *pszNodeValue = strtok((char *)szSrc, seperator); while (pszNodeValue != NULL) { if (FILE_ATTRIBUTE_DIRECTORY == GetFileAttributes(pszNodeValue)) do_findFilesFromDir(pszNodeValue, strDest, seperator); else do_findFile(pszNodeValue, strDest, seperator); pszNodeValue = strtok(NULL, seperator); } strDest = strDest.substr( 0, strDest.length()-1 ); return TRUE; }

---

总结

开始处理这个任务时,是直接在源程序的线程添加的代码——直接解析传来的字符串，把所有文件的路径添加到vector<string> vFilePaths中(是单个文件直接加入到vFilePaths中,是文件夹就调用封装好的函数,把路径加入到vFilePaths中).然后再遍历vFilePaths,处理文件.

缺点：程序有4个线程,每个都要改的话,修改比较大.（原程序只支持文件、不支持目录,过程是：strtok字符串的过程中处理了文件）.

教训：功能的扩展应以不大动筋骨为前提的.