Add stats mode profile option to ydb query exec command. ()

2 files changed, 4 insertions, 1 deletions

diff --git a/ydb/public/lib/ydb_cli/commands/ydb_service_table.cpp b/ydb/public/lib/ydb_cli/commands/ydb_service_table.cpp
index 0b3a7dc0b4..9b401976fd 100644
--- a/ydb/public/lib/ydb_cli/commands/ydb_service_table.cpp
+++ b/ydb/public/lib/ydb_cli/commands/ydb_service_table.cpp
@@ -539,7 +539,8 @@ void TCommandExecuteQuery::PrintScanQueryResponse(NTable::TScanQueryPartIterator
 
     if (fullStats) {
         Cout << Endl << "Full statistics:" << Endl;
-        TQueryPlanPrinter queryPlanPrinter(EOutputFormat::Pretty, /* analyzeMode */ true);
+
+        TQueryPlanPrinter queryPlanPrinter(OutputFormat, /* analyzeMode */ true);
         queryPlanPrinter.Print(*fullStats);
     }
 
diff --git a/ydb/public/lib/ydb_cli/common/query_stats.cpp b/ydb/public/lib/ydb_cli/common/query_stats.cpp
index b53f8e746e..d6669cb575 100644
--- a/ydb/public/lib/ydb_cli/common/query_stats.cpp
+++ b/ydb/public/lib/ydb_cli/common/query_stats.cpp
@@ -15,6 +15,8 @@ NTable::ECollectQueryStatsMode ParseQueryStatsMode(const TString& statsMode,
             return NTable::ECollectQueryStatsMode::Basic;
         } else if (statsMode == "full") {
             return NTable::ECollectQueryStatsMode::Full;
+        } else if (statsMode == "profile") {
+            return NTable::ECollectQueryStatsMode::Profile;
         } else {
             throw TMisuseException() << "Unknown stats collection mode.";
         }