Pretty printing for TNode

1 files changed, 14 insertions, 0 deletions

diff --git a/library/cpp/yson/node/node.cpp b/library/cpp/yson/node/node.cpp
index b39e070718..3733c3cae2 100644
--- a/library/cpp/yson/node/node.cpp
+++ b/library/cpp/yson/node/node.cpp
@@ -5,6 +5,9 @@
 #include <library/cpp/yson/writer.h>
 
 #include <util/generic/overloaded.h>
+#include <util/string/escape.h>
+
+#include <iostream>
 
 namespace NYT {
 
@@ -910,6 +913,17 @@ bool GetBool(const TNode& node)
     }
 }
 
+void PrintTo(const TNode& node, std::ostream* out)
+{
+    if (node.IsUndefined()) {
+        (*out) << "NYT::TNode::Undefined";
+    } else {
+        (*out) << "NYT::TNode("
+        << NodeToCanonicalYsonString(node)
+        << ")";
+    }
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 
 } // namespace NYT