C++ 常用代码片段整理

2021-11-09 18:31:24 浏览数 (1)

C 常用代码片段整理

一丶文件 文件夹相关

1.1 递归删除文件

代码语言:javascript复制
wstring Utilstring2wstring(string str)
{
	wstring result;
	//获取缓冲区大小,并申请空间,缓冲区大小按字符计算  
	int len = MultiByteToWideChar(CP_ACP, 0, str.c_str(), str.size(), NULL, 0);
	wchar_t* buffer = NULL;
	buffer = new wchar_t[len   1];
	//多字节编码转换成宽字节编码  
	MultiByteToWideChar(CP_ACP, 0, str.c_str(), str.size(), buffer, len);
	buffer[len] = L'';             //添加字符串结尾  
	//删除缓冲区并返回值  
	result.append(buffer);
	if (buffer != NULL)
	{
		delete[] buffer;
		buffer = NULL;
	}
	return result;
}

//将wstring转换成string  
string Utilwstring2string(wstring wstr)
{
	string result;
	//获取缓冲区大小,并申请空间,缓冲区大小事按字节计算的  
	int len = WideCharToMultiByte(CP_ACP, 0, wstr.c_str(), wstr.size(), NULL, 0, NULL, NULL);
	char* buffer = new char[len   1];
	//宽字节编码转换成多字节编码  
	WideCharToMultiByte(CP_ACP, 0, wstr.c_str(), wstr.size(), buffer, len, NULL, NULL);
	buffer[len] = '';
	//删除缓冲区并返回值  
	result.append(buffer);
	delete[] buffer;
	return result;
}

bool DeleteAllFileA(std::string FileName)
{
	string RootFileName = FileName;
	wstring wRootFileName = Utilstring2wstring(RootFileName);
	
	RootFileName = endsWith(wRootFileName, L"\") ? RootFileName : RootFileName   "\";
	string RmFilter = RootFileName   "*.*";
	WIN32_FIND_DATAA windata = { 0 };
	HANDLE hFileHandle = FindFirstFileA(RmFilter.c_str(), &windata);
	do
	{
		if (windata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
		{
			string compareDirName = windata.cFileName;
			if (compareDirName.find(".") != compareDirName.npos
				|| compareDirName.find("..") != compareDirName.npos)
			{
				continue;
			}
			//递归删除
			string RmFileName = RootFileName   compareDirName;
			DeleteAllFileA(RmFileName);
			RemoveDirectoryA(RmFileName.c_str()); //递归删除文件夹
			continue;
		}
		//直接删除文件
		string RmFileName = RootFileName   windata.cFileName;
		DeleteFileA(RmFileName.c_str());
		

	} while (FindNextFileA(hFileHandle, &windata));
	
	return true;
}

0 人点赞