feat: Adding new fields for Serverless analytics

google-cloud-spanner/src/main/java/com/google/cloud/spanner/v1/SpannerClient.java

@@ -940,6 +940,7 @@ public final UnaryCallable<DeleteSessionRequest, Empty> deleteSessionCallable()
    *           .setSeqno(109325920)
    *           .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
    *           .setRequestOptions(RequestOptions.newBuilder().build())
+   *           .setDataBoostEnabled(true)
    *           .build();
    *   ResultSet response = spannerClient.executeSql(request);
    * }
@@ -987,6 +988,7 @@ public final ResultSet executeSql(ExecuteSqlRequest request) {
    *           .setSeqno(109325920)
    *           .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
    *           .setRequestOptions(RequestOptions.newBuilder().build())
+   *           .setDataBoostEnabled(true)
    *           .build();
    *   ApiFuture<ResultSet> future = spannerClient.executeSqlCallable().futureCall(request);
    *   // Do something.
@@ -1027,6 +1029,7 @@ public final UnaryCallable<ExecuteSqlRequest, ResultSet> executeSqlCallable() {
    *           .setSeqno(109325920)
    *           .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
    *           .setRequestOptions(RequestOptions.newBuilder().build())
+   *           .setDataBoostEnabled(true)
    *           .build();
    *   ServerStream<PartialResultSet> stream =
    *       spannerClient.executeStreamingSqlCallable().call(request);
@@ -1162,6 +1165,7 @@ public final ExecuteBatchDmlResponse executeBatchDml(ExecuteBatchDmlRequest requ
    *           .setResumeToken(ByteString.EMPTY)
    *           .setPartitionToken(ByteString.EMPTY)
    *           .setRequestOptions(RequestOptions.newBuilder().build())
+   *           .setDataBoostEnabled(true)
    *           .build();
    *   ResultSet response = spannerClient.read(request);
    * }
@@ -1210,6 +1214,7 @@ public final ResultSet read(ReadRequest request) {
    *           .setResumeToken(ByteString.EMPTY)
    *           .setPartitionToken(ByteString.EMPTY)
    *           .setRequestOptions(RequestOptions.newBuilder().build())
+   *           .setDataBoostEnabled(true)
    *           .build();
    *   ApiFuture<ResultSet> future = spannerClient.readCallable().futureCall(request);
    *   // Do something.
@@ -1250,6 +1255,7 @@ public final UnaryCallable<ReadRequest, ResultSet> readCallable() {
    *           .setResumeToken(ByteString.EMPTY)
    *           .setPartitionToken(ByteString.EMPTY)
    *           .setRequestOptions(RequestOptions.newBuilder().build())
+   *           .setDataBoostEnabled(true)
    *           .build();
    *   ServerStream<PartialResultSet> stream = spannerClient.streamingReadCallable().call(request);
    *   for (PartialResultSet response : stream) {

google-cloud-spanner/src/test/java/com/google/cloud/spanner/v1/SpannerClientHttpJsonTest.java

@@ -598,6 +598,7 @@ public void executeSqlTest() throws Exception {
             .setSeqno(109325920)
             .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
             .setRequestOptions(RequestOptions.newBuilder().build())
+            .setDataBoostEnabled(true)
             .build();
 
     ResultSet actualResponse = client.executeSql(request);
@@ -639,6 +640,7 @@ public void executeSqlExceptionTest() throws Exception {
               .setSeqno(109325920)
               .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
               .setRequestOptions(RequestOptions.newBuilder().build())
+              .setDataBoostEnabled(true)
               .build();
       client.executeSql(request);
       Assert.fail("No exception raised");
@@ -742,6 +744,7 @@ public void readTest() throws Exception {
             .setResumeToken(ByteString.EMPTY)
             .setPartitionToken(ByteString.EMPTY)
             .setRequestOptions(RequestOptions.newBuilder().build())
+            .setDataBoostEnabled(true)
             .build();
 
     ResultSet actualResponse = client.read(request);
@@ -783,6 +786,7 @@ public void readExceptionTest() throws Exception {
               .setResumeToken(ByteString.EMPTY)
               .setPartitionToken(ByteString.EMPTY)
               .setRequestOptions(RequestOptions.newBuilder().build())
+              .setDataBoostEnabled(true)
               .build();
       client.read(request);
       Assert.fail("No exception raised");

google-cloud-spanner/src/test/java/com/google/cloud/spanner/v1/SpannerClientTest.java

@@ -554,6 +554,7 @@ public void executeSqlTest() throws Exception {
             .setSeqno(109325920)
             .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
             .setRequestOptions(RequestOptions.newBuilder().build())
+            .setDataBoostEnabled(true)
             .build();
 
     ResultSet actualResponse = client.executeSql(request);
@@ -574,6 +575,7 @@ public void executeSqlTest() throws Exception {
     Assert.assertEquals(request.getSeqno(), actualRequest.getSeqno());
     Assert.assertEquals(request.getQueryOptions(), actualRequest.getQueryOptions());
     Assert.assertEquals(request.getRequestOptions(), actualRequest.getRequestOptions());
+    Assert.assertEquals(request.getDataBoostEnabled(), actualRequest.getDataBoostEnabled());
     Assert.assertTrue(
         channelProvider.isHeaderSent(
             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
@@ -599,6 +601,7 @@ public void executeSqlExceptionTest() throws Exception {
               .setSeqno(109325920)
               .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
               .setRequestOptions(RequestOptions.newBuilder().build())
+              .setDataBoostEnabled(true)
               .build();
       client.executeSql(request);
       Assert.fail("No exception raised");
@@ -631,6 +634,7 @@ public void executeStreamingSqlTest() throws Exception {
             .setSeqno(109325920)
             .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
             .setRequestOptions(RequestOptions.newBuilder().build())
+            .setDataBoostEnabled(true)
             .build();
 
     MockStreamObserver<PartialResultSet> responseObserver = new MockStreamObserver<>();
@@ -661,6 +665,7 @@ public void executeStreamingSqlExceptionTest() throws Exception {
             .setSeqno(109325920)
             .setQueryOptions(ExecuteSqlRequest.QueryOptions.newBuilder().build())
             .setRequestOptions(RequestOptions.newBuilder().build())
+            .setDataBoostEnabled(true)
             .build();
 
     MockStreamObserver<PartialResultSet> responseObserver = new MockStreamObserver<>();
@@ -761,6 +766,7 @@ public void readTest() throws Exception {
             .setResumeToken(ByteString.EMPTY)
             .setPartitionToken(ByteString.EMPTY)
             .setRequestOptions(RequestOptions.newBuilder().build())
+            .setDataBoostEnabled(true)
             .build();
 
     ResultSet actualResponse = client.read(request);
@@ -780,6 +786,7 @@ public void readTest() throws Exception {
     Assert.assertEquals(request.getResumeToken(), actualRequest.getResumeToken());
     Assert.assertEquals(request.getPartitionToken(), actualRequest.getPartitionToken());
     Assert.assertEquals(request.getRequestOptions(), actualRequest.getRequestOptions());
+    Assert.assertEquals(request.getDataBoostEnabled(), actualRequest.getDataBoostEnabled());
     Assert.assertTrue(
         channelProvider.isHeaderSent(
             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
@@ -805,6 +812,7 @@ public void readExceptionTest() throws Exception {
               .setResumeToken(ByteString.EMPTY)
               .setPartitionToken(ByteString.EMPTY)
               .setRequestOptions(RequestOptions.newBuilder().build())
+              .setDataBoostEnabled(true)
               .build();
       client.read(request);
       Assert.fail("No exception raised");
@@ -837,6 +845,7 @@ public void streamingReadTest() throws Exception {
             .setResumeToken(ByteString.EMPTY)
             .setPartitionToken(ByteString.EMPTY)
             .setRequestOptions(RequestOptions.newBuilder().build())
+            .setDataBoostEnabled(true)
             .build();
 
     MockStreamObserver<PartialResultSet> responseObserver = new MockStreamObserver<>();
@@ -867,6 +876,7 @@ public void streamingReadExceptionTest() throws Exception {
             .setResumeToken(ByteString.EMPTY)
             .setPartitionToken(ByteString.EMPTY)
             .setRequestOptions(RequestOptions.newBuilder().build())
+            .setDataBoostEnabled(true)
             .build();
 
     MockStreamObserver<PartialResultSet> responseObserver = new MockStreamObserver<>();

proto-google-cloud-spanner-v1/src/main/java/com/google/spanner/v1/ExecuteSqlRequest.java

@@ -1972,6 +1972,27 @@ public com.google.spanner.v1.RequestOptionsOrBuilder getRequestOptionsOrBuilder(
         : requestOptions_;
   }
 
+  public static final int DATA_BOOST_ENABLED_FIELD_NUMBER = 15;
+  private boolean dataBoostEnabled_ = false;
+  /**
+   *
+   *
+   * <pre>
+   * If this is for a partitioned read and this field is set to `true`, the
+   * request will be executed via Spanner independent compute resources.
+   * If the field is set to `true` but the request does not set
+   * `partition_token`, the API will return an `INVALID_ARGUMENT` error.
+   * </pre>
+   *
+   * <code>bool data_boost_enabled = 15;</code>
+   *
+   * @return The dataBoostEnabled.
+   */
+  @java.lang.Override
+  public boolean getDataBoostEnabled() {
+    return dataBoostEnabled_;
+  }
+
   private byte memoizedIsInitialized = -1;
 
   @java.lang.Override
@@ -2018,6 +2039,9 @@ public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io
     if (requestOptions_ != null) {
       output.writeMessage(11, getRequestOptions());
     }
+    if (dataBoostEnabled_ != false) {
+      output.writeBool(15, dataBoostEnabled_);
+    }
     getUnknownFields().writeTo(output);
   }
 
@@ -2067,6 +2091,9 @@ public int getSerializedSize() {
     if (requestOptions_ != null) {
       size += com.google.protobuf.CodedOutputStream.computeMessageSize(11, getRequestOptions());
     }
+    if (dataBoostEnabled_ != false) {
+      size += com.google.protobuf.CodedOutputStream.computeBoolSize(15, dataBoostEnabled_);
+    }
     size += getUnknownFields().getSerializedSize();
     memoizedSize = size;
     return size;
@@ -2105,6 +2132,7 @@ public boolean equals(final java.lang.Object obj) {
     if (hasRequestOptions()) {
       if (!getRequestOptions().equals(other.getRequestOptions())) return false;
     }
+    if (getDataBoostEnabled() != other.getDataBoostEnabled()) return false;
     if (!getUnknownFields().equals(other.getUnknownFields())) return false;
     return true;
   }
@@ -2148,6 +2176,8 @@ public int hashCode() {
       hash = (37 * hash) + REQUEST_OPTIONS_FIELD_NUMBER;
       hash = (53 * hash) + getRequestOptions().hashCode();
     }
+    hash = (37 * hash) + DATA_BOOST_ENABLED_FIELD_NUMBER;
+    hash = (53 * hash) + com.google.protobuf.Internal.hashBoolean(getDataBoostEnabled());
     hash = (29 * hash) + getUnknownFields().hashCode();
     memoizedHashCode = hash;
     return hash;
@@ -2335,6 +2365,7 @@ public Builder clear() {
         requestOptionsBuilder_.dispose();
         requestOptionsBuilder_ = null;
       }
+      dataBoostEnabled_ = false;
       return this;
     }
 
@@ -2408,6 +2439,9 @@ private void buildPartial0(com.google.spanner.v1.ExecuteSqlRequest result) {
         result.requestOptions_ =
             requestOptionsBuilder_ == null ? requestOptions_ : requestOptionsBuilder_.build();
       }
+      if (((from_bitField0_ & 0x00000800) != 0)) {
+        result.dataBoostEnabled_ = dataBoostEnabled_;
+      }
     }
 
     @java.lang.Override
@@ -2491,6 +2525,9 @@ public Builder mergeFrom(com.google.spanner.v1.ExecuteSqlRequest other) {
       if (other.hasRequestOptions()) {
         mergeRequestOptions(other.getRequestOptions());
       }
+      if (other.getDataBoostEnabled() != false) {
+        setDataBoostEnabled(other.getDataBoostEnabled());
+      }
       this.mergeUnknownFields(other.getUnknownFields());
       onChanged();
       return this;
@@ -2590,6 +2627,12 @@ public Builder mergeFrom(
                 bitField0_ |= 0x00000400;
                 break;
               } // case 90
+            case 120:
+              {
+                dataBoostEnabled_ = input.readBool();
+                bitField0_ |= 0x00000800;
+                break;
+              } // case 120
             default:
               {
                 if (!super.parseUnknownField(input, extensionRegistry, tag)) {
@@ -4210,6 +4253,68 @@ public com.google.spanner.v1.RequestOptionsOrBuilder getRequestOptionsOrBuilder(
       return requestOptionsBuilder_;
     }
 
+    private boolean dataBoostEnabled_;
+    /**
+     *
+     *
+     * <pre>
+     * If this is for a partitioned read and this field is set to `true`, the
+     * request will be executed via Spanner independent compute resources.
+     * If the field is set to `true` but the request does not set
+     * `partition_token`, the API will return an `INVALID_ARGUMENT` error.
+     * </pre>
+     *
+     * <code>bool data_boost_enabled = 15;</code>
+     *
+     * @return The dataBoostEnabled.
+     */
+    @java.lang.Override
+    public boolean getDataBoostEnabled() {
+      return dataBoostEnabled_;
+    }
+    /**
+     *
+     *
+     * <pre>
+     * If this is for a partitioned read and this field is set to `true`, the
+     * request will be executed via Spanner independent compute resources.
+     * If the field is set to `true` but the request does not set
+     * `partition_token`, the API will return an `INVALID_ARGUMENT` error.
+     * </pre>
+     *
+     * <code>bool data_boost_enabled = 15;</code>
+     *
+     * @param value The dataBoostEnabled to set.
+     * @return This builder for chaining.
+     */
+    public Builder setDataBoostEnabled(boolean value) {
+
+      dataBoostEnabled_ = value;
+      bitField0_ |= 0x00000800;
+      onChanged();
+      return this;
+    }
+    /**
+     *
+     *
+     * <pre>
+     * If this is for a partitioned read and this field is set to `true`, the
+     * request will be executed via Spanner independent compute resources.
+     * If the field is set to `true` but the request does not set
+     * `partition_token`, the API will return an `INVALID_ARGUMENT` error.
+     * </pre>
+     *
+     * <code>bool data_boost_enabled = 15;</code>
+     *
+     * @return This builder for chaining.
+     */
+    public Builder clearDataBoostEnabled() {
+      bitField0_ = (bitField0_ & ~0x00000800);
+      dataBoostEnabled_ = false;
+      onChanged();
+      return this;
+    }
+
     @java.lang.Override
     public final Builder setUnknownFields(final com.google.protobuf.UnknownFieldSet unknownFields) {
       return super.setUnknownFields(unknownFields);

proto-google-cloud-spanner-v1/src/main/java/com/google/spanner/v1/ExecuteSqlRequestOrBuilder.java

@@ -429,4 +429,20 @@ com.google.spanner.v1.Type getParamTypesOrDefault(
    * <code>.google.spanner.v1.RequestOptions request_options = 11;</code>
    */
   com.google.spanner.v1.RequestOptionsOrBuilder getRequestOptionsOrBuilder();
+
+  /**
+   *
+   *
+   * <pre>
+   * If this is for a partitioned read and this field is set to `true`, the
+   * request will be executed via Spanner independent compute resources.
+   * If the field is set to `true` but the request does not set
+   * `partition_token`, the API will return an `INVALID_ARGUMENT` error.
+   * </pre>
+   *
+   * <code>bool data_boost_enabled = 15;</code>
+   *
+   * @return The dataBoostEnabled.
+   */
+  boolean getDataBoostEnabled();
 }