Added support of part_bytes_limit in QueryService

Author: alex268

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

@@ -200,6 +200,10 @@ GrpcReadStream<YdbQuery.ExecuteQueryResponsePart> createGrpcStream(
             request.setPoolId(resourcePool);
         }
 
+        if (settings.getPartBytesLimit() >= 0) {
+            request.setResponsePartLimitBytes(settings.getPartBytesLimit());
+        }
+
         if (tx != null) {
             request.setTxControl(tx);
         }

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

@@ -13,6 +13,7 @@ public class ExecuteQuerySettings extends BaseRequestSettings {
     private final boolean concurrentResultSets;
     private final String resourcePool;
     private final GrpcFlowControl flowControl;
+    private final long partBytesLimit;
 
     private ExecuteQuerySettings(Builder builder) {
         super(builder);
@@ -21,6 +22,7 @@ private ExecuteQuerySettings(Builder builder) {
         this.concurrentResultSets = builder.concurrentResultSets;
         this.resourcePool = builder.resourcePool;
         this.flowControl = builder.flowControl;
+        this.partBytesLimit = builder.partBytesLimit;
     }
 
     public QueryExecMode getExecMode() {
@@ -47,6 +49,10 @@ public GrpcFlowControl getGrpcFlowControl() {
         return flowControl;
     }
 
+    public long getPartBytesLimit() {
+        return partBytesLimit;
+    }
+
     public static Builder newBuilder() {
         return new Builder();
     }
@@ -57,6 +63,7 @@ public static class Builder extends BaseBuilder<Builder> {
         private boolean concurrentResultSets = false;
         private String resourcePool = null;
         private GrpcFlowControl flowControl = null;
+        private long partBytesLimit = -1;
 
         public Builder withExecMode(QueryExecMode mode) {
             this.execMode = mode;
@@ -92,6 +99,16 @@ public Builder withGrpcFlowControl(GrpcFlowControl ctrl) {
             return this;
         }
 
+        /**
+         * Allows to set size limitation (in bytes) for one result part
+         * @param limit maximum length if one result set part
+         * @return builder
+         */
+        public Builder withPartBytesLimit(long limit) {
+            this.partBytesLimit = limit;
+            return this;
+        }
+
         @Override
         public ExecuteQuerySettings build() {
             return new ExecuteQuerySettings(this);