Fixes

Author: shnikd

ydb/public/lib/ydb_cli/commands/ydb_sql.cpp

@@ -9,6 +9,7 @@
 #include <ydb/public/lib/ydb_cli/common/query_stats.h>
 #include <ydb/public/lib/ydb_cli/common/waiting_bar.h>
 #include <ydb/public/sdk/cpp/client/ydb_proto/accessor.h>
+#include <util/generic/guid.h>
 #include <util/generic/queue.h>
 #include <google/protobuf/text_format.h>
 
@@ -40,6 +41,8 @@ void TCommandSql::Config(TConfig& config) {
     config.Opts->AddLongOption("syntax", "Query syntax [yql, pg]")
         .RequiredArgument("[String]").DefaultValue("yql").StoreResult(&Syntax)
         .Hidden();
+    config.Opts->AddLongOption("collect-diagnostics", "Collects diagnostics and saves it to file")
+        .StoreTrue(&CollectFullDiagnostics);
 
     AddOutputFormats(config, {
         EDataFormat::Pretty,
@@ -135,6 +138,10 @@ int TCommandSql::RunCommand(TConfig& config) {
         throw TMisuseException() << "Unknow syntax option \"" << Syntax << "\"";
     }
 
+    if (CollectFullDiagnostics) {
+        settings.CollectFullDiagnostics(true);
+    }
+
     if (!Parameters.empty() || InputParamStream) {
         // Execute query with parameters
         THolder<TParamsBuilder> paramBuilder;
@@ -171,6 +178,7 @@ int TCommandSql::RunCommand(TConfig& config) {
 int TCommandSql::PrintResponse(NQuery::TExecuteQueryIterator& result) {
     TMaybe<TString> stats;
     TMaybe<TString> plan;
+    TString diagnostics;
     {
         TResultSetPrinter printer(OutputFormat, &IsInterrupted);
 
@@ -192,6 +200,8 @@ int TCommandSql::PrintResponse(NQuery::TExecuteQueryIterator& result) {
                     plan = queryStats.GetPlan();
                 }
             }
+
+            diagnostics = streamPart.GetDiagnostics();
         }
     } // TResultSetPrinter destructor should be called before printing stats
 
@@ -212,6 +222,11 @@ int TCommandSql::PrintResponse(NQuery::TExecuteQueryIterator& result) {
         queryPlanPrinter.Print(*plan);
     }
 
+    if (CollectFullDiagnostics) {
+        TFileOutput file(TStringBuilder() << "diagnostics_" << TGUID::Create().AsGuidString() << ".txt");
+        file << diagnostics;
+    }
+
     if (IsInterrupted()) {
         Cerr << "<INTERRUPTED>" << Endl;
         return EXIT_FAILURE;

ydb/public/lib/ydb_cli/commands/ydb_sql.h

@@ -33,6 +33,7 @@ class TCommandSql : public TYdbCommand, public TCommandWithOutput, public TComma
     TString Syntax;
     bool ExplainMode = false;
     bool ExplainAnalyzeMode = false;
+    bool CollectFullDiagnostics = false;
 };
 
 }