o i<@sddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl m Z ddl mZddl mZddl mZdd l mZdd lmZd d lmZd d lmZeeZeeeZeeeZedZedZGdddeZdddZ Gddde j!Z"dS)) annotationsN)Any) Awaitable)Callable) NamedTuple)Optional)overload)TypeVar) ParamSpec)log_obj) log_prefixPTc@seZdZUded<ded<dS) LoopControlzasyncio.AbstractEventLooploopz asyncio.Event stop_eventN)__name__ __module__ __qualname____annotations__rrX/home/ubuntu/.local/lib/python3.10/site-packages/libsql_client/dbapi2/_async_executor.pyrs  rr strqqueue.Queue[LoopControl]returnNonecsLddddfd d dfd d }dt|ddS)Nexc_infomsgrargsobjectrOptional[BaseException]rrcs@rdnd}|d}t|tj|g|Rd|idS)N z: r) _log_prefixloggingDEBUG)r rr!loop_strprefix)r rrrdbg(s z_thread_main..dbgc st}tz=zdt||IdHdWnty<}z d||dWYd}~n d}~wwWddSWddSdw)Nzstarted main()zfinished main()zfailed main(): %sr)asyncioEventget_running_loop put_nowaitrwait Exception)re)r+rrrrmain2s  z_thread_main..mainzstarted threadzfinished thread)r rr!r"rr#rrrr)r,run)r rr3r)r+r rrr _thread_main%s   r6cseZdZUded<ded<dZd*fdd Zd*d d Zd+d dZe e Z e e e jZe e e jZe e e jZd*ddZejdddkr^ed,ddZed-ddZed.d dZned/d#dZed0d$dZed1d%dZd2d'dZd2d(d)ZZS)3 AsyncExecutorzthreading.Lock_lockzOptional[LoopControl]_control)r9r8rrcsvd|_td}d|jjdt|dd}tjdt||fd| d| t |_ ||_| d dS) Nr < at x>T)daemontargetr!createdz thread ready)r9queueQueue __class__ridsuper__init__r6_infstart threadingLockr8get)selfrr rCrrrFJs    zAsyncExecutor.__init__cCs |d|jdusJddS)N destroyedzThread is still running)rGr9)rLrrr__del__Us zAsyncExecutor.__del__rc Cs\tt|}|r dnd}|jdur|d|jj7}d|jjd|d|jd|d S) Nstartedstoppedz loop=r:r;z name=r$r=)hexrDis_aliver9rrCrname)rLaddrsrrr__repr__[s  $zAsyncExecutor.__repr__csdjjdurtdj\}d_Wdn1s$wYd fdd }d|}t||}|ddS) Nzshuting down thread...zthread already downrrcsddS)Nznotifying _stop_event)_dbgsetrrLrrrrun_in_main_loopps  z0AsyncExecutor.shutdown..run_in_main_loopzthread will stopzthread did stopr4) rXr8r9 RuntimeErrorr,run_coroutine_threadsaferesultjoinrG)rLrr[corofuturerrZrshutdowngs     zAsyncExecutor.shutdownN) fn)Callable[P, Awaitable[asyncio.Future[T]]]r!P.argskwargsP.kwargsconcurrent.futures.Future[T]cOdSNrrLrfr!rirrrsubmit}zAsyncExecutor.submitCallable[P, Awaitable[T]]cOrlrmrrnrrrrorpCallable[P, T]cOrlrmrrnrrrrorp&Callable[P, Awaitable[asyncio.Future]]concurrent.futures.FuturecOrlrmrrnrrrrorpcOrlrmrrnrrrrorpcOrlrmrrnrrrrorprcOsB|j|j|g|Ri|WdS1swYdSrm)r8_unlocked_submitrnrrrros$cs<jdur tddfdd }|}t|jjS)Nz already downrrc sz4di}t|st|r*|IdH}t|st|sd||WStyM}z jd||dd}~ww)Nzcalling: %s, args=%s, kwargs=%sz+finished: %s, args=%s, kwargs=%s, result=%sz&failed: %s, args=%s, kwargs=%s, exc=%sr)rXr, iscoroutineisfuturer1)rr2r!rfrirLrrr[s6 z8AsyncExecutor._unlocked_submit..run_in_main_loop)rr)r9r\r,r]r)rLrfr!rir[r`rryrrus zAsyncExecutor._unlocked_submitr4)rr)rfrgr!rhrirjrrk)rfrqr!rhrirjrrk)rfrrr!rhrirjrrk)rfrsr!rhrirjrrt)rfrqr!rhrirjrrt)rfrrr!rhrirjrrt)rfrrr!rhrirjrr)rrrr __slots__rFrOrW functoolspartial_log_obj_log partialmethodr'r(rXINFOrGERROR_errrbsys version_inforroru __classcell__rrrMrr7Es6           r7)r rrrrr)# __future__rr,concurrent.futures concurrentr{r'rArrItypingrrrrrrr typing_extensionsr _utilsr r getLoggerr_loggerr|r}r&rrrr6Threadr7rrrrs4