Skip to content
This repository has been archived by the owner on Dec 31, 2023. It is now read-only.

feat: add RPC retry information for MacSign, MacVerify, and GenerateRandomBytes #186

Merged
merged 3 commits into from Sep 30, 2021
Merged

feat: add RPC retry information for MacSign, MacVerify, and GenerateRandomBytes #186

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
65099cc
feat: add RPC retry information for MacSign, MacVerify, and GenerateR…
gcf-owl-bot[bot] Sep 30, 2021
2d98a94
🦉 Updates from OwlBot
gcf-owl-bot[bot] Sep 30, 2021
259cf20
Merge branch 'main' into owl-bot-d67d7bdd-ff68-44ef-bfaf-3cea57027d66
busunkim96 Sep 30, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
36 changes: 33 additions & 3 deletions google/cloud/kms_v1/services/key_management_service/async_client.py
View file
Open in desktop
Expand Up @@ -2628,7 +2628,17 @@ async def mac_sign(
# and friendly error handling.
rpc = gapic_v1.method_async.wrap_method(
self._client._transport.mac_sign,
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=60.0,
),
default_timeout=60.0,
client_info=DEFAULT_CLIENT_INFO,
)

Expand Down Expand Up @@ -2724,7 +2734,17 @@ async def mac_verify(
# and friendly error handling.
rpc = gapic_v1.method_async.wrap_method(
self._client._transport.mac_verify,
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=60.0,
),
default_timeout=60.0,
client_info=DEFAULT_CLIENT_INFO,
)

Expand Down Expand Up @@ -2822,7 +2842,17 @@ async def generate_random_bytes(
# and friendly error handling.
rpc = gapic_v1.method_async.wrap_method(
self._client._transport.generate_random_bytes,
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=60.0,
),
default_timeout=60.0,
client_info=DEFAULT_CLIENT_INFO,
)

Expand Down
40 changes: 37 additions & 3 deletions google/cloud/kms_v1/services/key_management_service/transports/base.py
View file
Open in desktop
Expand Up @@ -484,14 +484,48 @@ def _prep_wrapped_messages(self, client_info):
client_info=client_info,
),
self.mac_sign: gapic_v1.method.wrap_method(
self.mac_sign, default_timeout=None, client_info=client_info,
self.mac_sign,
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=60.0,
),
default_timeout=60.0,
client_info=client_info,
),
self.mac_verify: gapic_v1.method.wrap_method(
self.mac_verify, default_timeout=None, client_info=client_info,
self.mac_verify,
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=60.0,
),
default_timeout=60.0,
client_info=client_info,
),
self.generate_random_bytes: gapic_v1.method.wrap_method(
self.generate_random_bytes,
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=60.0,
),
default_timeout=60.0,
client_info=client_info,
),
}
Expand Down