fs/io/FileOutputStream: merge all classes into one, add enum Mode

Prepare to add more modes.
This commit is contained in:
Max Kellermann
2016-09-04 12:07:36 +02:00
parent b630afdeda
commit d775f13a03
3 changed files with 122 additions and 140 deletions

View File

@@ -38,7 +38,7 @@
class Path;
class BaseFileOutputStream : public OutputStream {
class FileOutputStream final : public OutputStream {
const AllocatedPath path;
#ifdef WIN32
@@ -47,40 +47,58 @@ class BaseFileOutputStream : public OutputStream {
FileDescriptor fd = FileDescriptor::Undefined();
#endif
protected:
#ifdef WIN32
template<typename P>
BaseFileOutputStream(P &&_path)
:path(std::forward<P>(_path)) {}
#else
template<typename P>
BaseFileOutputStream(P &&_path)
:path(std::forward<P>(_path)) {}
#ifdef HAVE_LINKAT
/**
* Was O_TMPFILE used? If yes, then linkat() must be used to
* create a link to this file.
*/
bool is_tmpfile = false;
#endif
~BaseFileOutputStream() {
assert(!IsDefined());
public:
enum class Mode : uint8_t {
/**
* Create a new file, or replace an existing file.
* File contents may not be visible until Commit() has
* been called.
*/
CREATE,
/**
* Append to a file that already exists. If it does
* not, an exception is thrown.
*/
APPEND_EXISTING,
};
private:
Mode mode;
public:
FileOutputStream(Path _path, Mode _mode=Mode::CREATE);
~FileOutputStream() {
if (IsDefined())
Cancel();
}
#ifdef WIN32
void SetHandle(HANDLE _handle) {
assert(!IsDefined());
handle = _handle;
assert(IsDefined());
}
#else
FileDescriptor &SetFD() {
assert(!IsDefined());
return fd;
public:
Path GetPath() const {
return path;
}
const FileDescriptor &GetFD() const {
return fd;
}
#endif
gcc_pure
uint64_t Tell() const;
/* virtual methods from class OutputStream */
void Write(const void *data, size_t size) override;
void Commit();
void Cancel();
private:
void OpenCreate();
void OpenAppendExisting();
bool Close() {
assert(IsDefined());
@@ -108,50 +126,6 @@ protected:
return fd.IsDefined();
#endif
}
public:
Path GetPath() const {
return path;
}
gcc_pure
uint64_t Tell() const;
/* virtual methods from class OutputStream */
void Write(const void *data, size_t size) override;
};
class FileOutputStream final : public BaseFileOutputStream {
#ifdef HAVE_LINKAT
/**
* Was O_TMPFILE used? If yes, then linkat() must be used to
* create a link to this file.
*/
bool is_tmpfile;
#endif
public:
FileOutputStream(Path _path);
~FileOutputStream() {
if (IsDefined())
Cancel();
}
void Commit();
void Cancel();
};
class AppendFileOutputStream final : public BaseFileOutputStream {
public:
AppendFileOutputStream(Path _path);
~AppendFileOutputStream() {
if (IsDefined())
Close();
}
void Commit();
};
#endif