blob: 2d78107a93f670185c0c2c4304e4672c88356ae7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
#pragma once
#include "buffer.h"
#include <util/generic/ptr.h>
#include <util/generic/strbuf.h>
#include <util/system/types.h>
#include <util/system/yassert.h>
#include <cstddef>
#include <cstdio>
#include <cstring>
#include <memory>
//! \brief Output stream adaptor interface.
//!
//! Represents a model of an optionally-buffered writer.
namespace NYsonPull {
namespace NOutput {
class IStream {
output_buffer buffer_;
public:
virtual ~IStream() = default;
output_buffer& buffer() noexcept {
return buffer_;
}
const output_buffer& buffer() const noexcept {
return buffer_;
}
void flush_buffer(TStringBuf extra = {}) {
if (!extra.empty() || !buffer_.is_empty()) {
do_flush_buffer(extra);
}
while (!buffer_.is_empty()) {
do_flush_buffer({});
}
}
protected:
//! \brief Flush data to underlying stream.
//!
//! The implementation is to flush the buffer contents AND
//! extra argument to underlying stream.
//!
//! This way, at zero buffer size this interface implements an unbuffered
//! stream (with an added cost of a virtual call per each write).
//!
//! Write is assumed to always succeed unless it throws an exception.
virtual void do_flush_buffer(TStringBuf extra) = 0;
};
//! \brief Write data to C FILE* object.
THolder<IStream> FromStdioFile(FILE* file, size_t buffer_size = 0);
//! \brief Write data to POSIX file descriptor
THolder<IStream> FromPosixFd(int fd, size_t buffer_size = 65536);
THolder<IStream> FromOutputStream(IOutputStream* output, size_t buffer_size = 65536);
THolder<IStream> FromString(TString* output, size_t buffer_size = 1024);
}
}
|
