intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 154 insertions, 0 deletions

diff --git a/library/cpp/yson/node/node_io.cpp b/library/cpp/yson/node/node_io.cpp
new file mode 100644
index 0000000000..294a7f7217
--- /dev/null
+++ b/library/cpp/yson/node/node_io.cpp
@@ -0,0 +1,154 @@
+#include "node_io.h"
+
+#include "node_builder.h"
+#include "node_visitor.h"
+
+#include <library/cpp/yson/json/json_writer.h>
+#include <library/cpp/yson/parser.h>
+#include <library/cpp/yson/writer.h>
+#include <library/cpp/yson/json/yson2json_adapter.h>
+
+#include <library/cpp/json/json_reader.h>
+#include <library/cpp/json/json_value.h>
+
+#include <util/stream/input.h>
+#include <util/stream/output.h>
+#include <util/stream/str.h>
+#include <util/stream/mem.h>
+
+namespace NYT {
+
+static void WalkJsonTree(const NJson::TJsonValue& jsonValue, NJson::TJsonCallbacks* callbacks)
+{
+    using namespace NJson;
+    switch (jsonValue.GetType()) {
+        case JSON_NULL:
+            callbacks->OnNull();
+            return;
+        case JSON_BOOLEAN:
+            callbacks->OnBoolean(jsonValue.GetBoolean());
+            return;
+        case JSON_INTEGER:
+            callbacks->OnInteger(jsonValue.GetInteger());
+            return;
+        case JSON_UINTEGER:
+            callbacks->OnUInteger(jsonValue.GetUInteger());
+            return;
+        case JSON_DOUBLE:
+            callbacks->OnDouble(jsonValue.GetDouble());
+            return;
+        case JSON_STRING:
+            callbacks->OnString(jsonValue.GetString());
+            return;
+        case JSON_MAP:
+            {
+                callbacks->OnOpenMap();
+                for (const auto& item : jsonValue.GetMap()) {
+                    callbacks->OnMapKey(item.first);
+                    WalkJsonTree(item.second, callbacks);
+                }
+                callbacks->OnCloseMap();
+            }
+            return;
+        case JSON_ARRAY:
+            {
+                callbacks->OnOpenArray();
+                for (const auto& item : jsonValue.GetArray()) {
+                    WalkJsonTree(item, callbacks);
+                }
+                callbacks->OnCloseArray();
+            }
+            return;
+        case JSON_UNDEFINED:
+            ythrow yexception() << "cannot consume undefined json value";
+            return;
+    }
+    Y_UNREACHABLE();
+}
+
+static TNode CreateEmptyNodeByType(::NYson::EYsonType type)
+{
+    TNode result;
+    switch (type) {
+        case ::NYson::EYsonType::ListFragment:
+            result = TNode::CreateList();
+            break;
+        case ::NYson::EYsonType::MapFragment:
+            result = TNode::CreateMap();
+            break;
+        default:
+            break;
+    }
+    return result;
+}
+
+TNode NodeFromYsonString(const TStringBuf input, ::NYson::EYsonType type)
+{
+    TMemoryInput stream(input);
+    return NodeFromYsonStream(&stream, type);
+}
+
+TString NodeToYsonString(const TNode& node, NYson::EYsonFormat format)
+{
+    TStringStream stream;
+    NodeToYsonStream(node, &stream, format);
+    return stream.Str();
+}
+
+TString NodeToCanonicalYsonString(const TNode& node, NYson::EYsonFormat format)
+{
+    TStringStream stream;
+    NodeToCanonicalYsonStream(node, &stream, format);
+    return stream.Str();
+}
+
+TNode NodeFromYsonStream(IInputStream* input, ::NYson::EYsonType type)
+{
+    TNode result = CreateEmptyNodeByType(type);
+
+    TNodeBuilder builder(&result);
+    ::NYson::TYsonParser parser(&builder, input, type);
+    parser.Parse();
+    return result;
+}
+
+void NodeToYsonStream(const TNode& node, IOutputStream* output, NYson::EYsonFormat format)
+{
+    ::NYson::TYsonWriter writer(output, format);
+    TNodeVisitor visitor(&writer);
+    visitor.Visit(node);
+}
+
+void NodeToCanonicalYsonStream(const TNode& node, IOutputStream* output, NYson::EYsonFormat format)
+{
+    ::NYson::TYsonWriter writer(output, format);
+    TNodeVisitor visitor(&writer, /*sortMapKeys*/ true);
+    visitor.Visit(node);
+}
+
+TNode NodeFromJsonString(const TStringBuf input)
+{
+    TMemoryInput stream(input);
+
+    TNode result;
+
+    TNodeBuilder builder(&result);
+    TYson2JsonCallbacksAdapter callbacks(&builder, /*throwException*/ true);
+    NJson::TJsonReaderConfig config;
+    config.DontValidateUtf8 = true;
+    NJson::ReadJson(&stream, &config, &callbacks);
+    return result;
+}
+
+TNode NodeFromJsonValue(const NJson::TJsonValue& input)
+{
+    TNode result;
+    TNodeBuilder builder(&result);
+    TYson2JsonCallbacksAdapter callbacks(&builder, /*throwException*/ true);
+    WalkJsonTree(input, &callbacks);
+    return result;
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT