Fixed i/o for pg_proc (#10914)

ydb-platform · Oct 25, 2024 · 7108e37 · 7108e37
1 parent bb8be1a
commit 7108e37
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/ydb/library/yql/parser/pg_catalog/catalog.cpp b/ydb/library/yql/parser/pg_catalog/catalog.cpp
@@ -3835,6 +3835,9 @@ TString ExportExtensions(const TMaybe<TSet<ui32>>& filter) {
 
         protoProc->SetIsStrict(desc.IsStrict);
         protoProc->SetLang(desc.Lang);
+        protoProc->SetResultType(desc.ResultType);
+        protoProc->SetReturnSet(desc.ReturnSet);
+        protoProc->SetKind((ui32)desc.Kind);
     }
 
     TVector<TTableInfoKey> extTables;
@@ -4104,6 +4107,9 @@ void ImportExtensions(const TString& exported, bool typesOnly, IExtensionLoader*
             desc.Src = protoProc.GetSrc();
             desc.IsStrict = protoProc.GetIsStrict();
             desc.Lang = protoProc.GetLang();
+            desc.ResultType = protoProc.GetResultType();
+            desc.ReturnSet = protoProc.GetReturnSet();
+            desc.Kind = (EProcKind)protoProc.GetKind();
             for (const auto t : protoProc.GetArgType()) {
                 desc.ArgTypes.push_back(t);
             }

diff --git a/ydb/library/yql/parser/pg_catalog/proto/pg_catalog.proto b/ydb/library/yql/parser/pg_catalog/proto/pg_catalog.proto
@@ -41,6 +41,9 @@ message TPgProc {
     optional string VariadicArgName = 13;
     optional bool IsStrict = 14;
     optional uint32 Lang = 15;
+    optional uint32 ResultType = 16;
+    optional bool ReturnSet = 17;
+    optional uint32 Kind = 18;
 }
 
 message TPgTable {