Skip to content

Commit 9553004

Browse files
committed
Fixes
1 parent 1eaad6c commit 9553004

File tree

3 files changed

+20
-15
lines changed

3 files changed

+20
-15
lines changed

ydb/core/kqp/ut/query/kqp_query_ut.cpp

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -211,10 +211,13 @@ Y_UNIT_TEST_SUITE(KqpQuery) {
211211

212212
UNIT_ASSERT_VALUES_EQUAL_C(result.GetStatus(), EStatus::SUCCESS, result.GetIssues().ToString().c_str());
213213

214-
UNIT_ASSERT_C(!result.GetMeta().empty(), "Query result meta is empty");
214+
auto stats = result.GetStats();
215+
UNIT_ASSERT(stats.has_value());
216+
217+
UNIT_ASSERT_C(stats->GetMeta().has_value(), "Query result meta is empty");
215218

216219
TStringStream in;
217-
in << result.GetMeta();
220+
in << stats->GetMeta().value();
218221
NJson::TJsonValue value;
219222
ReadJsonTree(&in, &value);
220223

@@ -234,12 +237,16 @@ Y_UNIT_TEST_SUITE(KqpQuery) {
234237

235238
{
236239
auto settings = TExecDataQuerySettings();
240+
settings.CollectQueryStats(ECollectQueryStatsMode::Basic);
237241

238242
auto result = session.ExecuteDataQuery(query, TTxControl::BeginTx().CommitTx(), settings).ExtractValueSync();
239243

240244
UNIT_ASSERT_VALUES_EQUAL_C(result.GetStatus(), EStatus::SUCCESS, result.GetIssues().ToString().c_str());
241245

242-
UNIT_ASSERT_C(result.GetMeta().empty(), "Query result meta should be empty, but it's not");
246+
auto stats = result.GetStats();
247+
UNIT_ASSERT(stats.has_value());
248+
249+
UNIT_ASSERT_C(!stats->GetMeta().has_value(), "Query result meta should be empty, but it's not");
243250
}
244251
}
245252
}

ydb/public/sdk/cpp/include/ydb-cpp-sdk/client/table/table.h

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1163,8 +1163,10 @@ struct TStreamExecScanQuerySettings : public TRequestSettings<TStreamExecScanQue
11631163

11641164
// Collect runtime statistics with a given detalization mode
11651165
FLUENT_SETTING_DEFAULT(ECollectQueryStatsMode, CollectQueryStats, ECollectQueryStatsMode::None);
1166-
// // Collect full query compilation diagnostics
1167-
// FLUENT_SETTING_DEFAULT_DEPRECATED(bool, CollectFullDiagnostics, false);
1166+
1167+
//Deprecated. Use CollectQueryStats >= ECollectQueryStatsMode::Full to get QueryMeta in QueryStats
1168+
// Collect full query compilation diagnostics
1169+
FLUENT_SETTING_DEFAULT(bool, CollectFullDiagnostics, false);
11681170
};
11691171

11701172
enum class EDataFormat {
@@ -2036,8 +2038,6 @@ class TDataQueryResult : public TStatus {
20362038

20372039
const std::string GetQueryPlan() const;
20382040

2039-
const std::string GetMeta() const;
2040-
20412041
private:
20422042
std::optional<TTransaction> Transaction_;
20432043
std::vector<TResultSet> ResultSets_;
@@ -2110,6 +2110,11 @@ class TScanQueryPart : public TStreamPartStatus {
21102110
const TQueryStats& GetQueryStats() const { return *QueryStats_; }
21112111
TQueryStats ExtractQueryStats() { return std::move(*QueryStats_); }
21122112

2113+
// Deprecated. Use GetMeta() of TQueryStats
2114+
bool HasDiagnostics() const { return FakeDiagnostics_.has_value(); }
2115+
const std::string& GetDiagnostics() const { return *FakeDiagnostics_; }
2116+
std::string&& ExtractDiagnostics() { return std::move(*FakeDiagnostics_); }
2117+
21132118
TScanQueryPart(TStatus&& status)
21142119
: TStreamPartStatus(std::move(status))
21152120
{}
@@ -2128,6 +2133,7 @@ class TScanQueryPart : public TStreamPartStatus {
21282133
private:
21292134
std::optional<TResultSet> ResultSet_;
21302135
std::optional<TQueryStats> QueryStats_;
2136+
std::optional<std::string> FakeDiagnostics_;
21312137
};
21322138

21332139
using TAsyncScanQueryPart = NThreading::TFuture<TScanQueryPart>;

ydb/public/sdk/cpp/src/client/table/table.cpp

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2220,14 +2220,6 @@ const std::string TDataQueryResult::GetQueryPlan() const {
22202220
}
22212221
}
22222222

2223-
const std::string TDataQueryResult::GetMeta() const {
2224-
if (QueryStats_.has_value()) {
2225-
return NYdb::TProtoAccessor::GetProto(*QueryStats_).query_meta();
2226-
} else {
2227-
return "";
2228-
}
2229-
}
2230-
22312223
////////////////////////////////////////////////////////////////////////////////
22322224

22332225
TBeginTransactionResult::TBeginTransactionResult(TStatus&& status, TTransaction transaction)

0 commit comments

Comments
 (0)