Make possible to add UserDataTable in TProgram

2 files changed, 11 insertions, 0 deletions

diff --git a/ydb/library/yql/core/facade/yql_facade.cpp b/ydb/library/yql/core/facade/yql_facade.cpp
index 7872f74a29d..6019a537d75 100644
--- a/ydb/library/yql/core/facade/yql_facade.cpp
+++ b/ydb/library/yql/core/facade/yql_facade.cpp
@@ -410,6 +410,15 @@ TString TProgram::TakeSessionId() {
     }
 }
 
+void TProgram::AddUserDataTable(const TUserDataTable& userDataTable) {
+    for (const auto& p : userDataTable) {
+        if (!SavedUserDataTable_.emplace(p).second) {
+            ythrow yexception() << "UserDataTable already has user data block with key " << p.first;
+        }
+        UserDataStorage_->AddUserDataBlock(p.first, p.second);
+    }
+}
+
 bool TProgram::ParseYql() {
     YQL_PROFILE_FUNC(TRACE);
     YQL_ENSURE(SourceSyntax_ == ESourceSyntax::Unknown);
diff --git a/ydb/library/yql/core/facade/yql_facade.h b/ydb/library/yql/core/facade/yql_facade.h
index 149c49c1b66..26529ed3883 100644
--- a/ydb/library/yql/core/facade/yql_facade.h
+++ b/ydb/library/yql/core/facade/yql_facade.h
@@ -102,6 +102,8 @@ public:
 public:
     ~TProgram();
 
+    void AddUserDataTable(const TUserDataTable& userDataTable);
+
     bool ParseYql();
     bool ParseSql();
     bool ParseSql(const NSQLTranslation::TTranslationSettings& settings);