blob: d937d1869190e5ff1c971c1dd37459bd4cd13a33 (
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#pragma once
#include "consumer.h"
#include "output.h"
#include "scalar.h"
#include "stream_type.h"
#include <memory>
namespace NYsonPull {
//! \brief YSON writer facade class
//!
//! Owns a YSON consumer and a corresponding output stream.
//! Methods invoke corresponding \p NYsonPull::IConsumer methods and can be chained.
class TWriter {
THolder<NOutput::IStream> Stream_;
THolder<IConsumer> Impl_;
public:
TWriter(
THolder<NOutput::IStream> stream,
THolder<IConsumer> impl)
: Stream_{std::move(stream)}
, Impl_{std::move(impl)} {
}
//! \brief Get a reference to underlying consumer.
//!
//! Useful with \p NYsonPull::bridge
IConsumer& GetConsumer() {
return *Impl_;
}
TWriter& BeginStream() {
Impl_->OnBeginStream();
return *this;
}
TWriter& EndStream() {
Impl_->OnEndStream();
return *this;
}
TWriter& BeginList() {
Impl_->OnBeginList();
return *this;
}
TWriter& EndList() {
Impl_->OnEndList();
return *this;
}
TWriter& BeginMap() {
Impl_->OnBeginMap();
return *this;
}
TWriter& EndMap() {
Impl_->OnEndMap();
return *this;
}
TWriter& BeginAttributes() {
Impl_->OnBeginAttributes();
return *this;
}
TWriter& EndAttributes() {
Impl_->OnEndAttributes();
return *this;
}
TWriter& Key(TStringBuf name) {
Impl_->OnKey(name);
return *this;
}
TWriter& Entity() {
Impl_->OnEntity();
return *this;
}
TWriter& Boolean(bool value) {
Impl_->OnScalarBoolean(value);
return *this;
}
TWriter& Int64(i64 value) {
Impl_->OnScalarInt64(value);
return *this;
}
TWriter& UInt64(ui64 value) {
Impl_->OnScalarUInt64(value);
return *this;
}
TWriter& Float64(double value) {
Impl_->OnScalarFloat64(value);
return *this;
}
TWriter& String(TStringBuf value) {
Impl_->OnScalarString(value);
return *this;
}
TWriter& Scalar(const TScalar& value) {
Impl_->OnScalar(value);
return *this;
}
TWriter& Event(const TEvent& value) {
Impl_->OnEvent(value);
return *this;
}
};
//! \brief Construct a writer for binary YSON format.
TWriter MakeBinaryWriter(
THolder<NOutput::IStream> stream,
EStreamType mode);
//! \brief Construct a writer for text YSON format.
TWriter MakeTextWriter(
THolder<NOutput::IStream> stream,
EStreamType mode);
//! \brief Construct a writer for pretty text YSON format.
TWriter MakePrettyTextWriter(
THolder<NOutput::IStream> stream,
EStreamType mode,
size_t indent_size = 4);
}
|
