File tree Expand file tree Collapse file tree 2 files changed +13
-0
lines changed
query/src/main/java/tech/ydb/query Expand file tree Collapse file tree 2 files changed +13
-0
lines changed Original file line number Diff line number Diff 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 ) {
Original file line number Diff line number Diff line change 99public 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 );
You can’t perform that action at this time.
0 commit comments