import asyncio
import json

from google import pubsub_v1
from google.api_core import retry

PROJECT_ID = "your-project-id"
SUBSCRIPTION_ID = "your-subscription-id"


async def main():
    async def request_generator():
        yield pubsub_v1.StreamingPullRequest(
            subscription=f"projects/{PROJECT_ID}/subscriptions/{SUBSCRIPTION_ID}",
            max_outstanding_messages=1,
            stream_ack_deadline_seconds=60,
        )

    retry_config = retry.AsyncRetry(
        predicate=retry.if_transient_error,
        initial=1.0,
        maximum=60.0,
        multiplier=2.0,
        timeout=None,
    )

    subscriber = pubsub_v1.SubscriberAsyncClient()
    stream = await subscriber.streaming_pull(requests=request_generator(), retry=retry_config)
    async for response in stream:
        for message in response.received_messages:
            print("Received message: ", json.loads(message.message.data))


if __name__ == "__main__":
    asyncio.run(main())