library/yson: Add NodeFromJsonStringIterative

AI для

1 files changed, 11 insertions, 1 deletions

diff --git a/library/cpp/yson/json/yson2json_adapter.cpp b/library/cpp/yson/json/yson2json_adapter.cpp
index b5e7c49d4d..1dbe770e5b 100644
--- a/library/cpp/yson/json/yson2json_adapter.cpp
+++ b/library/cpp/yson/json/yson2json_adapter.cpp
@@ -1,9 +1,13 @@
 #include "yson2json_adapter.h"
 
 namespace NYT {
-    TYson2JsonCallbacksAdapter::TYson2JsonCallbacksAdapter(::NYson::TYsonConsumerBase* impl, bool throwException)
+    TYson2JsonCallbacksAdapter::TYson2JsonCallbacksAdapter(
+        ::NYson::TYsonConsumerBase* impl,
+        bool throwException,
+        ui64 maxDepth)
         : NJson::TJsonCallbacks(throwException)
         , Impl_(impl)
+        , MaxDepth_(maxDepth)
     {
     }
 
@@ -46,6 +50,9 @@ namespace NYT {
     bool TYson2JsonCallbacksAdapter::OnOpenArray() {
         WrapIfListItem();
         State_.ContextStack.push(true);
+        if (State_.ContextStack.size() > MaxDepth_) {
+            return false;
+        }
         Impl_->OnBeginList();
         return true;
     }
@@ -59,6 +66,9 @@ namespace NYT {
     bool TYson2JsonCallbacksAdapter::OnOpenMap() {
         WrapIfListItem();
         State_.ContextStack.push(false);
+        if (State_.ContextStack.size() > MaxDepth_) {
+            return false;
+        }
         Impl_->OnBeginMap();
         return true;
     }