Add ExecuteMode to YQL plugin for YT. Support validate and explain modes

Author: Krock21

ydb/library/yql/yt/bridge/interface.h

@@ -97,7 +97,8 @@ using TFuncBridgeRun = TBridgeQueryResult*(
     const char* queryText,
     const char* settings,
     const TBridgeQueryFile* files,
-    int fileCount);
+    int fileCount,
+    int executeMode);
 using TFuncBridgeGetProgress = TBridgeQueryResult*(TBridgeYqlPlugin* plugin, const char* queryId);
 using TFuncBridgeAbort = TBridgeAbortResult*(TBridgeYqlPlugin* plugin, const char* queryId);
 using TFuncBridgeFreeAbortResult = void(TBridgeAbortResult* result);

ydb/library/yql/yt/dynamic/impl.cpp

@@ -12,7 +12,7 @@ extern "C" {
 
 ssize_t BridgeGetAbiVersion()
 {
-    return 1; // EYqlPluginAbiVersion::AbortQuery
+    return 2; // EYqlPluginAbiVersion
 }
 
 TBridgeYqlPlugin* BridgeCreateYqlPlugin(const TBridgeYqlPluginOptions* bridgeOptions)
@@ -75,7 +75,8 @@ TBridgeQueryResult* BridgeRun(
     const char* queryText,
     const char* settings,
     const TBridgeQueryFile* bridgeFiles,
-    int bridgeFileCount)
+    int bridgeFileCount,
+    int executeMode)
 {
     static const auto EmptyMap = TYsonString(TString("{}"));
 
@@ -97,7 +98,8 @@ TBridgeQueryResult* BridgeRun(
         TString(impersonationUser),
         TString(queryText),
         settings ? TYsonString(TString(settings)) : EmptyMap,
-        files);
+        files,
+        executeMode);
     FillString(bridgeResult->YsonResult, bridgeResult->YsonResultLength, result.YsonResult);
     FillString(bridgeResult->Plan, bridgeResult->PlanLength, result.Plan);
     FillString(bridgeResult->Statistics, bridgeResult->StatisticsLength, result.Statistics);

ydb/library/yql/yt/native/plugin.cpp

@@ -254,7 +254,8 @@ class TYqlPlugin
         TString impersonationUser,
         TString queryText,
         TYsonString settings,
-        std::vector<TQueryFile> files)
+        std::vector<TQueryFile> files,
+        int executeMode)
     {
         auto program = ProgramFactory_->Create("-memory-", queryText);
         {
@@ -313,8 +314,45 @@ class TYqlPlugin
             ActiveQueriesProgress_[queryId].Compiled = true;
         }
 
+        NYql::NEmbedded::EExecuteMode executeModeEnum;
+
+        // Transforming NYT::NYqlClient::EExecuteMode (yt/yt/ytlib/yql_client/public.h)
+        // to NYql::EExecuteMode
+        switch (executeMode) {
+        case 0:
+            executeModeEnum = NYql::NEmbedded::EExecuteMode::Validate;
+            break;
+        case 1:
+            executeModeEnum = NYql::NEmbedded::EExecuteMode::Optimize;
+            break;
+        case 2:
+            executeModeEnum = NYql::NEmbedded::EExecuteMode::Run;
+            break;
+        case 3:
+            executeModeEnum = NYql::NEmbedded::EExecuteMode::Lineage;
+            break;
+        default:
+            return TQueryResult{
+                .YsonError = MessageToYtErrorYson(Format("Unknown execution mode: %v", executeMode)),
+            };
+        }
+
         NYql::TProgram::TStatus status = NYql::TProgram::TStatus::Error;
-        status = program->RunWithConfig(impersonationUser, pipelineConfigurator);
+        switch (executeModeEnum) {
+        case NYql::NEmbedded::EExecuteMode::Run:
+            status = program->RunWithConfig(impersonationUser, pipelineConfigurator);
+            break;
+        case NYql::NEmbedded::EExecuteMode::Optimize:
+            status = program->OptimizeWithConfig(impersonationUser, pipelineConfigurator);
+            break;
+        case NYql::NEmbedded::EExecuteMode::Validate:
+            status = program->Validate(impersonationUser, nullptr);
+            break;
+        case NYql::NEmbedded::EExecuteMode::Lineage:
+            return TQueryResult{
+                .YsonError = MessageToYtErrorYson("Lineage execution mode is not supported"),
+            };
+        }
 
         if (status == NYql::TProgram::TStatus::Error) {
             return TQueryResult{
@@ -354,10 +392,11 @@ class TYqlPlugin
         TString impersonationUser,
         TString queryText,
         TYsonString settings,
-        std::vector<TQueryFile> files) noexcept override
+        std::vector<TQueryFile> files,
+        int executeMode) noexcept override
     {
         try {
-            return GuardedRun(queryId, impersonationUser, queryText, settings, files);
+            return GuardedRun(queryId, impersonationUser, queryText, settings, files, executeMode);
         } catch (const std::exception& ex) {
             {
                 auto guard = WriterGuard(ProgressSpinLock);

ydb/library/yql/yt/plugin.h

@@ -75,7 +75,8 @@ struct IYqlPlugin
         TString impersonationUser,
         TString queryText,
         NYson::TYsonString settings,
-        std::vector<TQueryFile> files) noexcept = 0;
+        std::vector<TQueryFile> files,
+        int executeMode) noexcept = 0;
     virtual TQueryResult GetProgress(TQueryId queryId) noexcept = 0;
 
     virtual TAbortResult Abort(TQueryId queryId) noexcept = 0;