%ddlmZmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZddlmZmZddlmZddlmZdd lmZdd lmZmZdd lmZdd lmZdd lm Z m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5cm6cm7cm8cm9Z:ddl;mZ>ddl?m@Z@ddlAmBZBmCZCmDZDmEZEmFZFmGZGddlHmIZIddlJmKZKddlLmMZMdZNd e>d!eOd"e d#ee d$eeeOeOfd%e'f d&ZPd e>d!eOd"e d#ee d$eeeOeOfd'ed(e d)ed%efd*ZQ d0d e>d!eOd+ed(ee d,ee d)eed-eee=d#ee d"ee d.eeeOeOfd%e#fd/ZRy)1)OptionalMappingSet AsyncIteratorCallable)uuid4) ClientOptions) Credentials)ReassignmentHandlerDefaultReassignmentHandler)to_cps_subscribe_message#add_id_to_cps_subscribe_transformer) ClientCache)FlowControlSettings)AckSetTrackerImpl)PartitionSubscriberFactoryAssigningSingleSubscriber)SinglePartitionSingleSubscriber)MessageTransformer) NackHandlerDefaultNackHandler)AsyncSingleSubscriber)regional_endpoint)Assigner) AssignerImpl) CommitterImplFixedSetAssigner)GapicConnectionFactory)merge_metadata)pubsub_contextN)SubscriberResetHandler) PartitionSubscriptionPath)subscription_routing_metadata)SubscribeRequestInitialSubscribeRequestStreamingCommitCursorRequestPartitionAssignmentRequest!InitialPartitionAssignmentRequestInitialCommitCursorRequestSubscriberServiceAsyncClient)%PartitionAssignmentServiceAsyncClientCursorServiceAsyncClientg? subscription transportclient_options credentials base_metadatareturncidttffd }ttt |t j t|S)Nrequestsctt}|j|tjS)Nr4r2r3metadata)r.assign_partitionslistitems)r8assignment_clientr5r3r4r2s Olib/third_party/google/cloud/pubsublite/cloudpubsub/internal/make_subscriber.pyassignment_connection_factoryz=_make_dynamic_assigner..assignment_connection_factorybsIBken@NO 22 tM$7$7$9:3  )r1 client_id)rr)rr*strrbytesr)r1r2r3r4r5rBs ```` rA_make_dynamic_assignerrGXsZ   :;   )\*egmm  <=  rCflow_control_settings nack_handlermessage_transformerc tfd tfd dtdtf  fd }|S)Nc tSNr:r,r3r4r2srAz4_make_partition_subscriber_factory..}s,#y rCc tSrMr/rNsrArOz4_make_partition_subscriber_factory..s(#y rC partitionr6c Ztt dttf fd dttffd }dt f fd }t tt jtt|}t|}t| | t S)Nr8ctjj|tjSNr;)get subscriber>r?)r8final_metadatasubscribe_client_caches rAsubscribe_connection_factoryzY_make_partition_subscriber_factory..factory..subscribe_connection_factorys7)--/994(<(<(>#?: rCctjj|tjSrT)rUstreaming_commit_cursorr>r?)r8cursor_client_cacherWs rAcursor_connection_factoryzV_make_partition_subscriber_factory..factory..cursor_connection_factorys:'**,DD4(<(<(>#?E rC reset_handlerctjttjt t |S)Nr1rQ)wire_subscriberSubscriberImplr'rEvalue_DEFAULT_FLUSH_SECONDSr)r^rQrYr1s rAsubscriber_factoryzO_make_partition_subscriber_factory..factory..subscriber_factorys="11'!$\!2ioo'&'CD  rCr`)r r%rr&r(r"rr+rErcrdrrrr)rQr]re committerack_set_trackerrWrYr5r\rHrJrIrXr1s` @@rAfactoryz3_make_partition_subscriber_factory..factorys' 8yQ  =AQ3R   #$@A  .D " & .)//  # "#< =  ,I6.  !   / ;N O   rC)rr#r) r1r2r3r4r5rHrIrJrhr\rXs ```````` @@rA"_make_partition_subscriber_factoryrirsI)  &  ) 9) )>) ) V NrC#per_partition_flow_control_settingsreassignment_handlerfixed_partitionsr<c V ttd  )ttjj rfd} n  fd} | t }| t}|tjt}t |||} t| | |S)a Make a Pub/Sub Lite AsyncSubscriber. Args: subscription: The subscription to subscribe to. transport: The transport type to use. per_partition_flow_control_settings: The flow control settings for each partition subscribed to. Note that these settings apply to each partition individually, not in aggregate. nack_handler: An optional handler for when nack() is called on a Message. The default will fail the client. message_transformer: An optional transformer from Pub/Sub Lite messages to Cloud Pub/Sub messages. fixed_partitions: A fixed set of partitions to subscribe to. If not present, will instead use auto-assignment. credentials: The credentials to use to connect. GOOGLE_DEFAULT_CREDENTIALS is used if None. client_options: Other options to pass to the client. Note that if you pass any you must set api_endpoint. metadata: Additional metadata to send with the RPC. Returns: A new AsyncSubscriber. CLOUD_PUBSUB_SHIM) framework) api_endpointctSNr)rlsrArOz'make_async_subscriber..s #34D#ErCc"tSrr)rG)r3r4r<r1r2srArOz'make_async_subscriber..s#9      $ rC) r r!r rlocationregionrr r of_callabler rir) r1r2rjrIrkrJrlr4r3r<assigner_factorypartition_subscriber_factorys `` ```` rAmake_async_subscriberrys<n7JKXVH&*<+@+@+G+GH E )+ #9;"0<<=UV#E+ $  %68L rC)NNNNNNN)Stypingrrrrruuidrgoogle.api_core.client_optionsr google.auth.credentialsr 8google.cloud.pubsublite.cloudpubsub.reassignment_handlerr r 6google.cloud.pubsublite.cloudpubsub.message_transformsr r2google.cloud.pubsublite.internal.wire.client_cachergoogle.cloud.pubsublite.typesrAgoogle.cloud.pubsublite.cloudpubsub.internal.ack_set_tracker_implrAgoogle.cloud.pubsublite.cloudpubsub.internal.assigning_subscriberrrHgoogle.cloud.pubsublite.cloudpubsub.internal.single_partition_subscriberr7google.cloud.pubsublite.cloudpubsub.message_transformerr0google.cloud.pubsublite.cloudpubsub.nack_handlerrr>google.cloud.pubsublite.cloudpubsub.internal.single_subscriberr*google.cloud.pubsublite.internal.endpointsr.google.cloud.pubsublite.internal.wire.assignerr3google.cloud.pubsublite.internal.wire.assigner_implr4google.cloud.pubsublite.internal.wire.committer_implr8google.cloud.pubsublite.internal.wire.fixed_set_assignerr6google.cloud.pubsublite.internal.wire.gapic_connectionr4google.cloud.pubsublite.internal.wire.merge_metadatar 4google.cloud.pubsublite.internal.wire.pubsub_contextr!5google.cloud.pubsublite.internal.wire.subscriber_implcloud pubsubliteinternalwiresubscriber_implra>google.cloud.pubsublite.internal.wire.subscriber_reset_handlerr"r#r$1google.cloud.pubsublite.internal.routing_metadatar%google.cloud.pubsublite_v1r&r'r(r)r*r+Cgoogle.cloud.pubsublite_v1.services.subscriber_service.async_clientr-Mgoogle.cloud.pubsublite_v1.services.partition_assignment_service.async_clientr.?google.cloud.pubsublite_v1.services.cursor_service.async_clientr0rdrErGriryrCrArs0CB8/K=WICLNUPOOOOF""+&  GCH-.   4@"@@"@+& @ GCH-. @ / @@,@ @N+/:>8<15)-.2,0A"AA*=A;' A ##67 A ""45 As9~.A+&A]+AwsCx()AArC