'ddlmZddlZddlmZddlmZddlmZddlmZejrddl m Z Gdd ejZ Gd d ejZ y) )absolute_importN)Any)Union)futures)AcknowledgeStatus)StreamingPullManagercLeZdZdZd fd ZdddefdZdefdZdefd Z xZ S) StreamingPullFuturea<Represents a process that asynchronously performs streaming pull and schedules messages to be processed. This future is resolved when the process is stopped (via :meth:`cancel`) or if it encounters an unrecoverable error. Calling `.result()` will cause the calling thread to block indefinitely. managerrctt| ||_|jj |j d|_y)NF)superr __init___StreamingPullFuture__manageradd_close_callback_on_close_callback_StreamingPullFuture__cancelled)selfr __class__s OOD !   v &rreturncFd|_|jjy)aStops pulling messages and shutdowns the background thread consuming messages. The method always returns ``True``, as the shutdown is always initiated. However, if the background stream is already being shut down or the shutdown has completed, this method is a no-op. .. versionchanged:: 2.4.1 The method does not block anymore, it just triggers the shutdown and returns immediately. To block until the background stream is terminated, call :meth:`result()` after cancelling the future. .. versionchanged:: 2.10.0 The method always returns ``True`` instead of ``None``. T)rrclosers rcancelzStreamingPullFuture.cancel8s$  rc|jS)zW Returns: ``True`` if the subscription has been cancelled. )rrs r cancelledzStreamingPullFuture.cancelledNs r)r r) __name__ __module__ __qualname____doc__rrrboolr r" __classcell__rs@rr r s9! '*@ '# ', 4 rr cPeZdZdZdefdZdefdZddeee fde ffd Z xZ S)FuturezThis future object is for subscribe-side calls. Calling :meth:`result` will resolve the future by returning the message ID, unless an error occurs. rcyziActions in Pub/Sub generally may not be canceled. This method always returns ``False``. Frs rr z Future.cancel] rcyr-r.rs rr"zFuture.cancelleddr/rtimeoutc$t||S)aDReturn a success code or raise an exception. This blocks until the operation completes successfully and returns the error code unless an exception is raised. Args: timeout: The number of seconds before this call times out and raises TimeoutError. Returns: AcknowledgeStatus.SUCCESS if the operation succeeded. Raises: concurrent.futures.TimeoutError: If the request times out. AcknowledgeError: If the operation did not succeed for another reason. )r1)r r)rr1rs rrz Future.resultks$w~g~..r)N) r#r$r%r&r'r r"rintfloatrrr(r)s@rr+r+VsB 4/eCJ//;L//rr+) __future__rtypingrrgoogle.cloud.pubsub_v1r,google.cloud.pubsub_v1.subscriber.exceptionsr TYPE_CHECKINGBgoogle.cloud.pubsub_v1.subscriber._protocol.streaming_pull_managerrr+r r.rrr;sH' *J  5 '..5 p'/W^^'/r