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
|
#include "json2yson.h"
#include <library/cpp/yson/parser.h>
#include <library/cpp/yson/json/json_writer.h>
#include <library/cpp/yson/json/yson2json_adapter.h>
namespace NJson2Yson {
static void WriteJsonValue(const NJson::TJsonValue& jsonValue, NYT::TYson2JsonCallbacksAdapter* adapter) {
switch (jsonValue.GetType()) {
default:
case NJson::JSON_NULL:
adapter->OnNull();
break;
case NJson::JSON_BOOLEAN:
adapter->OnBoolean(jsonValue.GetBoolean());
break;
case NJson::JSON_DOUBLE:
adapter->OnDouble(jsonValue.GetDouble());
break;
case NJson::JSON_INTEGER:
adapter->OnInteger(jsonValue.GetInteger());
break;
case NJson::JSON_UINTEGER:
adapter->OnUInteger(jsonValue.GetUInteger());
break;
case NJson::JSON_STRING:
adapter->OnString(jsonValue.GetString());
break;
case NJson::JSON_ARRAY: {
adapter->OnOpenArray();
const NJson::TJsonValue::TArray& arr = jsonValue.GetArray();
for (const auto& it : arr)
WriteJsonValue(it, adapter);
adapter->OnCloseArray();
break;
}
case NJson::JSON_MAP: {
adapter->OnOpenMap();
const NJson::TJsonValue::TMapType& map = jsonValue.GetMap();
for (const auto& it : map) {
adapter->OnMapKey(it.first);
WriteJsonValue(it.second, adapter);
}
adapter->OnCloseMap();
break;
}
}
}
void SerializeJsonValueAsYson(const NJson::TJsonValue& inputValue, NYson::TYsonWriter* ysonWriter) {
NYT::TYson2JsonCallbacksAdapter adapter(ysonWriter);
WriteJsonValue(inputValue, &adapter);
}
void SerializeJsonValueAsYson(const NJson::TJsonValue& inputValue, IOutputStream* outputStream) {
NYson::TYsonWriter ysonWriter(outputStream, NYson::EYsonFormat::Binary, ::NYson::EYsonType::Node, false);
SerializeJsonValueAsYson(inputValue, &ysonWriter);
}
void SerializeJsonValueAsYson(const NJson::TJsonValue& inputValue, TString& result) {
TStringOutput resultStream(result);
SerializeJsonValueAsYson(inputValue, &resultStream);
}
TString SerializeJsonValueAsYson(const NJson::TJsonValue& inputValue) {
TString result;
SerializeJsonValueAsYson(inputValue, result);
return result;
}
bool DeserializeYsonAsJsonValue(IInputStream* inputStream, NJson::TJsonValue* outputValue, bool throwOnError) {
NJson::TParserCallbacks parser(*outputValue);
NJson2Yson::TJsonBuilder consumer(&parser);
NYson::TYsonParser ysonParser(&consumer, inputStream, ::NYson::EYsonType::Node);
try {
ysonParser.Parse();
} catch (...) {
if (throwOnError) {
throw;
}
return false;
}
return true;
}
bool DeserializeYsonAsJsonValue(TStringBuf str, NJson::TJsonValue* outputValue, bool throwOnError) {
TMemoryInput inputStream(str);
return DeserializeYsonAsJsonValue(&inputStream, outputValue, throwOnError);
}
void ConvertYson2Json(IInputStream* inputStream, IOutputStream* outputStream) {
NYT::TJsonWriter writer(outputStream, ::NYson::EYsonType::Node, NYT::JF_TEXT, NYT::JAM_ON_DEMAND, NYT::SBF_BOOLEAN);
NYson::TYsonParser ysonParser(&writer, inputStream, ::NYson::EYsonType::Node);
ysonParser.Parse();
}
void ConvertYson2Json(TStringBuf yson, IOutputStream* outputStream) {
TMemoryInput inputStream(yson);
ConvertYson2Json(&inputStream, outputStream);
}
TString ConvertYson2Json(TStringBuf yson) {
TString json;
TStringOutput outputStream(json);
ConvertYson2Json(yson, &outputStream);
return json;
}
}
|
