From 91d522066eb5a7759a5c7804837de08449ba1704 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=90=D0=BB=D0=B5=D0=BA=D1=81=D0=B0=D0=BD=D0=B4=D1=80=20?=
 =?UTF-8?q?=D0=9D=D0=BE=D0=B2=D0=BE=D0=B6=D0=B8=D0=BB=D0=BE=D0=B2?=
 <mrlolthe1st@ydb.tech>
Date: Wed, 27 Dec 2023 09:50:23 +0300
Subject: [PATCH] Add PG provider to pgrun (#732)

---
 ydb/library/yql/tools/pgrun/pgrun.cpp | 3 +++
 ydb/library/yql/tools/pgrun/ya.make   | 1 +
 2 files changed, 4 insertions(+)

diff --git a/ydb/library/yql/tools/pgrun/pgrun.cpp b/ydb/library/yql/tools/pgrun/pgrun.cpp
index a26b41ddacfe..1ad3d59d8aed 100644
--- a/ydb/library/yql/tools/pgrun/pgrun.cpp
+++ b/ydb/library/yql/tools/pgrun/pgrun.cpp
@@ -9,6 +9,7 @@
 #include <ydb/library/yql/providers/common/provider/yql_provider_names.h>
 #include <ydb/library/yql/providers/common/proto/gateways_config.pb.h>
 #include <ydb/library/yql/providers/yt/provider/yql_yt_provider.h>
+#include <ydb/library/yql/providers/pg/provider/yql_pg_provider.h>
 #include <ydb/library/yql/public/issue/yql_issue.h>
 
 #include <library/cpp/getopt/last_getopt.h>
@@ -971,6 +972,7 @@ int Main(int argc, char* argv[])
 
     static const TString DefaultCluster{"plato"};
     clusterMapping[DefaultCluster] = YtProviderName;
+    clusterMapping["pg_catalog"] = PgProviderName;
 
     opts.AddHelpOption();
     opts.AddLongOption("datadir", "directory for tables").StoreResult<TString>(&rawDataDir);
@@ -1001,6 +1003,7 @@ int Main(int argc, char* argv[])
 
     TVector<TDataProviderInitializer> dataProvidersInit;
     dataProvidersInit.push_back(GetYtNativeDataProviderInitializer(ytNativeGateway));
+    dataProvidersInit.push_back(GetPgDataProviderInitializer());
 
     TExprContext ctx;
     TExprContext::TFreezeGuard freezeGuard(ctx);
diff --git a/ydb/library/yql/tools/pgrun/ya.make b/ydb/library/yql/tools/pgrun/ya.make
index 8420b3a60235..43fb6932b39c 100644
--- a/ydb/library/yql/tools/pgrun/ya.make
+++ b/ydb/library/yql/tools/pgrun/ya.make
@@ -32,6 +32,7 @@ PEERDIR(
     ydb/library/yql/providers/yt/gateway/file
     ydb/library/yql/providers/yt/comp_nodes/llvm
     ydb/library/yql/providers/yt/codec/codegen
+    ydb/library/yql/providers/pg/provider
     ydb/library/yql/core/url_preprocessing
     ydb/library/yql/parser/pg_wrapper
 )