summaryrefslogtreecommitdiffstats
path: root/util/stream/direct_io.h
diff options
context:
space:
mode:
authorDevtools Arcadia <[email protected]>2022-02-07 18:08:42 +0300
committerDevtools Arcadia <[email protected]>2022-02-07 18:08:42 +0300
commit1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch)
treee26c9fed0de5d9873cce7e00bc214573dc2195b7 /util/stream/direct_io.h
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'util/stream/direct_io.h')
-rw-r--r--util/stream/direct_io.h43
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;
+};