mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-27 18:12:02 +00:00
78 lines
1.4 KiB
C++
78 lines
1.4 KiB
C++
#include <mutex>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
|
|
#include <Common/ProfileEvents.h>
|
|
#include <Common/Exception.h>
|
|
#include <IO/OpenedFile.h>
|
|
|
|
|
|
namespace ProfileEvents
|
|
{
|
|
extern const Event FileOpen;
|
|
}
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int FILE_DOESNT_EXIST;
|
|
extern const int CANNOT_OPEN_FILE;
|
|
extern const int CANNOT_CLOSE_FILE;
|
|
}
|
|
|
|
|
|
void OpenedFile::open() const
|
|
{
|
|
ProfileEvents::increment(ProfileEvents::FileOpen);
|
|
|
|
fd = ::open(file_name.c_str(), (flags == -1 ? 0 : flags) | O_RDONLY | O_CLOEXEC);
|
|
|
|
if (-1 == fd)
|
|
throwFromErrnoWithPath("Cannot open file " + file_name, file_name,
|
|
errno == ENOENT ? ErrorCodes::FILE_DOESNT_EXIST : ErrorCodes::CANNOT_OPEN_FILE);
|
|
}
|
|
|
|
int OpenedFile::getFD() const
|
|
{
|
|
std::lock_guard l(mutex);
|
|
if (fd == -1)
|
|
open();
|
|
return fd;
|
|
}
|
|
|
|
std::string OpenedFile::getFileName() const
|
|
{
|
|
return file_name;
|
|
}
|
|
|
|
|
|
OpenedFile::OpenedFile(const std::string & file_name_, int flags_)
|
|
: file_name(file_name_), flags(flags_)
|
|
{
|
|
}
|
|
|
|
|
|
OpenedFile::~OpenedFile()
|
|
{
|
|
close(); /// Exceptions will lead to std::terminate and that's Ok.
|
|
}
|
|
|
|
|
|
void OpenedFile::close()
|
|
{
|
|
std::lock_guard l(mutex);
|
|
if (fd == -1)
|
|
return;
|
|
|
|
if (0 != ::close(fd))
|
|
throw Exception("Cannot close file", ErrorCodes::CANNOT_CLOSE_FILE);
|
|
|
|
fd = -1;
|
|
metric_increment.destroy();
|
|
}
|
|
|
|
}
|
|
|