您的位置:首页 > 编程语言

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

2011-01-17 23:22 169 查看
原程序要扩展功能,支持右键菜单选择目录.为了减少改动,只得在接受到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字符串的过程中处理了文件).
教训:功能的扩展应以不大动筋骨为前提的.
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐