Support of PG extensions in YQL embedded (#10117)

Author: vitstn

ydb/library/yql/public/embedded/no_llvm/ya.make

@@ -49,6 +49,8 @@ PEERDIR(
     ydb/library/yql/providers/yt/provider
     ydb/library/yql/providers/yt/codec/codegen/no_llvm
     ydb/library/yql/providers/yt/comp_nodes/no_llvm
+    ydb/library/yql/parser/pg_wrapper/interface
+    ydb/library/yql/parser/pg_catalog
 )
 
 YQL_LAST_ABI_VERSION()

ydb/library/yql/public/embedded/ya.make

@@ -41,6 +41,8 @@ PEERDIR(
     ydb/library/yql/providers/yt/provider
     ydb/library/yql/providers/yt/codec/codegen
     ydb/library/yql/providers/yt/comp_nodes/llvm14
+    ydb/library/yql/parser/pg_wrapper/interface
+    ydb/library/yql/parser/pg_catalog
 )
 
 YQL_LAST_ABI_VERSION()

ydb/library/yql/public/embedded/yql_embedded.cpp

@@ -24,6 +24,8 @@
 #include <ydb/library/yql/core/services/mounts/yql_mounts.h>
 #include <ydb/library/yql/utils/log/log.h>
 #include <ydb/library/yql/utils/backtrace/backtrace.h>
+#include <ydb/library/yql/parser/pg_wrapper/interface/comp_factory.h>
+#include <ydb/library/yql/parser/pg_wrapper/interface/parser.h>
 
 #include <yt/cpp/mapreduce/interface/config.h>
 
@@ -319,6 +321,12 @@ namespace NYql {
                     NKikimr::NMiniKQL::FillStaticModules(*FuncRegistry_);
                 }
 
+                if (!Options_.PgExtensions_.empty()) {
+                    NPg::RegisterExtensions(Options_.PgExtensions_, false,
+                        *NSQLTranslationPG::CreateExtensionSqlParser(),
+                        NKikimr::NMiniKQL::CreateExtensionLoader().get());
+                }
+
                 TUserDataTable userDataTable = GetYqlModuleResolver(ExprContext_, ModuleResolver_, Options_.UserData_, Clusters_, {});
 
                 if (!userDataTable) {

ydb/library/yql/public/embedded/yql_embedded.h

@@ -2,6 +2,7 @@
 
 #include <ydb/library/yql/core/user_data/yql_user_data.h>
 #include <ydb/library/yql/core/file_storage/defs/downloader.h>
+#include <ydb/library/yql/parser/pg_catalog/catalog.h>
 
 #include <yt/cpp/mapreduce/interface/logging/logger.h>
 
@@ -60,6 +61,7 @@ namespace NYql {
             TString UdfResolverBinary_;
             TString UdfsDir_;
             bool PreloadUdfs_ = false; // used when UdfResolverBinary_ is specified, if UdfResolverBinary_ is empty it is considered equal to true
+            TVector<NPg::TExtensionDesc> PgExtensions_;
             TVector<NUserData::TUserData> UserData_;
 
             ELogPriority LogLevel_ = TLOG_ERR;