ddlmZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZddlm Z ddl!m"Z"ddl#m$Z$m%Z%m&Z&Gddee Z'Gddee Z(y))ThreadPoolExecutor)OptionalUnionSet AsyncIterator) ClientOptions) Credentials)StreamingPullFuture)Message)ReassignmentHandlermake_async_subscriber) MultiplexedAsyncSubscriberClient)MultiplexedSubscriberClient)MessageTransformer) NackHandler)SubscriberClientInterfaceAsyncSubscriberClientInterfaceMessageCallback)ConstructableFromServiceAccount)RequireStarted)FlowControlSettings PartitionSubscriptionPathceZdZUdZeed<eed<dddddddddeedee d ee d ee d ee d e d eefdZ ddeee fdededeeedef dZdZdZy)SubscriberClientz A SubscriberClient reads messages similar to Google Pub/Sub. Any subscribe failures are unlikely to succeed if retried. Must be used in a `with` block or have __enter__() called before use. _impl_require_startedN grpc_asyncio)executor nack_handlerreassignment_handlermessage_transformer credentials transportclient_optionsr r!r"r#r$r%r&c x| t}t|fd|_t|_y)a Create a new SubscriberClient. Args: executor: A ThreadPoolExecutor to use. The client will shut it down on __exit__. If provided a single threaded executor, messages will be ordered per-partition, but take care that the callback does not block for too long as it will impede forward progress on all subscriptions. nack_handler: A handler for when `nack()` is called. The default NackHandler raises an exception and fails the subscribe stream. message_transformer: A transformer from Pub/Sub Lite messages to Cloud Pub/Sub messages. This may not return a message with "message_id" set. credentials: If provided, the credentials to use when connecting. transport: The transport to use. Must correspond to an asyncio transport. client_options: The client options to use when connecting. If used, must explicitly set `api_endpoint`. Nc ,t||| SN) subscriptionr%#per_partition_flow_control_settingsr!r"r#fixed_partitionsr$r&r r* partitionssettingsr&r$r#r!r"r%s Hlib/third_party/google/cloud/pubsublite/cloudpubsub/subscriber_client.pyz+SubscriberClient.__init__..X&7L)#4<)%9$7!+'- 8)rrrrr)selfr r!r"r#r$r%r&s ``````r0__init__zSubscriberClient.__init__>s9,  )+H0      !/ 0r3r*callbackr+r,returncr|jj|jj||||SNrrequire_startedr subscribe)r4r*r6r+r,s r0r<zSubscriberClient.subscribefs9 --/zz##   /    r3cn|jj|jj|Sr9)r __enter__rr4s r0r>zSubscriberClient.__enter__us) '')  r3cx|jj||||jj|||yr9)r__exit__rr4exc_type exc_value tracebacks r0rAzSubscriberClient.__exit__zs0 Hi; &&xIFr3r9)__name__ __module__ __qualname____doc__r__annotations__rrrrr rr strrr5rrrrrrr r<r>rAr3r0rr3s %$$$ 26.2>B<@-1'26&1-.&1{+ &1 '':; &1 &&89 &1k*&1&1!/&1Z6:  ,c12  "  .A  #3y>2     Gr3rceZdZUdZeed<eed<ddddddddeedee d ee d ee d e d ee f d Z ddeee fdedeeedeefdZdZdZy)AsyncSubscriberClienta  An AsyncSubscriberClient reads messages similar to Google Pub/Sub, but must be used in an async context. Any subscribe failures are unlikely to succeed if retried. Must be used in an `async with` block or have __aenter__() awaited before use. rrNr)r!r"r#r$r%r&r!r"r#r$r%r&c^tfd|_t|_y)a Create a new AsyncSubscriberClient. Args: nack_handler: A handler for when `nack()` is called. The default NackHandler raises an exception and fails the subscribe stream. message_transformer: A transformer from Pub/Sub Lite messages to Cloud Pub/Sub messages. This may not return a message with "message_id" set. credentials: If provided, the credentials to use when connecting. transport: The transport to use. Must correspond to an asyncio transport. client_options: The client options to use when connecting. If used, must explicitly set `api_endpoint`. c ,t||| Sr)r r-s r0r1z0AsyncSubscriberClient.__init__..r2r3N)rrrr)r4r!r"r#r$r%r&s ``````r0r5zAsyncSubscriberClient.__init__s((6     !/ 0r3r*r+r,r7cK|jj|jj|||d{S7wr9r:)r4r*r+r,s r0r<zAsyncSubscriberClient.subscribesD --/ZZ)) =?O    s;AAAcK|jj|jjd{|S7wr9)rr>r __aenter__r?s r0rSz AsyncSubscriberClient.__aenter__s8 '')jj##%%%  &s8AAAcK|jj|||d{|jj|||y7"wr9)r __aexit__rrArBs r0rUzAsyncSubscriberClient.__aexit__s@jj""8Y BBB &&xIF Cs!AA#Ar9)rFrGrHrIrrJrrrr rr rKrr5rrrrrrr r<rSrUrLr3r0rNrNs *)$$ /3>B<@-1'26!1{+!1'':; !1 &&89 !1 k* !1!1!/!1N6:  ,c12  .A  #3y>2  w    Gr3rNN))concurrent.futures.threadrtypingrrrrgoogle.api_core.client_optionsrgoogle.auth.credentialsr )google.cloud.pubsub_v1.subscriber.futuresr )google.cloud.pubsub_v1.subscriber.messager 8google.cloud.pubsublite.cloudpubsub.reassignment_handlerr rfs9668/I=XWH LIG02QIGXCG"$CCGr3