diff options
| author | Devtools Arcadia <[email protected]> | 2022-02-07 18:08:42 +0300 |
|---|---|---|
| committer | Devtools Arcadia <[email protected]> | 2022-02-07 18:08:42 +0300 |
| commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
| tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /util/stream/direct_io.h | |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'util/stream/direct_io.h')
| -rw-r--r-- | util/stream/direct_io.h | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/util/stream/direct_io.h b/util/stream/direct_io.h new file mode 100644 index 00000000000..2e1f2e07dd7 --- /dev/null +++ b/util/stream/direct_io.h @@ -0,0 +1,43 @@ +#pragma once + +#include "input.h" +#include "output.h" +#include <util/system/direct_io.h> + +class TRandomAccessFileInput: public IInputStream { +public: + TRandomAccessFileInput(TDirectIOBufferedFile& file, ui64 position); + +protected: + size_t DoRead(void* buf, size_t len) override; + size_t DoSkip(size_t len) override; + +private: + TDirectIOBufferedFile& File; + ui64 Position; +}; + +class TRandomAccessFileOutput: public IOutputStream { +public: + TRandomAccessFileOutput(TDirectIOBufferedFile& file); + + TRandomAccessFileOutput(TRandomAccessFileOutput&&) noexcept = default; + TRandomAccessFileOutput& operator=(TRandomAccessFileOutput&&) noexcept = default; + +protected: + TDirectIOBufferedFile* File; + +private: + void DoWrite(const void* buf, size_t len) override; + void DoFlush() override; +}; + +class TBufferedFileOutputEx: public TRandomAccessFileOutput { +public: + TBufferedFileOutputEx(const TString& path, EOpenMode oMode, size_t buflen = 1 << 17); + +private: + void DoFlush() override; + void DoFinish() override; + THolder<TDirectIOBufferedFile> FileHolder; +}; |
