ClickHouse/src/Common/filesystemHelpers.h

#pragma once

#include <common/types.h>
#include <Common/Exception.h>

#include <filesystem>
#include <memory>
#include <string>
#include <cerrno>
#include <sys/statvfs.h>
#include <Poco/TemporaryFile.h>


namespace DB
{
namespace ErrorCodes
{
    extern const int CANNOT_STATVFS;
}

using TemporaryFile = Poco::TemporaryFile;

bool enoughSpaceInDirectory(const std::string & path, size_t data_size);
std::unique_ptr<TemporaryFile> createTemporaryFile(const std::string & path);

/// Returns mount point of filesystem where absolute_path (must exist) is located
std::filesystem::path getMountPoint(std::filesystem::path absolute_path);

/// Returns name of filesystem mounted to mount_point
#if !defined(__linux__)
[[noreturn]]
#endif
String getFilesystemName([[maybe_unused]] const String & mount_point);

inline struct statvfs getStatVFS(const String & path)
{
    struct statvfs fs;
    while (statvfs(path.c_str(), &fs) != 0)
    {
        if (errno == EINTR)
            continue;
        throwFromErrnoWithPath("Could not calculate available disk space (statvfs)", path, ErrorCodes::CANNOT_STATVFS);
    }
    return fs;
}

}