Skip to content

Commit 5e090f4

Browse files
committed
Add support pool_id in query requests
1 parent c831264 commit 5e090f4

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

query/src/main/java/tech/ydb/query/impl/SessionImpl.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -194,6 +194,7 @@ GrpcReadStream<YdbQuery.ExecuteQueryResponsePart> createGrpcStream(
194194
.setText(query)
195195
.build()
196196
)
197+
.setPoolId(settings.getResourcePool())
197198
.putAllParameters(prms.toPb());
198199

199200
if (tx != null) {

query/src/main/java/tech/ydb/query/settings/ExecuteQuerySettings.java

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,13 @@
99
public class ExecuteQuerySettings extends BaseRequestSettings {
1010
private final QueryExecMode execMode;
1111
private final QueryStatsMode statsMode;
12+
private final String resourcePool;
1213

1314
private ExecuteQuerySettings(Builder builder) {
1415
super(builder);
1516
this.execMode = builder.execMode;
1617
this.statsMode = builder.statsMode;
18+
this.resourcePool = builder.resourcePool;
1719
}
1820

1921
public QueryExecMode getExecMode() {
@@ -24,13 +26,18 @@ public QueryStatsMode getStatsMode() {
2426
return this.statsMode;
2527
}
2628

29+
public String getResourcePool() {
30+
return this.resourcePool;
31+
}
32+
2733
public static Builder newBuilder() {
2834
return new Builder();
2935
}
3036

3137
public static class Builder extends BaseBuilder<Builder> {
3238
private QueryExecMode execMode = QueryExecMode.EXECUTE;
3339
private QueryStatsMode statsMode = QueryStatsMode.NONE;
40+
private String resourcePool = "default";
3441

3542
public Builder withExecMode(QueryExecMode mode) {
3643
this.execMode = mode;
@@ -42,6 +49,11 @@ public Builder withStatsMode(QueryStatsMode mode) {
4249
return this;
4350
}
4451

52+
public Builder withResourcePool(String poolId) {
53+
this.resourcePool = poolId;
54+
return this;
55+
}
56+
4557
@Override
4658
public ExecuteQuerySettings build() {
4759
return new ExecuteQuerySettings(this);

0 commit comments

Comments
 (0)