google_api_core_retry_with_exponential_backoff_quickstart.py

1import datetime
2import random
3
4from google.api_core import retry
5
6
7# This function simulates an API call that fails occasionally.
8def flaky_function():
9    if random.random() < 0.75:
10        print(f"{datetime.datetime.now()}: Operation failed, retrying...")
11        raise RuntimeError("Transient error")
12    print(f"{datetime.datetime.now()}: Operation succeeded!")
13    return "Success"
14
15
16# Create a retry object that will retry on RuntimeError.
17# By default, this uses exponential backoff.
18retry_strategy = retry.Retry(predicate=retry.if_exception_type(RuntimeError))
19
20# Use the retry strategy to call the flaky function.
21result = retry_strategy(flaky_function)()
22
23print(f"Final Result: {result}")