Exposed extensions via pg_extension table (#8062)

Author: vitstn

ydb/library/yql/core/pg_ext/yql_pg_ext.cpp

@@ -16,6 +16,7 @@ void PgExtensionsFromProto(const NYql::NProto::TPgExtensions& proto,
         desc.LibraryPath = e.GetLibraryPath();
         desc.TypesOnly = e.GetTypesOnly();
         desc.LibraryMD5 = e.GetLibraryMD5();
+        desc.Version = e.GetVersion();
         extensions.emplace_back(desc);
     }
 }

ydb/library/yql/parser/pg_catalog/catalog.h

@@ -395,6 +395,7 @@ struct TExtensionDesc {
     TString LibraryPath;        // file path
     bool TypesOnly = false;     // Can't be loaded if true
     TString LibraryMD5;         // optional
+    TString Version;            // version of extension
 };
 
 class IExtensionSqlBuilder {

ydb/library/yql/parser/pg_wrapper/comp_factory.cpp

@@ -542,6 +542,17 @@ class TPgTableContent : public TMutableComputationNode<TPgTableContent> {
                 };
 
                 ApplyFillers(AllPgLanguageFillers, Y_ARRAY_SIZE(AllPgLanguageFillers), PgLanguageFillers_);
+            } else if (Table_ == "pg_extension") {
+                static const std::pair<const char*, TPgExtensionFiller> AllPgExtensionFillers[] = {
+                    {"oid", [](ui32 oid,const NPg::TExtensionDesc& desc) { return ScalarDatumToPod(ObjectIdGetDatum(oid)); }},
+                    {"extname", [](ui32,const NPg::TExtensionDesc& desc) { return PointerDatumToPod((Datum)MakeFixedString(desc.Name, NAMEDATALEN)); }},
+                    {"extowner", [](ui32,const NPg::TExtensionDesc&) { return ScalarDatumToPod(ObjectIdGetDatum(1)); }},
+                    {"extnamespace", [](ui32,const NPg::TExtensionDesc&) { return ScalarDatumToPod(ObjectIdGetDatum(PG_CATALOG_NAMESPACE)); }},
+                    {"extversion", [](ui32,const NPg::TExtensionDesc& desc) { return PointerDatumToPod((Datum)MakeVar(desc.Version)); }},
+                    {"extrelocatable", [](ui32,const NPg::TExtensionDesc&) { return ScalarDatumToPod(BoolGetDatum(false)); }},
+                };
+
+                ApplyFillers(AllPgExtensionFillers, Y_ARRAY_SIZE(AllPgExtensionFillers), PgExtensionFillers_);
             }
         } else {
             if (Table_ == "tables") {
@@ -927,6 +938,19 @@ class TPgTableContent : public TMutableComputationNode<TPgTableContent> {
                         }
                     }
 
+                    sysFiller.Fill(items);
+                    rows.emplace_back(row);
+                });
+            } else if (Table_ == "pg_extension") {
+                NPg::EnumExtensions([&](ui32 oid, const NPg::TExtensionDesc& desc) {
+                    NUdf::TUnboxedValue* items;
+                    auto row = compCtx.HolderFactory.CreateDirectArrayHolder(PgExtensionFillers_.size(), items);
+                    for (ui32 i = 0; i < PgExtensionFillers_.size(); ++i) {
+                        if (PgExtensionFillers_[i]) {
+                            items[i] = PgExtensionFillers_[i](oid, desc);
+                        }
+                    }
+
                     sysFiller.Fill(items);
                     rows.emplace_back(row);
                 });
@@ -1026,6 +1050,9 @@ class TPgTableContent : public TMutableComputationNode<TPgTableContent> {
 
     using TPgOperFiller = NUdf::TUnboxedValuePod(*)(const NPg::TOperDesc&);
     TVector<TPgOperFiller> PgOperFillers_;
+
+    using TPgExtensionFiller = NUdf::TUnboxedValuePod(*)(ui32,const NPg::TExtensionDesc&);
+    TVector<TPgExtensionFiller> PgExtensionFillers_;
 };
 
 class TFunctionCallInfo {

ydb/library/yql/protos/pg_ext.proto

@@ -8,6 +8,7 @@ message TPgExtension {
     optional string LibraryPath = 4;
     optional bool TypesOnly = 5;
     optional string LibraryMD5 = 6;
+    optional string Version = 7;
 }
 
 message TPgExtensions {

ydb/library/yql/tools/yqlrun/pg_ext.txt

@@ -4,5 +4,6 @@ Extension {
     SqlPath: "../../../../../yql/pg_ext/postgis/libs/postgis/postgis/postgis.sql"
     SqlPath: "../../../../../yql/pg_ext/postgis/libs/postgis/spatial_ref_sys.sql"
     LibraryPath: "../../../../../yql/pg_ext/postgis/libs/postgis/libpostgis.so"
+    Version: "3.4.3dev"
 }