- 1. 关键词
- 2. fileutil.h
- 3. fileutil.cpp
- 4. filesystem_win.h
- 5. filesystem_win.cpp
- 6. filesystem_unix.cpp
- 7. 源码地址
1. 关键词
C++ 文件系统操作 创建文件夹 创建多级目录文件夹 删除文件夹 删除文件夹下的所有文件和子目录 跨平台
2. fileutil.h
#pragma once
#include <string>
#include <cstdio>
#include <cstdint>
#include "filetype.h"
#include "filepath.h"
namespace cutl
{
/**
* @brief The file guard class to manage the FILE pointer automatically.
* file_guard object can close the FILE pointer automatically when his scope is exit.
*/
class file_guard
{
public:
/**
* @brief Construct a new file guard object
*
* @param file the pointer of the FILE object
*/
explicit file_guard(FILE *file);
/**
* @brief Destroy the file guard object
*
*/
~file_guard();
/**
* @brief Get the FILE pointer.
*
* @return FILE*
*/
FILE *getfd() const;
private:
FILE *file_;
};
/**
* @brief Create a new directory.
*
* @param path the filepath of the new directory to be created
* @param recursive whether to create parent directories, default is false.
* If true, means create parent directories if not exist, like the 'mkdir -p' command.
* @return true if the directory is created successfully, false otherwise.
*/
bool createdir(const filepath &path, bool recursive = false);
/**
* @brief Remove a directory.
*
* @param path the filepath of the directory to be removed
* @param recursive whether to remove the whole directory recursively, default is false.
* If true, means remove the whole directory recursively, like the 'rm -rf' command.
* @return true if the directory is removed successfully, false otherwise.
*/
bool removedir(const filepath &path, bool recursive = false);
} // namespace cutl
3. fileutil.cpp
#include <cstdio>
#include <map>
#include <iostream>
#include <cstring>
#include <sys/stat.h>
#include "fileutil.h"
#include "inner/logger.h"
#include "inner/filesystem.h"
#include "strutil.h"
namespace cutl
{
file_guard::file_guard(FILE *file)
: file_(file)
{
}
file_guard::~file_guard()
{
if (file_)
{
// CUTL_DEBUG("close file");
int ret = fclose(file_);
if (ret != 0)
{
CUTL_ERROR("fail to close file, ret" + std::to_string(ret));
}
file_ = nullptr;
}
// ROBOLOG_DCHECK(file_ == nullptr);
}
FILE *file_guard::getfd() const
{
return file_;
}
bool createdir(const filepath &path, bool recursive)
{
if (recursive)
{
constexpr int buf_size = MAX_PATH_LEN;
char buffer[buf_size] = {0};
int ret = snprintf(buffer, buf_size, "%s", path.str().c_str());
if (ret < 0 || ret >= buf_size)
{
CUTL_ERROR("invalid path: " + path.str());
return false;
}
int len = strlen(buffer);
if (buffer[len - 1] != filepath::separator())
{
buffer[len++] = filepath::separator();
}
int32_t idx = (buffer[0] == filepath::separator()) ? 1 : 0;
for (; idx < len; ++idx)
{
if (buffer[idx] != filepath::separator())
{
continue;
}
buffer[idx] = '\0';
filepath temp_path(buffer);
if (!temp_path.exists())
{
if (!create_dir(temp_path.str()))
{
CUTL_ERROR("createdir error. dir:" + temp_path.str());
return false;
}
}
buffer[idx] = filepath::separator();
}
return true;
}
else
{
auto dirPath = path.dirname();
if (dirPath.empty())
{
CUTL_ERROR("invalid path: " + path.str());
return false;
}
if (!cutl::path(dirPath).exists())
{
CUTL_ERROR("directory does not exist: " + dirPath);
return false;
}
return create_dir(path.str());
}
}
bool removedir(const filepath &path, bool recursive)
{
if (!path.exists())
{
CUTL_ERROR("directory does not exist: " + path.str());
return false;
}
if (recursive)
{
return remove_dir_recursive(path.str());
}
else
{
return remove_dir(path.str());
}
}
4. filesystem_win.h
#include <vector>
#include <string>
#pragma once
namespace cutl
{
bool create_dir(const std::string &dir_path);
// remove empty directory
bool remove_dir(const std::string &dir_path);
// remove directory recursively
bool remove_dir_recursive(const std::string &dir_path);
} // namespace cutl
5. filesystem_win.cpp
#if defined(_WIN32) || defined(__WIN32__)
#include <io.h>
#include <direct.h>
#include <Windows.h>
#include <stdlib.h>
#include "strutil.h"
#include "filesystem.h"
#include "logger.h"
namespace cutl
{
bool create_dir(const std::string &dir_path)
{
if (_mkdir(dir_path.c_str()) != 0)
{
CUTL_ERROR("mkdir error. dir_path:" + dir_path + ", error:" + strerror(errno));
return false;
}
return true;
}
// remove empty directory
// https://learn.microsoft.com/en-us/cpp/c-runtime-library/reference/rmdir-wrmdir?view=msvc-170
bool remove_dir(const std::string &dir_path)
{
if (_rmdir(dir_path.c_str()) != 0)
{
CUTL_ERROR("rmdir error. dir_path:" + dir_path + ", error:" + strerror(errno));
return false;
}
return true;
}
// remove directory recursively
bool remove_dir_recursive(const std::string &dir_path)
{
auto findpath = dir_path + win_separator + "*.*";
WIN32_FIND_DATAA data = {0};
HANDLE hFind = FindFirstFileA(findpath.c_str(), &data);
bool unicode = true;
if (hFind == INVALID_HANDLE_VALUE || hFind == NULL)
{
CUTL_ERROR("FindFirstFileA failed for " + findpath + ", errCode: " + std::to_string(GetLastError()));
return false;
}
do
{
auto dwAttrs = data.dwFileAttributes;
auto filename = std::string(data.cFileName);
if (is_special_dir(filename))
{
// “..”和“.”不做处理
continue;
}
std::string filepath = dir_path + win_separator + filename;
if ((dwAttrs & FILE_ATTRIBUTE_DIRECTORY))
{
// directory
if (!remove_dir_recursive(filepath))
{
FindClose(hFind);
return false;
}
}
else
{
// file
int ret = remove(filepath.c_str());
if (ret != 0)
{
CUTL_ERROR("remove " + filepath + " error, ret:" + std::to_string(ret));
return false;
}
}
} while (FindNextFileA(hFind, &data));
// 关闭句柄
FindClose(hFind);
// 删除当前文件夹
if (_rmdir(dir_path.c_str()) != 0)
{
CUTL_ERROR("rmdir error. dir_path:" + dir_path + ", error:" + strerror(errno));
return false;
}
return true;
}
} // namespace cutl
#endif // defined(_WIN32) || defined(__WIN32__)
6. filesystem_unix.cpp
#if defined(_WIN32) || defined(__WIN32__)
// do nothing
#else
#include <unistd.h>
#include <sys/stat.h>
#include <dirent.h>
#include <stack>
#include <cstring>
#include <utime.h>
#include <stdlib.h>
#include <sys/time.h>
#include "filesystem.h"
#include "inner/logger.h"
namespace cutl
{
bool create_dir(const std::string &dir_path)
{
if (mkdir(dir_path.c_str(), S_IRWXU | S_IRWXG | S_IRWXO | S_IWOTH) != 0)
{
CUTL_ERROR("mkdir error. dir_path:" + dir_path + ", error:" + strerror(errno));
return false;
}
return true;
}
bool remove_dir(const std::string &dir_path)
{
if (rmdir(dir_path.c_str()) != 0)
{
CUTL_ERROR("rmdir error. dir_path:" + dir_path + ", error:" + strerror(errno));
return false;
}
return true;
}
bool remove_dir_recursive(const std::string &dir_path)
{
DIR *dir = opendir(dir_path.c_str()); // 打开这个目录
if (dir == NULL)
{
CUTL_ERROR("opendir error. dir_path:" + dir_path + ", error:" + strerror(errno));
return false;
}
struct dirent *file_info = NULL;
// 逐个读取目录中的文件到file_info
while ((file_info = readdir(dir)) != NULL)
{
// 系统有个系统文件，名为“..”和“.”,对它不做处理
std::string filename(file_info->d_name);
if (is_special_dir(filename))
{
continue;
}
struct stat file_stat; // 文件的信息
std::string filepath = dir_path + unix_separator + filename;
int ret = lstat(filepath.c_str(), &file_stat);
if (0 != ret)
{
CUTL_ERROR("stat error. filepath:" + filepath + ", error:" + strerror(errno));
closedir(dir);
return false;
}
if (S_ISDIR(file_stat.st_mode))
{
if (!remove_dir_recursive(filepath))
{
closedir(dir);
return false;
}
}
else
{
int ret = remove(filepath.c_str());
if (ret != 0)
{
CUTL_ERROR("remove " + filepath + " error, ret:" + std::to_string(ret));
closedir(dir);
return false;
}
}
}
closedir(dir);
// 删除当前文件夹
int ret = rmdir(dir_path.c_str());
if (ret != 0)
{
CUTL_ERROR("rmdir error. dir_path:" + dir_path + ", error:" + strerror(errno));
return false;
}
return true;
}
} // namespace cutl
#endif // defined(_WIN32) || defined(__WIN32__)
7. 源码地址
更多详细代码，请查看本人写的C++ 通用工具库: common_util, 本项目已开源，代码简洁，且有详细的文档和Demo。
标签：return,C++,跨平台,文件夹,file,path,false,include,dir From： https://www.cnblogs.com/luoweifu/p/18282684