library/cpp/yson/json/yson2json_adapter.h


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

#pragma once 
 
#include <library/cpp/yson/consumer.h>
 
#include <library/cpp/json/json_reader.h>
 
#include <util/generic/stack.h> 
 
namespace NYT { 
    class TYson2JsonCallbacksAdapter 
       : public NJson::TJsonCallbacks { 
    public: 
        class TState {
        private:
            // Stores current context stack
            // If true - we are in a list
            // If false - we are in a map
            TStack<bool> ContextStack;

            friend class TYson2JsonCallbacksAdapter;
        };

    public:
        TYson2JsonCallbacksAdapter(::NYson::TYsonConsumerBase* impl, bool throwException = false);
 
        bool OnNull() override; 
        bool OnBoolean(bool val) override; 
        bool OnInteger(long long val) override; 
        bool OnUInteger(unsigned long long val) override; 
        bool OnString(const TStringBuf& val) override; 
        bool OnDouble(double val) override; 
        bool OnOpenArray() override; 
        bool OnCloseArray() override; 
        bool OnOpenMap() override; 
        bool OnCloseMap() override; 
        bool OnMapKey(const TStringBuf& val) override; 
 
        TState State() const {
            return State_;
        }

        void Reset(const TState& state) {
            State_ = state;
        }

    private: 
        void WrapIfListItem(); 
 
    private: 
        ::NYson::TYsonConsumerBase* Impl_;
        TState State_;
    }; 
}