diff options
| author | andrewproni <[email protected]> | 2023-08-18 16:05:37 +0300 |
|---|---|---|
| committer | andrewproni <[email protected]> | 2023-08-18 18:37:04 +0300 |
| commit | df6175995b904c7722bf8709a97815e13b1ccd70 (patch) | |
| tree | b41e37483f51a093a3b55ede515fc270e4215129 | |
| parent | 36366345dbbdb0d882ff2331c6e00ab007687f07 (diff) | |
AST for ScriptExecutionOperation
| -rw-r--r-- | ydb/core/kqp/session_actor/kqp_session_actor.cpp | 1 | ||||
| -rw-r--r-- | ydb/core/kqp/ut/service/kqp_query_service_ut.cpp | 2 |
2 files changed, 3 insertions, 0 deletions
diff --git a/ydb/core/kqp/session_actor/kqp_session_actor.cpp b/ydb/core/kqp/session_actor/kqp_session_actor.cpp index 6c2c2efa8f8..bdf855d08c2 100644 --- a/ydb/core/kqp/session_actor/kqp_session_actor.cpp +++ b/ydb/core/kqp/session_actor/kqp_session_actor.cpp @@ -1240,6 +1240,7 @@ public: if (QueryState->ReportStats()) { if (QueryState->GetStatsMode() >= Ydb::Table::QueryStatsCollection::STATS_COLLECTION_FULL) { response->SetQueryPlan(SerializeAnalyzePlan(*stats)); + response->SetQueryAst(QueryState->CompileResult->PreparedQuery->GetPhysicalQuery().GetQueryAst()); } response->MutableQueryStats()->Swap(stats); } diff --git a/ydb/core/kqp/ut/service/kqp_query_service_ut.cpp b/ydb/core/kqp/ut/service/kqp_query_service_ut.cpp index 09b2c7b7f3f..8f73db8520b 100644 --- a/ydb/core/kqp/ut/service/kqp_query_service_ut.cpp +++ b/ydb/core/kqp/ut/service/kqp_query_service_ut.cpp @@ -601,6 +601,7 @@ Y_UNIT_TEST_SUITE(KqpQueryService) { UNIT_ASSERT_STRING_CONTAINS(readyOp.Metadata().ScriptContent.Text, "SELECT 42"); ValidatePlan(readyOp.Metadata().ExecStats.query_plan()); + UNIT_ASSERT(readyOp.Metadata().ExecStats.query_ast()); } Y_UNIT_TEST(ParseScript) { @@ -715,6 +716,7 @@ Y_UNIT_TEST_SUITE(KqpQueryService) { } ValidatePlan(readyOp.Metadata().ExecStats.query_plan()); + UNIT_ASSERT(readyOp.Metadata().ExecStats.query_ast()); } Y_UNIT_TEST(ExecuteScriptStatsBasic) { |