diff options
author | udovichenko-r <udovichenko-r@yandex-team.com> | 2024-12-13 14:38:21 +0300 |
---|---|---|
committer | udovichenko-r <udovichenko-r@yandex-team.com> | 2024-12-13 15:57:44 +0300 |
commit | 1402a032649d20ffb80a541f706afcd31cd897fa (patch) | |
tree | 2c7d158a5ab885a6301dd0da832b16fd5f39c711 /yql/essentials | |
parent | 933c5664c69e565f3ad5752f5862e0ba860a52fe (diff) | |
download | ydb-1402a032649d20ffb80a541f706afcd31cd897fa.tar.gz |
Facade runner lib
commit_hash:06b99da07b603d38b18615794ce645139fcec9d0
Diffstat (limited to 'yql/essentials')
-rw-r--r-- | yql/essentials/core/facade/yql_facade.cpp | 4 | ||||
-rw-r--r-- | yql/essentials/tests/common/test_framework/yqlrun.py | 2 | ||||
-rw-r--r-- | yql/essentials/tools/arrow_kernels_dump/arrow_kernels_dump.cpp | 24 | ||||
-rw-r--r-- | yql/essentials/tools/arrow_kernels_dump/ya.make | 23 | ||||
-rw-r--r-- | yql/essentials/tools/ya.make | 4 | ||||
-rw-r--r-- | yql/essentials/tools/yql_facade_run/ya.make | 50 | ||||
-rw-r--r-- | yql/essentials/tools/yql_facade_run/yql_facade_run.cpp | 779 | ||||
-rw-r--r-- | yql/essentials/tools/yql_facade_run/yql_facade_run.h | 213 |
8 files changed, 1095 insertions, 4 deletions
diff --git a/yql/essentials/core/facade/yql_facade.cpp b/yql/essentials/core/facade/yql_facade.cpp index ee8e31eb35..a0b07e469f 100644 --- a/yql/essentials/core/facade/yql_facade.cpp +++ b/yql/essentials/core/facade/yql_facade.cpp @@ -1598,7 +1598,7 @@ TMaybe<TString> TProgram::GetStatistics(bool totalOnly, THashMap<TString, TStrin } TStringStream out; - NYson::TYsonWriter writer(&out); + NYson::TYsonWriter writer(&out, OutputFormat_); // Header writer.OnBeginMap(); writer.OnKeyedItem("ExecutionStatistics"); @@ -1677,7 +1677,7 @@ TMaybe<TString> TProgram::GetDiscoveredData() { } TStringStream out; - NYson::TYsonWriter writer(&out); + NYson::TYsonWriter writer(&out, OutputFormat_); writer.OnBeginMap(); for (auto& datasource: TypeCtx_->DataSources) { TStringStream providerOut; diff --git a/yql/essentials/tests/common/test_framework/yqlrun.py b/yql/essentials/tests/common/test_framework/yqlrun.py index 95aa794ec7..e23b81c92f 100644 --- a/yql/essentials/tests/common/test_framework/yqlrun.py +++ b/yql/essentials/tests/common/test_framework/yqlrun.py @@ -234,7 +234,7 @@ class YQLRun(object): for name in self.tables: cmd += '--table=yt.%s@%s ' % (name, self.tables[name].yqlrun_file) - if "--lineage" not in self.extra_args: + if "--lineage" not in self.extra_args and "--peephole" not in self.extra_args: if optimize_only: cmd += '-O ' else: diff --git a/yql/essentials/tools/arrow_kernels_dump/arrow_kernels_dump.cpp b/yql/essentials/tools/arrow_kernels_dump/arrow_kernels_dump.cpp new file mode 100644 index 0000000000..9c7778bfa7 --- /dev/null +++ b/yql/essentials/tools/arrow_kernels_dump/arrow_kernels_dump.cpp @@ -0,0 +1,24 @@ +#include <yql/essentials/minikql/invoke_builtins/mkql_builtins.h> +#include <yql/essentials/public/udf/udf_version.h> + +#include <util/stream/output.h> +#include <util/generic/algorithm.h> +#include <util/folder/path.h> + +int main(int argc, char **argv) { + Y_UNUSED(argc); + Cerr << TFsPath(argv[0]).GetName() << " ABI version: " << NKikimr::NUdf::CurrentAbiVersionStr() << Endl; + + auto builtins = NKikimr::NMiniKQL::CreateBuiltinRegistry(); + auto families = builtins->GetAllKernelFamilies(); + Sort(families, [](const auto& x, const auto& y) { return x.first < y.first; }); + ui64 totalKernels = 0; + for (const auto& f : families) { + auto numKernels = f.second->GetAllKernels().size(); + Cout << f.first << ": " << numKernels << " kernels" << Endl; + totalKernels += numKernels; + } + + Cout << "Total kernel families: " << families.size() << ", kernels: " << totalKernels << Endl; + return 0; +} diff --git a/yql/essentials/tools/arrow_kernels_dump/ya.make b/yql/essentials/tools/arrow_kernels_dump/ya.make new file mode 100644 index 0000000000..d33cfbd5b9 --- /dev/null +++ b/yql/essentials/tools/arrow_kernels_dump/ya.make @@ -0,0 +1,23 @@ +PROGRAM() + +SRCS( + arrow_kernels_dump.cpp +) + +IF (OS_LINUX) + # prevent external python extensions to lookup protobuf symbols (and maybe + # other common stuff) in main binary + EXPORTS_SCRIPT(${ARCADIA_ROOT}/yql/essentials/tools/exports.symlist) +ENDIF() + +PEERDIR( + yql/essentials/minikql/invoke_builtins + yql/essentials/minikql/invoke_builtins/llvm14 + yql/essentials/public/udf/service/terminate_policy + yql/essentials/public/udf + yql/essentials/parser/pg_wrapper +) + +YQL_LAST_ABI_VERSION() + +END() diff --git a/yql/essentials/tools/ya.make b/yql/essentials/tools/ya.make index 89ea1d9d22..8d74153182 100644 --- a/yql/essentials/tools/ya.make +++ b/yql/essentials/tools/ya.make @@ -1,5 +1,6 @@ RECURSE( - astdiff + arrow_kernels_dump + astdiff pg_catalog_dump pg-make-test pgrun @@ -9,4 +10,5 @@ RECURSE( udf_dep_stub udf_probe udf_resolver + yql_facade_run ) diff --git a/yql/essentials/tools/yql_facade_run/ya.make b/yql/essentials/tools/yql_facade_run/ya.make new file mode 100644 index 0000000000..e9d93e235d --- /dev/null +++ b/yql/essentials/tools/yql_facade_run/ya.make @@ -0,0 +1,50 @@ +LIBRARY() + +SRCS( + yql_facade_run.cpp +) + +PEERDIR( + yql/essentials/providers/pg/provider + yql/essentials/providers/common/provider + yql/essentials/providers/common/proto + yql/essentials/providers/common/udf_resolve + yql/essentials/core/file_storage + yql/essentials/core/file_storage/proto + yql/essentials/core/file_storage/defs + yql/essentials/core/url_lister/interface + yql/essentials/core/services/mounts + yql/essentials/core/services + yql/essentials/core/credentials + yql/essentials/core/pg_ext + yql/essentials/core/facade + yql/essentials/core/url_lister + yql/essentials/core/url_preprocessing + yql/essentials/core/peephole_opt + yql/essentials/core + yql/essentials/minikql/invoke_builtins + yql/essentials/minikql + yql/essentials/ast + yql/essentials/parser/pg_wrapper/interface + yql/essentials/parser/pg_catalog + yql/essentials/public/udf + yql/essentials/public/result_format + yql/essentials/utils/failure_injector + yql/essentials/utils/backtrace + yql/essentials/utils/log + yql/essentials/protos + yql/essentials/sql/settings + yql/essentials/sql/v1/format + + library/cpp/resource + library/cpp/getopt + library/cpp/yson/node + library/cpp/yson + library/cpp/logger + + contrib/libs/protobuf +) + +YQL_LAST_ABI_VERSION() + +END() diff --git a/yql/essentials/tools/yql_facade_run/yql_facade_run.cpp b/yql/essentials/tools/yql_facade_run/yql_facade_run.cpp new file mode 100644 index 0000000000..91a3a3eaf8 --- /dev/null +++ b/yql/essentials/tools/yql_facade_run/yql_facade_run.cpp @@ -0,0 +1,779 @@ +#include "yql_facade_run.h" + +#include <yql/essentials/providers/pg/provider/yql_pg_provider.h> +#include <yql/essentials/providers/common/provider/yql_provider_names.h> +#include <yql/essentials/providers/common/proto/gateways_config.pb.h> +#include <yql/essentials/providers/common/udf_resolve/yql_outproc_udf_resolver.h> +#include <yql/essentials/providers/common/udf_resolve/yql_simple_udf_resolver.h> +#include <yql/essentials/providers/common/udf_resolve/yql_udf_resolver_with_index.h> +#include <yql/essentials/core/yql_user_data_storage.h> +#include <yql/essentials/core/yql_udf_resolver.h> +#include <yql/essentials/core/yql_udf_index.h> +#include <yql/essentials/core/yql_udf_index_package_set.h> +#include <yql/essentials/core/yql_library_compiler.h> +#include <yql/essentials/core/yql_type_annotation.h> +#include <yql/essentials/core/pg_ext/yql_pg_ext.h> +#include <yql/essentials/core/services/mounts/yql_mounts.h> +#include <yql/essentials/core/services/yql_out_transformers.h> +#include <yql/essentials/core/file_storage/file_storage.h> +#include <yql/essentials/core/file_storage/proto/file_storage.pb.h> +#include <yql/essentials/core/peephole_opt/yql_opt_peephole_physical.h> +#include <yql/essentials/core/facade/yql_facade.h> +#include <yql/essentials/core/url_lister/url_lister_manager.h> +#include <yql/essentials/core/url_preprocessing/url_preprocessing.h> +#include <yql/essentials/minikql/invoke_builtins/mkql_builtins.h> +#include <yql/essentials/minikql/mkql_function_registry.h> +#include <yql/essentials/ast/yql_expr.h> +#include <yql/essentials/parser/pg_wrapper/interface/parser.h> +#include <yql/essentials/parser/pg_catalog/catalog.h> +#include <yql/essentials/public/udf/udf_version.h> +#include <yql/essentials/public/udf/udf_registrator.h> +#include <yql/essentials/public/udf/udf_validate.h> +#include <yql/essentials/public/result_format/yql_result_format_response.h> +#include <yql/essentials/public/result_format/yql_result_format_type.h> +#include <yql/essentials/public/result_format/yql_result_format_data.h> +#include <yql/essentials/utils/failure_injector/failure_injector.h> +#include <yql/essentials/utils/backtrace/backtrace.h> +#include <yql/essentials/utils/log/log.h> +#include <yql/essentials/protos/yql_mount.pb.h> +#include <yql/essentials/protos/pg_ext.pb.h> +#include <yql/essentials/sql/settings/translation_settings.h> +#include <yql/essentials/sql/v1/format/sql_format.h> + +#include <library/cpp/resource/resource.h> +#include <library/cpp/yson/node/node_io.h> +#include <library/cpp/yson/writer.h> + +#include <google/protobuf/text_format.h> +#include <google/protobuf/arena.h> + +#include <util/stream/output.h> +#include <util/stream/file.h> +#include <util/stream/null.h> +#include <util/system/user.h> +#include <util/string/split.h> +#include <util/string/join.h> +#include <util/string/builder.h> +#include <util/generic/vector.h> +#include <util/generic/ptr.h> +#include <util/generic/yexception.h> +#include <util/datetime/base.h> + +#ifdef __unix__ +#include <sys/resource.h> +#endif + +namespace { + +const ui32 PRETTY_FLAGS = NYql::TAstPrintFlags::PerLine | NYql::TAstPrintFlags::ShortQuote | + NYql::TAstPrintFlags::AdaptArbitraryContent; + +template <typename TMessage> +THolder<TMessage> ParseProtoConfig(const TString& cfgFile) { + auto config = MakeHolder<TMessage>(); + TString configData = TFileInput(cfgFile).ReadAll(); + + using ::google::protobuf::TextFormat; + if (!TextFormat::ParseFromString(configData, config.Get())) { + throw yexception() << "Bad format of config file " << cfgFile; + } + + return config; +} + +template <typename TMessage> +THolder<TMessage> ParseProtoFromResource(TStringBuf resourceName) { + if (!NResource::Has(resourceName)) { + return {}; + } + auto config = MakeHolder<TMessage>(); + TString configData = NResource::Find(resourceName); + + using ::google::protobuf::TextFormat; + if (!TextFormat::ParseFromString(configData, config.Get())) { + throw yexception() << "Bad format of config " << resourceName; + } + return config; +} + +class TOptPipelineConfigurator : public NYql::IPipelineConfigurator { +public: + TOptPipelineConfigurator(NYql::TProgramPtr prg, IOutputStream* planStream, IOutputStream* exprStream, bool withTypes) + : Program_(std::move(prg)) + , PlanStream_(planStream) + , ExprStream_(exprStream) + , WithTypes_(withTypes) + { + } + + void AfterCreate(NYql::TTransformationPipeline* pipeline) const final { + Y_UNUSED(pipeline); + } + + void AfterTypeAnnotation(NYql::TTransformationPipeline* pipeline) const final { + pipeline->Add(NYql::TExprLogTransformer::Sync("OptimizedExpr", NYql::NLog::EComponent::Core, NYql::NLog::ELevel::TRACE), + "OptTrace", NYql::TIssuesIds::CORE, "OptTrace"); + } + + void AfterOptimize(NYql::TTransformationPipeline* pipeline) const final { + if (ExprStream_) { + pipeline->Add(NYql::TExprOutputTransformer::Sync(Program_->ExprRoot(), ExprStream_, WithTypes_), "AstOutput"); + } + if (PlanStream_) { + pipeline->Add(NYql::TPlanOutputTransformer::Sync(PlanStream_, Program_->GetPlanBuilder(), Program_->GetOutputFormat()), "PlanOutput"); + } + } +private: + NYql::TProgramPtr Program_; + IOutputStream* PlanStream_; + IOutputStream* ExprStream_; + bool WithTypes_; +}; + +class TPeepHolePipelineConfigurator : public NYql::IPipelineConfigurator { +public: + TPeepHolePipelineConfigurator() { + } + + void AfterCreate(NYql::TTransformationPipeline* pipeline) const final { + Y_UNUSED(pipeline); + } + + void AfterTypeAnnotation(NYql::TTransformationPipeline* pipeline) const final { + pipeline->Add(NYql::TExprLogTransformer::Sync("OptimizedExpr", NYql::NLog::EComponent::Core, NYql::NLog::ELevel::TRACE), + "OptTrace", NYql::TIssuesIds::CORE, "OptTrace"); + } + + void AfterOptimize(NYql::TTransformationPipeline* pipeline) const final { + pipeline->Add(NYql::MakePeepholeOptimization(pipeline->GetTypeAnnotationContext()), "PeepHole"); + } +}; + +} // unnamed + + +namespace NYql { + +TFacadeRunOptions::TFacadeRunOptions() { + User = GetUsername(); +} + +TFacadeRunOptions::~TFacadeRunOptions() { +} + +void TFacadeRunOptions::InitLogger() { + if (Verbosity != LOG_DEF_PRIORITY) { + NYql::NLog::ELevel level = NYql::NLog::ELevelHelpers::FromInt(Verbosity); + NYql::NLog::EComponentHelpers::ForEach([level](NYql::NLog::EComponent c) { + NYql::NLog::YqlLogger().SetComponentLevel(c, level); + }); + } + + if (TraceOptStream) { + NYql::NLog::YqlLogger().SetComponentLevel(NYql::NLog::EComponent::Core, NYql::NLog::ELevel::TRACE); + NYql::NLog::YqlLogger().SetComponentLevel(NYql::NLog::EComponent::CoreEval, NYql::NLog::ELevel::TRACE); + NYql::NLog::YqlLogger().SetComponentLevel(NYql::NLog::EComponent::CorePeepHole, NYql::NLog::ELevel::TRACE); + } else if (ShowLog) { + NYql::NLog::YqlLogger().SetComponentLevel(NYql::NLog::EComponent::Core, NYql::NLog::ELevel::DEBUG); + } +} + +void TFacadeRunOptions::PrintInfo(const TString& msg) { + if (!NoDebug && Verbosity >= TLOG_INFO) { + Cerr << msg << Endl; + } +} + +void TFacadeRunOptions::Parse(int argc, const char *argv[]) { + + NLastGetopt::TOpts opts = NLastGetopt::TOpts::Default(); + + opts.AddHelpOption(); + opts.AddLongOption('p', "program", "Program file").Required().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + ProgramFile = file; + if (ProgramFile == "-") { + ProgramFile = "-stdin-"; + ProgramText = Cin.ReadAll(); + } else { + ProgramText = TFileInput(ProgramFile).ReadAll(); + } + User = GetUsername(); + }); + opts.AddLongOption('s', "sql", "Program is SQL query").NoArgument().StoreValue(&ProgramType, EProgramType::Sql); + if (PgSupport) { + opts.AddLongOption("pg", "Program has PG syntax").NoArgument().StoreValue(&ProgramType, EProgramType::Pg); + opts.AddLongOption("pg-ext", "pg extensions config file").Optional().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + PgExtConfig = ParseProtoConfig<NProto::TPgExtensions>(file); + }); + } + opts.AddLongOption('f', "file", "Additional files").RequiredArgument("name@path") + .KVHandler([this](TString name, TString path) { + if (name.empty() || path.empty()) { + throw yexception() << "Incorrect file mapping, expected form name@path, e.g. MyFile@file.txt"; + } + + auto& entry = DataTable[NYql::TUserDataKey::File(NYql::GetDefaultFilePrefix() + name)]; + entry.Type = NYql::EUserDataType::PATH; + entry.Data = path; + }, '@'); + + opts.AddLongOption('U', "url", "Additional urls").RequiredArgument("name@path") + .KVHandler([this](TString name, TString url) { + if (name.empty() || url.empty()) { + throw yexception() << "url mapping, expected form name@url, e.g. MyUrl@http://example.com/file"; + } + + auto& entry = DataTable[NYql::TUserDataKey::File(NYql::GetDefaultFilePrefix() + name)]; + entry.Type = NYql::EUserDataType::URL; + entry.Data = url; + }, '@'); + + opts.AddLongOption('m', "mounts", "Mount points config file.").Optional().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + MountConfig = ParseProtoConfig<NYqlMountConfig::TMountConfig>(file); + }); + opts.AddLongOption("params-file", "Query parameters values in YSON format").Optional().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + Params = TFileInput(file).ReadAll(); + }); + opts.AddLongOption('G', "gateways", "Used gateways").DefaultValue(JoinSeq(",", SupportedGateways_)) + .Handler1T<TString>([this](const TString& gateways) { + ::StringSplitter(gateways).Split(',').Consume([&](const TStringBuf& val) { + if (!SupportedGateways_.contains(val)) { + throw yexception() << "Unsupported gateway \"" << val << '"'; + } + GatewayTypes.emplace(val); + }); + }); + opts.AddLongOption("gateways-cfg", "Gateways configuration file").Optional().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + GatewaysConfig = ParseProtoConfig<TGatewaysConfig>(file); + }); + opts.AddLongOption("fs-cfg", "Fs configuration file").Optional().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + FsConfig = MakeHolder<TFileStorageConfig>(); + LoadFsConfigFromFile(file, *FsConfig); + }); + opts.AddLongOption('u', "udf", "Load shared library with UDF by given path").AppendTo(&UdfsPaths); + opts.AddLongOption("udfs-dir", "Load all shared libraries with UDFs found in given directory") + .Handler1T<TString>([this](const TString& dir) { + NKikimr::NMiniKQL::FindUdfsInDir(dir, &UdfsPaths); + }); + opts.AddLongOption("udf-resolver", "Path to udf-resolver").Optional().RequiredArgument("PATH").StoreResult(&UdfResolverPath); + opts.AddLongOption("udf-resolver-filter-syscalls", "Filter syscalls in udf resolver").Optional().NoArgument().SetFlag(&UdfResolverFilterSyscalls); + opts.AddLongOption("scan-udfs", "Scan specified udfs with external udf-resolver to use static function registry").NoArgument().SetFlag(&ScanUdfs); + + opts.AddLongOption("parse-only", "Exit after program has been parsed").NoArgument().StoreValue(&Mode, ERunMode::Parse); + opts.AddLongOption("compile-only", "Exit after program has been compiled").NoArgument().StoreValue(&Mode, ERunMode::Compile); + opts.AddLongOption("validate", "Exit after program has been validated").NoArgument().StoreValue(&Mode, ERunMode::Validate); + opts.AddLongOption("lineage", "Exit after data lineage has been calculated").NoArgument().StoreValue(&Mode, ERunMode::Lineage); + opts.AddLongOption('O',"optimize", "Optimize expression").NoArgument().StoreValue(&Mode, ERunMode::Optimize); + opts.AddLongOption('R',"run", "Run expression using input/output tables").NoArgument().StoreValue(&Mode, ERunMode::Run); + opts.AddLongOption('D', "discover", "Discover tables in the program").NoArgument().StoreValue(&Mode, ERunMode::Discover); + opts.AddLongOption("peephole", "Perform peephole stage of expression using input/output tables").NoArgument().StoreValue(&Mode, ERunMode::Peephole); + + opts.AddLongOption('L', "show-log", "Show transformation log").Optional().NoArgument().SetFlag(&ShowLog); + opts.AddLongOption('v', "verbosity", "Log verbosity level").Optional().RequiredArgument("LEVEL").StoreResult(&Verbosity); + opts.AddLongOption("print-ast", "Print AST after loading").NoArgument().SetFlag(&PrintAst); + opts.AddLongOption("print-expr", "Print rebuild AST before execution").NoArgument() + .Handler0([this]() { + if (!ExprStream) { + ExprStream = &Cout; + } + }); + opts.AddLongOption("with-types", "Print types annotation").NoArgument().SetFlag(&WithTypes); + opts.AddLongOption("trace-opt", "Print AST in the begin of each transformation").NoArgument() + .Handler0([this]() { + TraceOptStream = &Cerr; + }); + opts.AddLongOption("expr-file", "Print AST to that file instead of stdout").Optional().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + ExprStreamHolder = MakeHolder<TFixedBufferFileOutput>(file); + ExprStream = ExprStreamHolder.Get(); + }); + opts.AddLongOption("print-result", "Print program execution result to stdout").NoArgument() + .Handler0([this]() { + if (!ResultStream) { + ResultStream = &Cout; + } + }); + opts.AddLongOption("format", "Results format") + .Optional() + .RequiredArgument("STR") + .Choices(THashSet<TString>{"text", "binary", "pretty"}) + .Handler1T<TString>([this](const TString& val) { + if (val == "text") { + ResultsFormat = NYson::EYsonFormat::Text; + } else if (val == "binary") { + ResultsFormat = NYson::EYsonFormat::Binary; + } else if (val == "pretty") { + ResultsFormat = NYson::EYsonFormat::Pretty; + } else { + throw yexception() << "Unknown result format " << val; + } + }); + + opts.AddLongOption("result-file", "Print program execution result to file").Optional().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + ResultStreamHolder = MakeHolder<TFixedBufferFileOutput>(file); + ResultStream = ResultStreamHolder.Get(); + }); + opts.AddLongOption('P',"trace-plan", "Print plan before execution").NoArgument() + .Handler0([this]() { + if (!PlanStream) { + PlanStream = &Cerr; + } + }); + opts.AddLongOption("plan-file", "Print program plan to file").Optional().RequiredArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + PlanStreamHolder = MakeHolder<TFixedBufferFileOutput>(file); + PlanStream = PlanStreamHolder.Get(); + }); + opts.AddLongOption("err-file", "Print validate/optimize/runtime errors to file") + .Handler1T<TString>([this](const TString& file) { + ErrStreamHolder = MakeHolder<TFixedBufferFileOutput>(file); + ErrStream = ErrStreamHolder.Get(); + }); + opts.AddLongOption("full-expr", "Avoid buffering of expr/plan").NoArgument().SetFlag(&FullExpr); + opts.AddLongOption("mem-limit", "Set memory limit in megabytes") + .Handler1T<ui32>(0, [](ui32 memLimit) { + if (memLimit) { +#ifdef __unix__ + auto memLimitBytes = memLimit * 1024 * 1024; + + struct rlimit rl; + if (getrlimit(RLIMIT_AS, &rl)) { + throw TSystemError() << "Cannot getrlimit(RLIMIT_AS)"; + } + + rl.rlim_cur = memLimitBytes; + if (setrlimit(RLIMIT_AS, &rl)) { + throw TSystemError() << "Cannot setrlimit(RLIMIT_AS) to " << memLimitBytes << " bytes"; + } +#else + throw yexception() << "Memory limit can not be set on this platfrom"; +#endif + } + }); + + opts.AddLongOption("validate-mode", "Validate udf mode, available values: " + NUdf::ValidateModeAvailables()) + .DefaultValue(NUdf::ValidateModeAsStr(NUdf::EValidateMode::Greedy)) + .Handler1T<TString>([this](const TString& mode) { + ValidateMode = NUdf::ValidateModeByStr(mode); + }); + opts.AddLongOption("stat", "Print execution statistics").Optional().OptionalArgument("FILE") + .Handler1T<TString>([this](const TString& file) { + if (file) { + StatStreamHolder = MakeHolder<TFileOutput>(file); + StatStream = StatStreamHolder.Get(); + } else { + StatStream = &Cerr; + } + }); + opts.AddLongOption("full-stat", "Output full execution statistics").Optional().NoArgument().SetFlag(&FullStatistics); + + opts.AddLongOption("sql-flags", "SQL translator pragma flags").SplitHandler(&SqlFlags, ','); + opts.AddLongOption("syntax-version", "SQL syntax version").StoreResult(&SyntaxVersion).DefaultValue(1); + opts.AddLongOption("ansi-lexer", "Use ansi lexer").NoArgument().SetFlag(&AnsiLexer); + opts.AddLongOption("assume-ydb-on-slash", "Assume YDB provider if cluster name starts with '/'").NoArgument().SetFlag(&AssumeYdbOnClusterWithSlash); + opts.AddLongOption("test-antlr4", "Check antlr4 parser").NoArgument().SetFlag(&TestAntlr4); + + opts.AddLongOption("with-final-issues", "Include some final messages (like statistic) in issues").NoArgument().SetFlag(&WithFinalIssues); + if (FailureInjectionSupport) { + opts.AddLongOption("failure-inject", "Activate failure injection") + .Optional() + .RequiredArgument("INJECTION_NAME=FAIL_COUNT or INJECTION_NAME=SKIP_COUNT/FAIL_COUNT") + .KVHandler([](TString name, TString value) { + TFailureInjector::Activate(); + TStringBuf fail = value; + TStringBuf skip; + if (TStringBuf(value).TrySplit('/', skip, fail)) { + TFailureInjector::Set(name, FromString<ui32>(skip), FromString<ui32>(fail)); + } else { + TFailureInjector::Set(name, 0, FromString<ui32>(fail)); + } + }); + } + + opts.SetFreeArgsMax(0); + + for (auto& ext: OptExtenders_) { + ext(opts); + } + + auto res = NLastGetopt::TOptsParseResult(&opts, argc, argv); + + for (auto& handle: OptHandlers_) { + handle(res); + } + + if (Mode >= ERunMode::Validate && GatewayTypes.empty()) { + throw yexception() << "At least one gateway from the list " << JoinSeq(",", SupportedGateways_).Quote() << " must be specified"; + } + + if (!GatewaysConfig) { + GatewaysConfig = ParseProtoFromResource<TGatewaysConfig>("gateways.conf"); + } + + if (GatewaysConfig && GatewaysConfig->HasSqlCore()) { + SqlFlags.insert(GatewaysConfig->GetSqlCore().GetTranslationFlags().begin(), GatewaysConfig->GetSqlCore().GetTranslationFlags().end()); + } + + if (!FsConfig) { + FsConfig = MakeHolder<TFileStorageConfig>(); + if (NResource::Has("fs.conf")) { + LoadFsConfigFromResource("fs.conf", *FsConfig); + } + } +} + +int TFacadeRunner::Main(int argc, const char *argv[]) { + NYql::NBacktrace::RegisterKikimrFatalActions(); + NYql::NBacktrace::EnableKikimrSymbolize(); + + NYql::NLog::YqlLoggerScope logger(&Cerr); + try { + return DoMain(argc, argv); + } + catch (...) { + Cerr << CurrentExceptionMessage() << Endl; + return 1; + } +} + +int TFacadeRunner::DoMain(int argc, const char *argv[]) { + Y_UNUSED(NUdf::GetStaticSymbols()); + + RunOptions_.Parse(argc, argv); + + if (!RunOptions_.NoDebug) { + Cerr << Name_ << " ABI version: " << NKikimr::NUdf::CurrentAbiVersionStr() << Endl; + } + + RunOptions_.InitLogger(); + + if (RunOptions_.PgSupport) { + ClusterMapping_["pg_catalog"] = PgProviderName; + ClusterMapping_["information_schema"] = PgProviderName; + + NPg::SetSqlLanguageParser(NSQLTranslationPG::CreateSqlLanguageParser()); + NPg::LoadSystemFunctions(*NSQLTranslationPG::CreateSystemFunctionsParser()); + if (RunOptions_.PgExtConfig) { + TVector<NPg::TExtensionDesc> extensions; + PgExtensionsFromProto(*RunOptions_.PgExtConfig, extensions); + NPg::RegisterExtensions(extensions, false, + *NSQLTranslationPG::CreateExtensionSqlParser(), + NKikimr::NMiniKQL::CreateExtensionLoader().get()); + } + + NPg::GetSqlLanguageParser()->Freeze(); + } + + FuncRegistry_ = NKikimr::NMiniKQL::CreateFunctionRegistry(&NYql::NBacktrace::KikimrBackTrace, + NKikimr::NMiniKQL::CreateBuiltinRegistry(), true, RunOptions_.UdfsPaths); + + TExprContext ctx; + if (RunOptions_.PgSupport) { + ctx.NextUniqueId = NPg::GetSqlLanguageParser()->GetContext().NextUniqueId; + } + IModuleResolver::TPtr moduleResolver; + if (RunOptions_.MountConfig) { + TModulesTable modules; + FillUserDataTableFromFileSystem(*RunOptions_.MountConfig, RunOptions_.DataTable); + + if (!CompileLibraries(RunOptions_.DataTable, ctx, modules)) { + *RunOptions_.ErrStream << "Errors on compile libraries:" << Endl; + ctx.IssueManager.GetIssues().PrintTo(*RunOptions_.ErrStream); + return -1; + } + + moduleResolver = std::make_shared<TModuleResolver>(std::move(modules), ctx.NextUniqueId, ClusterMapping_, RunOptions_.SqlFlags, RunOptions_.Mode >= ERunMode::Validate); + } else { + if (!GetYqlDefaultModuleResolver(ctx, moduleResolver, ClusterMapping_, RunOptions_.Mode >= ERunMode::Validate)) { + *RunOptions_.ErrStream << "Errors loading default YQL libraries:" << Endl; + ctx.IssueManager.GetIssues().PrintTo(*RunOptions_.ErrStream); + return -1; + } + } + + TExprContext::TFreezeGuard freezeGuard(ctx); + + if (RunOptions_.Mode >= ERunMode::Validate) { + std::vector<NFS::IDownloaderPtr> downloaders; + for (auto& factory: FsDownloadFactories_) { + downloaders.push_back(factory()); + } + + FileStorage_ = WithAsync(CreateFileStorage(*RunOptions_.FsConfig, downloaders)); + } + + IUdfResolver::TPtr udfResolver; + TUdfIndex::TPtr udfIndex; + if (FileStorage_ && RunOptions_.ScanUdfs) { + if (!RunOptions_.UdfResolverPath) { + Cerr << "udf-resolver path must be specified when use 'scan-udfs'"; + return -1; + } + + udfResolver = NCommon::CreateOutProcUdfResolver(FuncRegistry_.Get(), FileStorage_, RunOptions_.UdfResolverPath, {}, {}, RunOptions_.UdfResolverFilterSyscalls, {}); + + RunOptions_.PrintInfo(TStringBuilder() << TInstant::Now().ToStringLocalUpToSeconds() << " Udf scanning started for " << RunOptions_.UdfsPaths.size() << " udfs ..."); + udfIndex = new TUdfIndex(); + LoadRichMetadataToUdfIndex(*udfResolver, RunOptions_.UdfsPaths, false, TUdfIndex::EOverrideMode::RaiseError, *udfIndex); + RunOptions_.PrintInfo(TStringBuilder() << TInstant::Now().ToStringLocalUpToSeconds() << " UdfIndex done."); + + udfResolver = NCommon::CreateUdfResolverWithIndex(udfIndex, udfResolver, FileStorage_); + RunOptions_.PrintInfo(TStringBuilder() << TInstant::Now().ToStringLocalUpToSeconds() << " Udfs scanned"); + } else { + udfResolver = FileStorage_ && RunOptions_.UdfResolverPath + ? NCommon::CreateOutProcUdfResolver(FuncRegistry_.Get(), FileStorage_, RunOptions_.UdfResolverPath, {}, {}, RunOptions_.UdfResolverFilterSyscalls, {}) + : NCommon::CreateSimpleUdfResolver(FuncRegistry_.Get(), FileStorage_, true); + } + + TVector<TDataProviderInitializer> dataProvidersInit; + if (RunOptions_.PgSupport) { + dataProvidersInit.push_back(GetPgDataProviderInitializer()); + } + for (auto& factory: ProviderFactories_) { + dataProvidersInit.push_back(factory()); + } + + TVector<IUrlListerPtr> urlListers; + for (auto& factory: UrlListerFactories_) { + urlListers.push_back(factory()); + } + + TProgramFactory factory(RunOptions_.UseRepeatableRandomAndTimeProviders, FuncRegistry_.Get(), ctx.NextUniqueId, dataProvidersInit, Name_); + factory.AddUserDataTable(RunOptions_.DataTable); + factory.SetModules(moduleResolver); + factory.SetFileStorage(FileStorage_); + if (RunOptions_.GatewaysConfig && RunOptions_.GatewaysConfig->HasFs()) { + factory.SetUrlPreprocessing(new NYql::TUrlPreprocessing(*RunOptions_.GatewaysConfig)); + } + factory.SetUdfIndex(udfIndex, new TUdfIndexPackageSet()); + factory.SetUdfResolver(udfResolver); + factory.SetGatewaysConfig(RunOptions_.GatewaysConfig.Get()); + factory.SetCredentials(Credentials_); + factory.EnableRangeComputeFor(); + if (!urlListers.empty()) { + factory.SetUrlListerManager(MakeUrlListerManager(urlListers)); + } + + return RunProgram(factory); +} + +int TFacadeRunner::RunProgram(TProgramFactory& factory) { + + TProgramPtr program = factory.Create(RunOptions_.ProgramFile, RunOptions_.ProgramText);; + if (RunOptions_.Params) { + program->SetParametersYson(RunOptions_.Params); + } + + if (RunOptions_.EnableResultPosition) { + program->EnableResultPosition(); + } + + if (ProgressWriter_) { + program->SetProgressWriter(ProgressWriter_); + } + program->SetUseTableMetaFromGraph(RunOptions_.UseMetaFromGrpah); + program->SetValidateOptions(RunOptions_.ValidateMode); + + bool fail = false; + if (RunOptions_.ProgramType != EProgramType::SExpr) { + RunOptions_.PrintInfo("Parse SQL..."); + google::protobuf::Arena arena; + NSQLTranslation::TTranslationSettings settings; + settings.Arena = &arena; + settings.PgParser = EProgramType::Pg == RunOptions_.ProgramType; + settings.ClusterMapping = ClusterMapping_; + settings.Flags = RunOptions_.SqlFlags; + settings.SyntaxVersion = RunOptions_.SyntaxVersion; + settings.AnsiLexer = RunOptions_.AnsiLexer; + settings.TestAntlr4 = RunOptions_.TestAntlr4; + settings.V0Behavior = NSQLTranslation::EV0Behavior::Report; + settings.AssumeYdbOnClusterWithSlash = RunOptions_.AssumeYdbOnClusterWithSlash; + if (ERunMode::Discover == RunOptions_.Mode) { + settings.Mode = NSQLTranslation::ESqlMode::DISCOVERY; + } + if (!program->ParseSql(settings)) { + program->PrintErrorsTo(*RunOptions_.ErrStream); + fail = true; + } + if (!fail && RunOptions_.TestSqlFormat && 1 == RunOptions_.SyntaxVersion) { + TString formattedProgramText; + NYql::TIssues issues; + auto formatter = NSQLFormat::MakeSqlFormatter(settings); + if (!formatter->Format(RunOptions_.ProgramText, formattedProgramText, issues)) { + *RunOptions_.ErrStream << "Format failed" << Endl; + issues.PrintTo(*RunOptions_.ErrStream); + return -1; + } + + auto frmProgram = factory.Create("formatted SQL", formattedProgramText); + if (!frmProgram->ParseSql(settings)) { + frmProgram->PrintErrorsTo(*RunOptions_.ErrStream); + return -1; + } + + TStringStream srcQuery, frmQuery; + + program->AstRoot()->PrettyPrintTo(srcQuery, PRETTY_FLAGS); + frmProgram->AstRoot()->PrettyPrintTo(frmQuery, PRETTY_FLAGS); + if (srcQuery.Str() != frmQuery.Str()) { + *RunOptions_.ErrStream << "source query's AST and formatted query's AST are not same" << Endl; + return -1; + } + } + } else { + RunOptions_.PrintInfo("Parse YQL..."); + if (!program->ParseYql()) { + program->PrintErrorsTo(*RunOptions_.ErrStream); + fail = true; + } + } + + if (RunOptions_.TraceOptStream) { + if (auto ast = program->GetQueryAst()) { + *RunOptions_.TraceOptStream << *ast << Endl; + } + } + if (fail) { + return -1; + } + + if (RunOptions_.PrintAst) { + program->AstRoot()->PrettyPrintTo(Cout, PRETTY_FLAGS); + } + + if (ERunMode::Parse == RunOptions_.Mode) { + return 0; + } + + RunOptions_.PrintInfo("Compile program..."); + if (!program->Compile(RunOptions_.User)) { + program->PrintErrorsTo(*RunOptions_.ErrStream); + fail = true; + } + + if (RunOptions_.TraceOptStream) { + program->Print(RunOptions_.TraceOptStream, nullptr); + } + if (fail) { + return -1; + } + + if (ERunMode::Compile == RunOptions_.Mode) { + if (RunOptions_.ExprStream) { + auto baseAst = ConvertToAst(*program->ExprRoot(), program->ExprCtx(), NYql::TExprAnnotationFlags::None, true); + baseAst.Root->PrettyPrintTo(*RunOptions_.ExprStream, PRETTY_FLAGS); + } + return 0; + } + + auto defOptConfig = TOptPipelineConfigurator(program, RunOptions_.FullExpr ? RunOptions_.PlanStream : nullptr, RunOptions_.FullExpr ? RunOptions_.ExprStream : nullptr, RunOptions_.WithTypes); + IPipelineConfigurator* optConfig = OptPipelineConfigurator_ ? OptPipelineConfigurator_ : &defOptConfig; + + TProgram::TStatus status = TProgram::TStatus::Ok; + if (ERunMode::Peephole == RunOptions_.Mode) { + RunOptions_.PrintInfo("Peephole..."); + auto defConfig = TPeepHolePipelineConfigurator(); + IPipelineConfigurator* config = PeepholePipelineConfigurator_ ? PeepholePipelineConfigurator_ : &defConfig; + status = program->OptimizeWithConfig(RunOptions_.User, *config); + + if (RunOptions_.ExprStream && program->ExprRoot()) { + auto ast = ConvertToAst(*program->ExprRoot(), program->ExprCtx(), RunOptions_.WithTypes ? TExprAnnotationFlags::Types : TExprAnnotationFlags::None, true); + ui32 prettyFlags = TAstPrintFlags::ShortQuote; + if (!RunOptions_.WithTypes) { + prettyFlags |= TAstPrintFlags::PerLine; + } + ast.Root->PrettyPrintTo(*RunOptions_.ExprStream, prettyFlags); + } + + } else if (ERunMode::Run == RunOptions_.Mode) { + RunOptions_.PrintInfo("Run program..."); + status = program->RunWithConfig(RunOptions_.User, *optConfig); + } else if (ERunMode::Optimize == RunOptions_.Mode) { + RunOptions_.PrintInfo("Optimize program..."); + status = program->OptimizeWithConfig(RunOptions_.User, *optConfig); + } else if (ERunMode::Validate == RunOptions_.Mode) { + RunOptions_.PrintInfo("Validate program..."); + status = program->Validate(RunOptions_.User, RunOptions_.ExprStream, RunOptions_.WithTypes); + } else if (ERunMode::Discover == RunOptions_.Mode) { + RunOptions_.PrintInfo("Discover program..."); + status = program->Discover(RunOptions_.User); + } else if (ERunMode::Lineage == RunOptions_.Mode) { + RunOptions_.PrintInfo("Calculating lineage in program..."); + status = program->LineageWithConfig(RunOptions_.User, *optConfig); + } + + if (RunOptions_.WithFinalIssues) { + program->FinalizeIssues(); + } + program->PrintErrorsTo(*RunOptions_.ErrStream); + if (status == TProgram::TStatus::Error) { + if (RunOptions_.TraceOptStream) { + program->Print(RunOptions_.TraceOptStream, nullptr); + } + return -1; + } + + if (!RunOptions_.FullExpr && ERunMode::Peephole != RunOptions_.Mode) { + program->Print(RunOptions_.ExprStream, RunOptions_.PlanStream, /*cleanPlan*/true); + } + + program->ConfigureYsonResultFormat(RunOptions_.ResultsFormat); + + if (RunOptions_.ResultStream) { + RunOptions_.PrintInfo("Getting results..."); + if (ERunMode::Discover == RunOptions_.Mode) { + if (auto data = program->GetDiscoveredData()) { + *RunOptions_.ResultStream << *data; + } + } else if (ERunMode::Lineage == RunOptions_.Mode) { + if (auto data = program->GetLineage()) { + TStringInput in(*data); + NYson::ReformatYsonStream(&in, RunOptions_.ResultStream, RunOptions_.ResultsFormat); + } + } else if (program->HasResults()) { + if (RunOptions_.ValidateResultFormat) { + auto str = program->ResultsAsString(); + if (!str.empty()) { + auto node = NYT::NodeFromYsonString(str); + for (const auto& r : NResult::ParseResponse(node)) { + for (const auto& write : r.Writes) { + if (write.Type) { + NResult::TEmptyTypeVisitor visitor; + NResult::ParseType(*write.Type, visitor); + } + + if (write.Type && write.Data) { + NResult::TEmptyDataVisitor visitor; + NResult::ParseData(*write.Type, *write.Data, visitor); + } + } + } + } + + RunOptions_.ResultStream->Write(str.data(), str.size()); + } else { + *RunOptions_.ResultStream << program->ResultsAsString(); + } + } + } + + if (RunOptions_.StatStream) { + if (auto st = program->GetStatistics(!RunOptions_.FullStatistics)) { + *RunOptions_.StatStream << *st; + } + } + + RunOptions_.PrintInfo(""); + RunOptions_.PrintInfo("Done"); + + return 0; +} + +} // NYql diff --git a/yql/essentials/tools/yql_facade_run/yql_facade_run.h b/yql/essentials/tools/yql_facade_run/yql_facade_run.h new file mode 100644 index 0000000000..0b7e7a9e8d --- /dev/null +++ b/yql/essentials/tools/yql_facade_run/yql_facade_run.h @@ -0,0 +1,213 @@ +#pragma once + +#include <yql/essentials/core/file_storage/defs/downloader.h> +#include <yql/essentials/core/file_storage/file_storage.h> +#include <yql/essentials/core/credentials/yql_credentials.h> +#include <yql/essentials/core/url_lister/interface/url_lister.h> +#include <yql/essentials/core/yql_data_provider.h> +#include <yql/essentials/core/yql_user_data.h> + +#include <library/cpp/getopt/last_getopt.h> +#include <library/cpp/yson/public.h> +#include <library/cpp/logger/priority.h> + +#include <util/generic/hash_set.h> +#include <util/generic/hash.h> +#include <util/generic/strbuf.h> +#include <util/generic/string.h> + +#include <functional> + +namespace NKikimr::NMiniKQL { + class IFunctionRegistry; +} + +namespace NYql { + class TFileStorageConfig; + class TGatewaysConfig; + class TProgramFactory; +} + +namespace NYql::NProto { + class TPgExtensions; +} + +namespace NYqlMountConfig { + class TMountConfig; +} + +namespace NYql { + +enum class ERunMode { + Parse /* "parse" */, + Compile /* "compile" */, + Validate /* "validate" */, + Optimize /* "optimize" */, + Peephole /* "peephole" */, + Lineage /* "lineage" */, + Discover /* "discover" */, + Run /* "run" */, +}; + +enum class EProgramType { + SExpr /* "s-expr" */, + Sql /* "sql" */, + Pg /* "pg" */, +}; + +class TFacadeRunOptions { +public: + TFacadeRunOptions(); + ~TFacadeRunOptions(); + + EProgramType ProgramType = EProgramType::SExpr; + NYson::EYsonFormat ResultsFormat = NYson::EYsonFormat::Text; + ERunMode Mode = ERunMode::Run; + TString ProgramFile; + TString ProgramText; + TString User; + ui64 MemLimit = 0; + + THashSet<TString> SqlFlags; + ui16 SyntaxVersion = 1; + bool AnsiLexer = false; + bool TestAntlr4 = false; + bool AssumeYdbOnClusterWithSlash = false; + + bool PrintAst = false; + bool FullExpr = false; + bool WithTypes = false; + bool FullStatistics = false; + int Verbosity = TLOG_ERR; + bool ShowLog = false; + bool WithFinalIssues = false; + + IOutputStream* TraceOptStream = nullptr; + + IOutputStream* ErrStream = &Cerr; + THolder<IOutputStream> ErrStreamHolder; + IOutputStream* PlanStream = nullptr; + THolder<IOutputStream> PlanStreamHolder; + IOutputStream* ExprStream = nullptr; + THolder<IOutputStream> ExprStreamHolder; + IOutputStream* ResultStream = nullptr; + THolder<IOutputStream> ResultStreamHolder; + IOutputStream* StatStream = nullptr; + THolder<IOutputStream> StatStreamHolder; + + NYql::TUserDataTable DataTable; + TVector<TString> UdfsPaths; + TString Params; + NUdf::EValidateMode ValidateMode = NUdf::EValidateMode::Greedy; + + THashSet<TString> GatewayTypes; + TString UdfResolverPath; + bool UdfResolverFilterSyscalls = false; + bool ScanUdfs = false; + THolder<NYqlMountConfig::TMountConfig> MountConfig; + THolder<TGatewaysConfig> GatewaysConfig; + THolder<TFileStorageConfig> FsConfig; + THolder<NProto::TPgExtensions> PgExtConfig; + + // No command line options for these settings. Should be configured in the inherited class + bool NoDebug = false; + bool PgSupport = true; + bool FailureInjectionSupport = false; + bool UseRepeatableRandomAndTimeProviders = false; + bool UseMetaFromGrpah = false; + bool TestSqlFormat = false; + bool ValidateResultFormat = false; + bool EnableResultPosition = false; + + void Parse(int argc, const char *argv[]); + + void AddOptExtension(std::function<void(NLastGetopt::TOpts& opts)> optExtender) { + OptExtenders_.push_back(std::move(optExtender)); + } + void AddOptHandler(std::function<void(const NLastGetopt::TOptsParseResult& res)> optHandler) { + OptHandlers_.push_back(std::move(optHandler)); + } + void SetSupportedGateways(std::initializer_list<TString> gateways) { + SupportedGateways_.insert(gateways); + } + + void InitLogger(); + + void PrintInfo(const TString& msg); + +private: + std::vector<std::function<void(NLastGetopt::TOpts&)>> OptExtenders_; + std::vector<std::function<void(const NLastGetopt::TOptsParseResult&)>> OptHandlers_; + THashSet<TString> SupportedGateways_; +}; + +class TFacadeRunner { +public: + TFacadeRunner(TString name) + : Name_(std::move(name)) + { + } + int Main(int argc, const char *argv[]); + + void AddFsDownloadFactory(std::function<NFS::IDownloaderPtr()> factory) { + FsDownloadFactories_.push_back(std::move(factory)); + } + void AddProviderFactory(std::function<NYql::TDataProviderInitializer()> factory) { + ProviderFactories_.push_back(std::move(factory)); + } + void AddUrlListerFactory(std::function<IUrlListerPtr()> factory) { + UrlListerFactories_.push_back(std::move(factory)); + } + void AddClusterMapping(TString name, TString provider) { + ClusterMapping_[name] = std::move(provider); + } + template <class TPbConfig> + void FillClusterMapping(const TPbConfig& config, const TString& provider) { + for (auto& cluster: config.GetClusterMapping()) { + ClusterMapping_.emplace(to_lower(cluster.GetName()), provider); + } + } + void SetOperationProgressWriter(TOperationProgressWriter writer) { + ProgressWriter_ = std::move(writer); + } + void SetOptPipelineConfigurator(IPipelineConfigurator* configurator) { + OptPipelineConfigurator_ = configurator; + } + void SetPeepholePipelineConfigurator(IPipelineConfigurator* configurator) { + PeepholePipelineConfigurator_ = configurator; + } + + TFileStoragePtr GetFileStorage() const { + return FileStorage_; + } + TIntrusivePtr<NKikimr::NMiniKQL::IFunctionRegistry> GetFuncRegistry() { + return FuncRegistry_; + } + TCredentials::TPtr GetCredentials() { + return Credentials_; + } + TFacadeRunOptions& GetRunOptions() { + return RunOptions_; + } + +private: + int DoMain(int argc, const char *argv[]); + int RunProgram(TProgramFactory& factory); + +private: + TString Name_; + std::vector<std::function<NFS::IDownloaderPtr()>> FsDownloadFactories_; + std::vector<std::function<TDataProviderInitializer()>> ProviderFactories_; + std::vector<std::function<IUrlListerPtr()>> UrlListerFactories_; + THashMap<TString, TString> ClusterMapping_; + THolder<TFileStorageConfig> FileStorageConfig_; + TFileStoragePtr FileStorage_; + TIntrusivePtr<NKikimr::NMiniKQL::IFunctionRegistry> FuncRegistry_; + TCredentials::TPtr Credentials_ = MakeIntrusive<TCredentials>(); + TOperationProgressWriter ProgressWriter_; + IPipelineConfigurator* OptPipelineConfigurator_ = nullptr; + IPipelineConfigurator* PeepholePipelineConfigurator_ = nullptr; + TFacadeRunOptions RunOptions_; +}; + +} // NYql |