feat: add ResourceExhausted to retryable error for Write API unary ca…

…lls (#2178) * feat: add ResourceExhausted to retryable error for Write API unary calls docs: add multiplexing documentation PiperOrigin-RevId: 545839491 Source-Link: googleapis/googleapis@2b006af Source-Link: googleapis/googleapis-gen@0d52d38 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMGQ1MmQzODViZDRlNzhjN2IyYzgzNzU1MDEzZmUxMDNlODA0YzM4NCJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
googleapis · Jul 17, 2023 · d9b526a · d9b526a
1 parent afc550a
commit d9b526a
Show file tree

Hide file tree

Showing 13 changed files with 418 additions and 171 deletions.
diff --git a/...ge/src/main/java/com/google/cloud/bigquery/storage/v1/stub/BigQueryWriteStubSettings.java b/...ge/src/main/java/com/google/cloud/bigquery/storage/v1/stub/BigQueryWriteStubSettings.java
@@ -259,7 +259,9 @@ public static class Builder extends StubSettings.Builder<BigQueryWriteStubSettin
           "retry_policy_4_codes",
           ImmutableSet.copyOf(
               Lists.<StatusCode.Code>newArrayList(
-                  StatusCode.Code.DEADLINE_EXCEEDED, StatusCode.Code.UNAVAILABLE)));
+                  StatusCode.Code.DEADLINE_EXCEEDED,
+                  StatusCode.Code.UNAVAILABLE,
+                  StatusCode.Code.RESOURCE_EXHAUSTED)));
       RETRYABLE_CODE_DEFINITIONS = definitions.build();
     }
 

diff --git a/...uerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/MockBigQueryWriteImpl.java b/...uerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/MockBigQueryWriteImpl.java
@@ -174,8 +174,7 @@ public void batchCommitWriteStreams(
       responseObserver.onError(
           new IllegalArgumentException(
               String.format(
-                  "Unrecognized response type %s for method BatchCommitWriteStreams, expected %s or"
-                      + " %s",
+                  "Unrecognized response type %s for method BatchCommitWriteStreams, expected %s or %s",
                   response == null ? "null" : response.getClass().getName(),
                   BatchCommitWriteStreamsResponse.class.getName(),
                   Exception.class.getName())));

diff --git a/...rage/src/test/java/com/google/cloud/bigquery/storage/v1beta1/MockBigQueryStorageImpl.java b/...rage/src/test/java/com/google/cloud/bigquery/storage/v1beta1/MockBigQueryStorageImpl.java
@@ -117,8 +117,7 @@ public void batchCreateReadSessionStreams(
       responseObserver.onError(
           new IllegalArgumentException(
               String.format(
-                  "Unrecognized response type %s for method BatchCreateReadSessionStreams, expected"
-                      + " %s or %s",
+                  "Unrecognized response type %s for method BatchCreateReadSessionStreams, expected %s or %s",
                   response == null ? "null" : response.getClass().getName(),
                   Storage.BatchCreateReadSessionStreamsResponse.class.getName(),
                   Exception.class.getName())));

diff --git a/...torage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/MockBigQueryWriteImpl.java b/...torage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/MockBigQueryWriteImpl.java
@@ -174,8 +174,7 @@ public void batchCommitWriteStreams(
       responseObserver.onError(
           new IllegalArgumentException(
               String.format(
-                  "Unrecognized response type %s for method BatchCommitWriteStreams, expected %s or"
-                      + " %s",
+                  "Unrecognized response type %s for method BatchCommitWriteStreams, expected %s or %s",
                   response == null ? "null" : response.getClass().getName(),
                   BatchCommitWriteStreamsResponse.class.getName(),
                   Exception.class.getName())));