Fixes

shnikd · Oct 3, 2024 · 4be3433 · 4be3433
1 parent 058b5e0
commit 4be3433
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/ydb/library/yql/core/pg_settings/guc_settings.cpp b/ydb/library/yql/core/pg_settings/guc_settings.cpp
@@ -50,6 +50,31 @@ void TGUCSettings::ExportToJson(NJson::TJsonValue& value) const {
     value.InsertValue("guc_settings", std::move(gucSettings));
 }
 
+void TGUCSettings::ImportFromJson(const NJson::TJsonValue& value)
+{
+    Settings_.clear();
+    RollbackSettings_.clear();
+    SessionSettings_.clear();
+    if (value.Has("guc_settings")) {
+        auto gucSettings = value["guc_settings"];
+        if (gucSettings.Has("settings")) {
+            for (const auto& [settingName, settingValue] : gucSettings["settings"].GetMapSafe()) {
+                Settings_[settingName] = settingValue.GetStringSafe();
+            }
+        }
+        if (gucSettings.Has("rollback_settings")) {
+            for (const auto& [settingName, settingValue] : gucSettings["rollback_settings"].GetMapSafe()) {
+                RollbackSettings_[settingName] = settingValue.GetStringSafe();
+            }
+        }
+        if (gucSettings.Has("session_settings")) {
+            for (const auto& [settingName, settingValue] : gucSettings["session_settings"].GetMapSafe()) {
+                SessionSettings_[settingName] = settingValue.GetStringSafe();
+            }
+        }
+    }
+}
+
 TString TGUCSettings::SerializeToString() const {
     NJson::TJsonValue gucJson;
     this->ExportToJson(gucJson);

diff --git a/ydb/library/yql/core/pg_settings/guc_settings.h b/ydb/library/yql/core/pg_settings/guc_settings.h
@@ -19,6 +19,7 @@ struct TGUCSettings {
     void Commit();
     void RollBack();
     void ExportToJson(NJson::TJsonValue& value) const;
+    void ImportFromJson(const NJson::TJsonValue& value);
     TString SerializeToString() const;
 
     size_t GetHash() const noexcept;