sddlZddlmZmZmZmZddlmZmZddl m Z edZ de defdZ Gd d ZGd d Zy) N) AwaitableTypeVarOptionalCallable)GoogleAPICallErrorUnknown)wait_ignore_errorsTereturncBt|tr|Std|gS)NzHad an unknown error)errors) isinstancerr)r s Klib/third_party/google/cloud/pubsublite/internal/wire/permanent_failable.py adapt_errorrs"!'( )1# 66c$eZdZdefdZdZdZy)_TaskWithCleanupac8tj||_yN)asyncio ensure_future_task)selfrs r__init__z_TaskWithCleanup.__init__ s**1- rc"K|jSwr)rrs r __aenter__z_TaskWithCleanup.__aenter__#szzs cK|jjs8|jjt|jd{yy7wr)rdonecancelr )rexc_typeexc_valexc_tbs r __aexit__z_TaskWithCleanup.__aexit__&s>zz JJ   $TZZ0 0 0! 0sA AAAN)__name__ __module__ __qualname__rrrr&rrrrs.).1rrceZdZUdZeej ed<dZe dej fdZ de e de fdZ dege d ffd Zd efd Zdeefd Zy )PermanentFailablezdA class that can experience permanent failures, with helpers for forwarding these to client actions._maybe_failure_taskcd|_yr)r-rs rrzPermanentFailable.__init__1s #' rr cd|jtj|_|jS)zaGet the failure task, initializing it lazily, since it needs to be initialized in the event loop.)r-rFuturers r _failure_taskzPermanentFailable._failure_task4s+  # # +'.~~'7D $'''r awaitablecKt|4d{}|jjr|jjt j ||jgtj d{\}}||vr|d{cdddd{S|jj77B737%#1d{7swYyxYww)z Await the awaitable, unless fail() is called first. Args: awaitable: An awaitable Returns: The result of the awaitable Raises: The permanent error if fail() is called or the awaitable raises one. N) return_when)rr1r! exceptionrwaitFIRST_COMPLETED)rr2taskr!_s rawait_unless_failedz%PermanentFailable.await_unless_failed;s$I..$!!&&(((2244#LLt))*8O8OGD!t|!z/..$$..0 0/"/...suCCCA)C?CCCC C C!C&CCCCCCC CC poll_actionNcK |j|d{7#tj$rYyt$r$}|j t |Yd}~yd}~wwxYww)z Run a polling loop, which runs poll_action forever unless this is failed. Args: poll_action: A callable returning an awaitable to run in a loop. Note that async functions which return once satisfy this. N)r:rCancelledError Exceptionfailr)rr;r s r run_pollerzPermanentFailable.run_pollerNs_ &..{}====%%   & IIk!n % % &s8A*&$&A'A*A'A"A*"A''A*errc|jjs6|jj||jjyyr)r1r! set_exceptionr5)rrAs rr?zPermanentFailable.fail]sA!!&&(    , ,S 1    ( ( * )rcl|jjsy|jjSr)r1r!r5rs rerrorzPermanentFailable.errords,!!&&(!!++--r)r'r(r)__doc__rrr0__annotations__rpropertyr1rr r:rr@rr?rEr*rrr,r,,sn!'..11((w~~(( 19Q<1A1& &HR45H,I &+*+.x 23.rr,)rtypingrrrrgoogle.api_core.exceptionsrr6google.cloud.pubsublite.internal.wait_ignore_cancelledr r r>rrr,r*rrrLsG99BU CL797!37 1 1;.;.r