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
|
#pragma once
#include "json_output.h"
#include "config.h"
#include <library/cpp/json/json_writer.h>
#include <util/string/builder.h>
#include <util/generic/store_policy.h>
namespace NProtobufJson {
class TBaseJsonWriterOutput: public IJsonOutput {
public:
TBaseJsonWriterOutput(NJson::TJsonWriter& writer)
: Writer(writer)
{
}
private:
void DoWrite(int i) override {
Writer.Write(i);
}
void DoWrite(unsigned int i) override {
Writer.Write(i);
}
void DoWrite(long long i) override {
Writer.Write(i);
}
void DoWrite(unsigned long long i) override {
Writer.Write(i);
}
void DoWrite(float f) override {
Writer.Write(f);
}
void DoWrite(double f) override {
Writer.Write(f);
}
void DoWrite(bool b) override {
Writer.Write(b);
}
void DoWriteNull() override {
Writer.WriteNull();
}
void DoWrite(const TStringBuf& s) override {
Writer.Write(s);
}
void DoWrite(const TString& s) override {
Writer.Write(s);
}
void DoBeginList() override {
Writer.OpenArray();
}
void DoEndList() override {
Writer.CloseArray();
}
void DoBeginObject() override {
Writer.OpenMap();
}
void DoWriteKey(const TStringBuf& key) override {
Writer.Write(key);
}
void DoEndObject() override {
Writer.CloseMap();
}
void DoWriteRawJson(const TStringBuf& str) override {
Writer.UnsafeWrite(str);
}
NJson::TJsonWriter& Writer;
};
class TJsonWriterOutput: public TEmbedPolicy<NJson::TJsonWriter>, public TBaseJsonWriterOutput {
public:
TJsonWriterOutput(IOutputStream* outputStream, const NJson::TJsonWriterConfig& cfg)
: TEmbedPolicy<NJson::TJsonWriter>(outputStream, cfg)
, TBaseJsonWriterOutput(*Ptr())
{
}
TJsonWriterOutput(IOutputStream* outputStream, const TProto2JsonConfig& cfg)
: TEmbedPolicy<NJson::TJsonWriter>(outputStream, CreateJsonWriterConfig(cfg))
, TBaseJsonWriterOutput(*Ptr())
{
}
protected:
static NJson::TJsonWriterConfig CreateJsonWriterConfig(const TProto2JsonConfig& cfg);
};
class TJsonStringWriterOutput: public TEmbedPolicy<TStringOutput>, public TJsonWriterOutput {
public:
template <typename TConfig>
TJsonStringWriterOutput(TString* str, const TConfig& cfg)
: TEmbedPolicy<TStringOutput>(*str)
// If Unbuffered = false, TJsonWriter uses its own string and then flushes it into TStringOutput.
, TJsonWriterOutput(TEmbedPolicy<TStringOutput>::Ptr(), CreateJsonWriterConfig(cfg).SetUnbuffered(true))
{
}
};
}
|
