fix: add missing read api retry setting on SplitReadStream (#311)

- [ ] Regenerate this pull request now. Committer: @yirutang PiperOrigin-RevId: 398057257 Source-Link: googleapis/googleapis@376e8e9 Source-Link: googleapis/googleapis-gen@32c05ed Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMzJjMDVlZDRlZDg4ZDgzYWQ1Y2FmYzc1NDQwNThhY2M1OGJmNTYyOCJ9
googleapis · Sep 23, 2021 · 66c09c0 · 66c09c0
1 parent f4fcc4e
commit 66c09c0
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/google/cloud/bigquery_storage_v1/services/big_query_read/async_client.py b/google/cloud/bigquery_storage_v1/services/big_query_read/async_client.py
@@ -441,7 +441,17 @@ async def split_read_stream(
         # and friendly error handling.
         rpc = gapic_v1.method_async.wrap_method(
             self._client._transport.split_read_stream,
-            default_timeout=None,
+            default_retry=retries.Retry(
+                initial=0.1,
+                maximum=60.0,
+                multiplier=1.3,
+                predicate=retries.if_exception_type(
+                    core_exceptions.DeadlineExceeded,
+                    core_exceptions.ServiceUnavailable,
+                ),
+                deadline=600.0,
+            ),
+            default_timeout=600.0,
             client_info=DEFAULT_CLIENT_INFO,
         )
 

diff --git a/google/cloud/bigquery_storage_v1/services/big_query_read/transports/base.py b/google/cloud/bigquery_storage_v1/services/big_query_read/transports/base.py
@@ -189,7 +189,19 @@ def _prep_wrapped_messages(self, client_info):
                 client_info=client_info,
             ),
             self.split_read_stream: gapic_v1.method.wrap_method(
-                self.split_read_stream, default_timeout=None, client_info=client_info,
+                self.split_read_stream,
+                default_retry=retries.Retry(
+                    initial=0.1,
+                    maximum=60.0,
+                    multiplier=1.3,
+                    predicate=retries.if_exception_type(
+                        core_exceptions.DeadlineExceeded,
+                        core_exceptions.ServiceUnavailable,
+                    ),
+                    deadline=600.0,
+                ),
+                default_timeout=600.0,
+                client_info=client_info,
             ),
         }