aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/protobuf/json/json_writer_output.h
blob: 975245a264b9a84f88e6a068f785686fd5475e14 (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
#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))
        {
        }
    };

}