U
|
Z ±dõ ã @ sn d dl Z d dlmZ ddlmZ ddlmZ e jdddG dd d Ze jd
|
d
|
dd G d d d edZ dS )é N)Ú OrderedDicté )Ú_core)ÚFinalT)ÚfrozenÚslotsc @ s e Zd Ze ¡ ZdS )Ú_ParkingLotStatisticsN)Ú__name__Ú
|
__module__Ú __qualname__ÚattrÚibÚ tasks_waiting© r r úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/_parking_lot.pyr P s r F)ÚeqÚhashr c @ s e Zd ZdZejeddZdd Zdd Z e
|
j dd Z d
|
d Z e
|
j d d ddZdd Ze
|
j d d ddZdd Zdd ZdS )Ú
|
ParkingLota¥ A fair wait queue with cancellation and requeueing.
|
|
This class encapsulates the tricky parts of implementing a wait
|
queue. It's useful for implementing higher-level synchronization
|
primitives like queues and locks.
|
|
In addition to the methods below, you can use ``len(parking_lot)`` to get
|
the number of parked tasks, and ``if parking_lot: ...`` to check whether
|
there are any parked tasks.
|
|
F)ÚfactoryÚinitc C s
|
t | jS )z#Returns the number of parked tasks.)ÚlenÚ_parked©Úselfr r r Ú__len__g s zParkingLot.__len__c C s
|
t | jS )z0True if there are parked tasks, False otherwise.)Úboolr r r r r Ú__bool__k s zParkingLot.__bool__c s8 t ¡ d| j < | _ fdd}t |¡I dH dS )zfPark the current task until woken by a call to :meth:`unpark` or
|
:meth:`unpark_all`.
|
|
Nc s j j = tjjS )N)Úcustom_sleep_datar r ZAbortZ SUCCEEDED)Ú_©Útaskr r Úabort_fn} s
|
z!ParkingLot.park.<locals>.abort_fn)r Z current_taskr r Zwait_task_rescheduled)r r! r r r Úparks s
|
|
zParkingLot.parkc c s6 t t|t| jD ] }| jjdd\}}|V qd S )NF)Úlast)ÚrangeÚminr r Úpopitem)r Úcountr r r r r Ú _pop_several s zParkingLot._pop_severalé ©r' c C s& t | |¡}|D ]}t |¡ q|S )a7 Unpark one or more tasks.
|
|
This wakes up ``count`` tasks that are blocked in :meth:`park`. If
|
there are fewer than ``count`` tasks parked, then wakes as many tasks
|
are available and then returns successfully.
|
|
Args:
|
count (int): the number of tasks to unpark.
|
|
)Úlistr( r Z
|
reschedule)r r' Ztasksr r r r Úunpark s zParkingLot.unparkc C s | j t| dS )zUnpark all parked tasks.r* )r, r r r r r Ú
|
unpark_all s zParkingLot.unpark_allc C s6 t |tstd| |¡D ]}d|j|< ||_q dS )a¿ Move parked tasks from one :class:`ParkingLot` object to another.
|
|
This dequeues ``count`` tasks from one lot, and requeues them on
|
another, preserving order. For example::
|
|
async def parker(lot):
|
print("sleeping")
|
await lot.park()
|
print("woken")
|
|
async def main():
|
lot1 = trio.lowlevel.ParkingLot()
|
lot2 = trio.lowlevel.ParkingLot()
|
async with trio.open_nursery() as nursery:
|
nursery.start_soon(parker, lot1)
|
await trio.testing.wait_all_tasks_blocked()
|
assert len(lot1) == 1
|
assert len(lot2) == 0
|
lot1.repark(lot2)
|
assert len(lot1) == 0
|
assert len(lot2) == 1
|
# This wakes up the task that was originally parked in lot1
|
lot2.unpark()
|
|
If there are fewer than ``count`` tasks parked, then reparks as many
|
tasks as are available and then returns successfully.
|
|
Args:
|
new_lot (ParkingLot): the parking lot to move tasks to.
|
count (int): the number of tasks to move.
|
|
z new_lot must be a ParkingLotN)Ú
|
isinstancer Ú TypeErrorr( r r )r Únew_lotr' r r r r Úrepark s
|
"
|
|
zParkingLot.reparkc C s | j |t| dS )zMove all parked tasks from one :class:`ParkingLot` object to
|
another.
|
|
See :meth:`repark` for details.
|
|
r* )r1 r )r r0 r r r Ú
|
repark_allÅ s zParkingLot.repark_allc C s t t| jdS )zØReturn an object containing debugging information.
|
|
Currently the following fields are defined:
|
|
* ``tasks_waiting``: The number of tasks blocked on this lot's
|
:meth:`park` method.
|
|
)r )r r r r r r r Ú
|
statisticsÎ s zParkingLot.statisticsN)r r
|
r Ú__doc__r r r r r r r Zenable_ki_protectionr" r( r, r- r1 r2 r3 r r r r r U s
|
' r )Ú metaclass)
|
r Ú collectionsr Ú r Z_utilr Úsr r r r r r Ú<module>I s
|
