Skip to content

Commit 27c607b

Browse files
committed
Moved commit "Added add field for pool id into gRPC api" from ydb repo
1 parent 8fe04eb commit 27c607b

File tree

3 files changed

+4
-0
lines changed

3 files changed

+4
-0
lines changed

include/ydb-cpp-sdk/client/query/query.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ struct TExecuteQuerySettings : public TRequestSettings<TExecuteQuerySettings> {
7474
FLUENT_SETTING_DEFAULT(EExecMode, ExecMode, EExecMode::Execute);
7575
FLUENT_SETTING_DEFAULT(EStatsMode, StatsMode, EStatsMode::None);
7676
FLUENT_SETTING_OPTIONAL(bool, ConcurrentResultSets);
77+
FLUENT_SETTING(std::string, PoolId);
7778
};
7879

7980
struct TBeginTxSettings : public TRequestSettings<TBeginTxSettings> {};
@@ -97,6 +98,7 @@ struct TExecuteScriptSettings : public TOperationRequestSettings<TExecuteScriptS
9798
FLUENT_SETTING_DEFAULT(EExecMode, ExecMode, EExecMode::Execute);
9899
FLUENT_SETTING_DEFAULT(EStatsMode, StatsMode, EStatsMode::None);
99100
FLUENT_SETTING(TDuration, ResultsTtl);
101+
FLUENT_SETTING(std::string, PoolId);
100102
};
101103

102104
class TQueryContent {

src/client/query/client.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ class TQueryClient::TImpl: public TClientImplCommon<TQueryClient::TImpl>, public
8989
auto request = MakeOperationRequest<ExecuteScriptRequest>(settings);
9090
request.set_exec_mode(::Ydb::Query::ExecMode(settings.ExecMode_));
9191
request.set_stats_mode(::Ydb::Query::StatsMode(settings.StatsMode_));
92+
request.set_pool_id(settings.PoolId_);
9293
request.mutable_script_content()->set_syntax(::Ydb::Query::Syntax(settings.Syntax_));
9394
request.mutable_script_content()->set_text(TStringType{script});
9495
SetDuration(settings.ResultsTtl_, *request.mutable_results_ttl());

src/client/query/impl/exec_query.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,7 @@ TFuture<std::pair<TPlainStatus, TExecuteQueryProcessorPtr>> StreamExecuteQueryIm
212212
auto request = MakeRequest<Ydb::Query::ExecuteQueryRequest>();
213213
request.set_exec_mode(::Ydb::Query::ExecMode(settings.ExecMode_));
214214
request.set_stats_mode(::Ydb::Query::StatsMode(settings.StatsMode_));
215+
request.set_pool_id(settings.PoolId_);
215216
request.mutable_query_content()->set_text(TStringType{query});
216217
request.mutable_query_content()->set_syntax(::Ydb::Query::Syntax(settings.Syntax_));
217218
if (session.has_value()) {

0 commit comments

Comments
 (0)