C++删除目标文件夹下所有文件

前言

在实际项目中会遇到需要定期清理软件生成的日志文件的需求
下面写一个简单清理函数

#include <windows.h>
#include <iostream>
#include <string>
#define _CRT_SECURE_NO_WARNINGS
typedef std::basic_string<TCHAR, std::char_traits<TCHAR>, std::allocator<TCHAR> > MYString;

bool DeleteDirectory(IN LPCTSTR lpstrDir)
{
	TCHAR szFind[MAX_PATH];
	WIN32_FIND_DATA	fd;
	_tcscpy_s(szFind, lpstrDir);
	_tcscat_s(szFind, _T("\\*.*"));

	HANDLE hFind = ::FindFirstFile(szFind, &fd);
	if (INVALID_HANDLE_VALUE == hFind)
	{
		//!Vista下调用此API可能导致该文件夹被锁住
		//if ( !::RemoveDirectory(lpstrDir) )
		//{
		//	return false;
		//}

		return true;
	}

	while (true)
	{
		if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
		{
			if (fd.cFileName[0] != _T('.'))
			{
				_tcscpy_s(szFind, lpstrDir);
				_tcscat_s(szFind, _T("\\"));
				_tcscat_s(szFind, fd.cFileName);

				DeleteDirectory(szFind);
			}
		}
		else
		{
			MYString strFile = lpstrDir;
			strFile += _T("\\");
			strFile += fd.cFileName;

			::DeleteFile(strFile.c_str());
		}

		if (!::FindNextFile(hFind, &fd))
		{
			break;
		}
	}

	if (hFind != NULL && hFind != INVALID_HANDLE_VALUE)
	{
		::FindClose(hFind);
		hFind = NULL;
	}

	return true;
}

int _tmain(int argc, _TCHAR* argv[])
{
	MYString file_name;

	std::wcout << "请输入文件名：" << std::endl;
	std::wcin >> file_name;

	DeleteDirectory(file_name.c_str());

	return 0;
}

---

欢迎关注我的微博