dZddlmZddlmZddlmZddlmZddlZddlm Z ddl m Z dd l m Z dd l mZdd lmZdd lmZdd lmZddlmZddlmZdZGddej0ZdZGddeZy)z)Utilities for Pub/Sub Lite subscriptions.)absolute_import)division)unicode_literals)futuresN)Optional) cloudpubsub)types) PubsubMessage) lite_util) exceptions) gapic_util)log)queuelceZdZdZy)SubscribeOperationExceptionz2Error when something went wrong while subscribing.N)__name__ __module__ __qualname____doc__7lib/googlecloudsdk/api_lib/pubsub/lite_subscriptions.pyrr&s:rrcRtjtjS)N) credentials)rSubscriberClientr GetGapicCredentialsrrrGetDefaultSubscriberClientr*s!  % %002 44rcDeZdZdZ d dZdZdZdZdZde e fd Z y) rz4GCloud wrapper client for a Pub/Sub Lite subscriber.Nc>|xs t|_tj|_|j ||_|Dchc]}tj|c}|_ tj|t|_ ||_ d|_ycc}w)N)messages_outstandingbytes_outstanding)r_clientrQueue _messages_SubscriptionResourceToPath _subscriptionr Partition _partitionsFlowControlSettings _MAX_INT64_flow_control_settings _auto_ack _pull_future)selfsubscription_resource partitions max_messagesauto_ackclient partitions r__init__zSubscriberClient.__init__2s 979DL[[]DN99:OPDDNOJy 2JOD"'";";)$#DDND PsBc|jj|jj|j|jj |j |j|_|S)N)callback#per_partition_flow_control_settingsfixed_partitions) r" __enter__ subscriber&r$putr+r(r-)r.s rr:zSubscriberClient.__enter__Cs]LL .. ##,0,G,G)) /+D Krc8tjd|jjs5 |jj |jj |jj|||y#t j$rY3wxYw)N) timesleepr-donecancelresultrCancelledErrorr"__exit__)r.exc_type exc_value tracebacks rrEzSubscriberClient.__exit__Ls|JJqM    ! ! #    "   " LL(Iy9 # #   s4BBBctjtj|jtj |j |jS)N)projectlocationname)r SubscriptionPathr ProjectIdToProjectNumber projectsIdLocationToZoneOrRegion locationsIdsubscriptionsId)r.resources rr%z,SubscriberClient._SubscriptionResourceToPathXsF  ! !2283F3FG11(2F2FG  % % ''rc|jjra|jj}|rtdj |t j dtjdy)Nz.Subscribe operation failed with error: {error})errorzNThe streaming pull future completed unexpectedly without raising an exception.z-The subscribe stream terminated unexpectedly.) r-rA exceptionrformatrdebugr InternalError)r.es r_RaiseIfFailedzSubscriberClient._RaiseIfFailed^sw     % % 'a ) < C C! C LN N ii()  $ $ 9 ;; rreturnc|j |jjd}|jr|j |S#t j $rYywxYw)zPulls and optionally acks a message from the provided subscription. Returns: A PubsubMessage pulled from the subscription. r>)timeoutN)r[r$getr,ackrEmpty)r.messages rPullzSubscriberClient.PullisW  ""1"-g  n ;; s9A A"!A")N) rrrrr5r:rEr%r[rr rcrrrrr/s4< " :' ; H]+ rr)r __future__rrr concurrentrr?typingrgoogle.cloud.pubsubliterr google.pubsub_v1r !googlecloudsdk.command_lib.pubsubr googlecloudsdk.corer r r six.movesrr*Errorrrobjectrrrrrns^ 0&' /)*7**#  ;*"2"2;4 GvGr