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

Commit 62591c8

Browse files
feat: add RPC retry information for MacSign, MacVerify, and GenerateRandomBytes (#186)
- [ ] Regenerate this pull request now. PiperOrigin-RevId: 399925901 Source-Link: googleapis/googleapis@ae7e399 Source-Link: https://github.com/googleapis/googleapis-gen/commit/3a8128528ceddba0c1775401eef57ecb582db350 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiM2E4MTI4NTI4Y2VkZGJhMGMxNzc1NDAxZWVmNTdlY2I1ODJkYjM1MCJ9
1 parent 92908af commit 62591c8

File tree

2 files changed

+70
-6
lines changed

2 files changed

+70
-6
lines changed

google/cloud/kms_v1/services/key_management_service/async_client.py

Lines changed: 33 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2628,7 +2628,17 @@ async def mac_sign(
26282628
# and friendly error handling.
26292629
rpc = gapic_v1.method_async.wrap_method(
26302630
self._client._transport.mac_sign,
2631-
default_timeout=None,
2631+
default_retry=retries.Retry(
2632+
initial=0.1,
2633+
maximum=60.0,
2634+
multiplier=1.3,
2635+
predicate=retries.if_exception_type(
2636+
core_exceptions.DeadlineExceeded,
2637+
core_exceptions.ServiceUnavailable,
2638+
),
2639+
deadline=60.0,
2640+
),
2641+
default_timeout=60.0,
26322642
client_info=DEFAULT_CLIENT_INFO,
26332643
)
26342644

@@ -2724,7 +2734,17 @@ async def mac_verify(
27242734
# and friendly error handling.
27252735
rpc = gapic_v1.method_async.wrap_method(
27262736
self._client._transport.mac_verify,
2727-
default_timeout=None,
2737+
default_retry=retries.Retry(
2738+
initial=0.1,
2739+
maximum=60.0,
2740+
multiplier=1.3,
2741+
predicate=retries.if_exception_type(
2742+
core_exceptions.DeadlineExceeded,
2743+
core_exceptions.ServiceUnavailable,
2744+
),
2745+
deadline=60.0,
2746+
),
2747+
default_timeout=60.0,
27282748
client_info=DEFAULT_CLIENT_INFO,
27292749
)
27302750

@@ -2822,7 +2842,17 @@ async def generate_random_bytes(
28222842
# and friendly error handling.
28232843
rpc = gapic_v1.method_async.wrap_method(
28242844
self._client._transport.generate_random_bytes,
2825-
default_timeout=None,
2845+
default_retry=retries.Retry(
2846+
initial=0.1,
2847+
maximum=60.0,
2848+
multiplier=1.3,
2849+
predicate=retries.if_exception_type(
2850+
core_exceptions.DeadlineExceeded,
2851+
core_exceptions.ServiceUnavailable,
2852+
),
2853+
deadline=60.0,
2854+
),
2855+
default_timeout=60.0,
28262856
client_info=DEFAULT_CLIENT_INFO,
28272857
)
28282858

google/cloud/kms_v1/services/key_management_service/transports/base.py

Lines changed: 37 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -484,14 +484,48 @@ def _prep_wrapped_messages(self, client_info):
484484
client_info=client_info,
485485
),
486486
self.mac_sign: gapic_v1.method.wrap_method(
487-
self.mac_sign, default_timeout=None, client_info=client_info,
487+
self.mac_sign,
488+
default_retry=retries.Retry(
489+
initial=0.1,
490+
maximum=60.0,
491+
multiplier=1.3,
492+
predicate=retries.if_exception_type(
493+
core_exceptions.DeadlineExceeded,
494+
core_exceptions.ServiceUnavailable,
495+
),
496+
deadline=60.0,
497+
),
498+
default_timeout=60.0,
499+
client_info=client_info,
488500
),
489501
self.mac_verify: gapic_v1.method.wrap_method(
490-
self.mac_verify, default_timeout=None, client_info=client_info,
502+
self.mac_verify,
503+
default_retry=retries.Retry(
504+
initial=0.1,
505+
maximum=60.0,
506+
multiplier=1.3,
507+
predicate=retries.if_exception_type(
508+
core_exceptions.DeadlineExceeded,
509+
core_exceptions.ServiceUnavailable,
510+
),
511+
deadline=60.0,
512+
),
513+
default_timeout=60.0,
514+
client_info=client_info,
491515
),
492516
self.generate_random_bytes: gapic_v1.method.wrap_method(
493517
self.generate_random_bytes,
494-
default_timeout=None,
518+
default_retry=retries.Retry(
519+
initial=0.1,
520+
maximum=60.0,
521+
multiplier=1.3,
522+
predicate=retries.if_exception_type(
523+
core_exceptions.DeadlineExceeded,
524+
core_exceptions.ServiceUnavailable,
525+
),
526+
deadline=60.0,
527+
),
528+
default_timeout=60.0,
495529
client_info=client_info,
496530
),
497531
}

0 commit comments

Comments
 (0)