
    )                     b    S SK r S SKJr  S SKJr  S SKJr  S SKJrJ	r	  S SK
Jr   " S S\5      rg)	    N)Mapping)	Publisher)RoutingPolicy)	PartitionMessageMetadata)PubSubMessagec                   n    \ rS rSr% \\S'   \\\4   \S'   S\S\\\4   4S jr	S r
S rS	\S
\4S jrSrg)RoutingPublisher   _routing_policy_publishersrouting_policy
publishersc                     Xl         X l        g N)r   r   )selfr   r   s      Jlib/third_party/google/cloud/pubsublite/internal/wire/routing_publisher.py__init__RoutingPublisher.__init__   s      .%    c                    #    U R                   R                  5        Vs/ s H  oR                  5       PM     nn[        R                  " U6 I S h  vN   U $ s  snf  N7fr   )r   values
__aenter__asynciogather)r   	publisherenter_futuress      r   r   RoutingPublisher.__aenter__"   s[     484D4D4K4K4M
4My  "4M 	 
 nnm,,,	
 	-s   AAAAAc                    #    U R                   R                  5        H  nUR                  XU5      I S h  vN   M     g  N	7fr   )r   r   	__aexit__)r   exc_typeexc_valexc_tbr   s        r   r    RoutingPublisher.__aexit__)   s7     ))002I%%h@@@ 3@s   3A?
Amessagereturnc                    #    U R                   R                  U5      nX R                  ;   d   eU R                  U   R                  U5      I S h  vN $  N7fr   )r   router   publish)r   r%   	partitions      r   r)   RoutingPublisher.publish-   sO     ((..w7	,,,,,%%i088AAAAs   AAAA)r   r   N)__name__
__module____qualname____firstlineno__r   __annotations__r   r   r   r   r   r    r   r   r)   __static_attributes__ r   r   r
   r
      sX    ""I-..&+&9@IAU9V&AB] B Br   r
   )r   typingr   /google.cloud.pubsublite.internal.wire.publisherr   4google.cloud.pubsublite.internal.wire.routing_policyr   google.cloud.pubsublite.typesr   r   google.cloud.pubsublite_v1r   r
   r2   r   r   <module>r8      s'      E N D 4By Br   