diff options
author | Anton Samokhvalov <pg83@yandex.ru> | 2022-02-10 16:45:15 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:45:15 +0300 |
commit | 72cb13b4aff9bc9cf22e49251bc8fd143f82538f (patch) | |
tree | da2c34829458c7d4e74bdfbdf85dff449e9e7fb8 /library/cpp/yson_pull/detail/output | |
parent | 778e51ba091dc39e7b7fcab2b9cf4dbedfb6f2b5 (diff) | |
download | ydb-72cb13b4aff9bc9cf22e49251bc8fd143f82538f.tar.gz |
Restoring authorship annotation for Anton Samokhvalov <pg83@yandex.ru>. Commit 1 of 2.
Diffstat (limited to 'library/cpp/yson_pull/detail/output')
-rw-r--r-- | library/cpp/yson_pull/detail/output/buffered.h | 80 | ||||
-rw-r--r-- | library/cpp/yson_pull/detail/output/stdio_file.h | 38 | ||||
-rw-r--r-- | library/cpp/yson_pull/detail/output/stream.h | 8 |
3 files changed, 63 insertions, 63 deletions
diff --git a/library/cpp/yson_pull/detail/output/buffered.h b/library/cpp/yson_pull/detail/output/buffered.h index 475cf34785..033d8859e0 100644 --- a/library/cpp/yson_pull/detail/output/buffered.h +++ b/library/cpp/yson_pull/detail/output/buffered.h @@ -6,46 +6,46 @@ #include <util/generic/strbuf.h> -namespace NYsonPull { - namespace NDetail { - namespace NOutput { - template <typename T> - class TBuffered: public NYsonPull::NOutput::IStream { - TArrayHolder<ui8> buffer_; - size_t size_; - - public: - TBuffered(size_t buffer_size) - : buffer_{new ui8[buffer_size]} - , size_{buffer_size} { - reset_buffer(); - } - - protected: - void do_flush_buffer(TStringBuf extra) override { - auto& buf = buffer(); - if (!buf.is_empty()) { +namespace NYsonPull { + namespace NDetail { + namespace NOutput { + template <typename T> + class TBuffered: public NYsonPull::NOutput::IStream { + TArrayHolder<ui8> buffer_; + size_t size_; + + public: + TBuffered(size_t buffer_size) + : buffer_{new ui8[buffer_size]} + , size_{buffer_size} { + reset_buffer(); + } + + protected: + void do_flush_buffer(TStringBuf extra) override { + auto& buf = buffer(); + if (!buf.is_empty()) { do_write({reinterpret_cast<const char*>(buf.begin()), buf.used()}); - reset_buffer(); - } - if (extra.size() >= buf.available()) { - do_write(extra); + reset_buffer(); + } + if (extra.size() >= buf.available()) { + do_write(extra); } else if (extra.size() > 0) { - ::memcpy(buf.pos(), extra.data(), extra.size()); - buf.advance(extra.size()); - } - } - - private: - void do_write(TStringBuf data) { - // CRTP dispatch - static_cast<T*>(this)->write(data); - } - - void reset_buffer() { - buffer().reset(buffer_.Get(), buffer_.Get() + size_); - } - }; + ::memcpy(buf.pos(), extra.data(), extra.size()); + buf.advance(extra.size()); + } + } + + private: + void do_write(TStringBuf data) { + // CRTP dispatch + static_cast<T*>(this)->write(data); + } + + void reset_buffer() { + buffer().reset(buffer_.Get(), buffer_.Get() + size_); + } + }; } - } // namespace NDetail -} + } // namespace NDetail +} diff --git a/library/cpp/yson_pull/detail/output/stdio_file.h b/library/cpp/yson_pull/detail/output/stdio_file.h index 03f2b40dc5..0b9cbba894 100644 --- a/library/cpp/yson_pull/detail/output/stdio_file.h +++ b/library/cpp/yson_pull/detail/output/stdio_file.h @@ -8,26 +8,26 @@ #include <cstdio> -namespace NYsonPull { - namespace NDetail { - namespace NOutput { - class TStdioFile: public TBuffered<TStdioFile> { - FILE* file_; +namespace NYsonPull { + namespace NDetail { + namespace NOutput { + class TStdioFile: public TBuffered<TStdioFile> { + FILE* file_; - public: - TStdioFile(FILE* file, size_t buffer_size) - : TBuffered<TStdioFile>(buffer_size) - , file_(file) - { - } + public: + TStdioFile(FILE* file, size_t buffer_size) + : TBuffered<TStdioFile>(buffer_size) + , file_(file) + { + } - void write(TStringBuf data) { - auto nwritten = ::fwrite(data.data(), 1, data.size(), file_); + void write(TStringBuf data) { + auto nwritten = ::fwrite(data.data(), 1, data.size(), file_); if (Y_UNLIKELY(static_cast<size_t>(nwritten) != data.size())) { - throw NException::TSystemError(); - } - } - }; + throw NException::TSystemError(); + } + } + }; } - } // namespace NDetail -} + } // namespace NDetail +} diff --git a/library/cpp/yson_pull/detail/output/stream.h b/library/cpp/yson_pull/detail/output/stream.h index d4810f3353..39f3b33c4c 100644 --- a/library/cpp/yson_pull/detail/output/stream.h +++ b/library/cpp/yson_pull/detail/output/stream.h @@ -27,12 +27,12 @@ namespace NYsonPull { } }; - template <typename TOutput> + template <typename TOutput> class TOwned: public TBuffered<TOwned<TOutput>> { TOutput Output; public: - template <typename... Args> + template <typename... Args> TOwned(size_t buffer_size, Args&&... args) : TBuffered<TOwned>(buffer_size) , Output(std::forward<Args>(args)...) @@ -52,5 +52,5 @@ namespace NYsonPull { } }; } - } // namespace NDetail -} + } // namespace NDetail +} |