o ?e@s dZddlZGdddeZdS)zLocking related utils.Nc@sZeZdZdZgdZdddZddZdd Zd d Zd d Z ddZ Gddde Z dS) GroupLockaA lock to allow many members of a group to access a resource exclusively. This lock provides a way to allow access to a resource by multiple threads belonging to a logical group at the same time, while restricting access to threads from all other groups. You can think of this as an extension of a reader-writer lock, where you allow multiple writers at the same time. We made it generic to support multiple groups instead of just two - readers and writers. Simple usage example with two groups accessing the same resource: ```python lock = GroupLock(num_groups=2) # In a member of group 0: with lock.group(0): # do stuff, access the resource # ... # In a member of group 1: with lock.group(1): # do stuff, access the resource # ... ``` Using as a context manager with `.group(group_id)` is the easiest way. You can also use the `acquire` and `release` method directly. )_ready _num_groups_group_member_countscCs>|dkr td|tt|_||_dg|j|_dS)a6Initialize a group lock. Args: num_groups: The number of groups that will be accessing the resource under consideration. Should be a positive number. Returns: A group lock that can then be used to synchronize code. Raises: ValueError: If num_groups is less than 1. zGArgument `num_groups` must be a positive integer. Received: num_groups=rN) ValueError threading ConditionLockrrr)selfZ num_groupsr a/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/util/lock_util.py__init__4s zGroupLock.__init__cCs|||||S)zEnter a context where the lock is with group `group_id`. Args: group_id: The group for which to acquire and release the lock. Returns: A context manager which will acquire the lock for `group_id`. )_validate_group_id_Contextr group_idr r rgroupIs zGroupLock.groupcCsR|||j||r|j||s|j|d7<|jdS)z7Acquire the group lock for a specific group `group_id`.rN)rracquire_another_group_activewaitrreleaserr r rrUs     zGroupLock.acquirecCsL|||j|j|d8<|j|dkr|j|jdS)z7Release the group lock for a specific group `group_id`.rrN)rrrr notify_allrrr r rr_s   zGroupLock.releasecstfddt|jDS)Nc3s$|] \}}|kr|dkVqdS)rNr ).0gcrr r js   z2GroupLock._another_group_active..)any enumeraterrr rrris zGroupLock._another_group_activecCs,|dks ||jkrtd|jd|dS)NrzQArgument `group_id` should verify `0 <= group_id < num_groups` (with `num_groups=z`). Received: group_id=)rrrr r rrmszGroupLock._validate_group_idc@s0eZdZdZddgZddZddZdd Zd S) zGroupLock._Contextz'Context manager helper for `GroupLock`._lock _group_idcCs||_||_dSN)r!r")r lockrr r rrys zGroupLock._Context.__init__cCs|j|jdSr#)r!rr")r r r r __enter__}szGroupLock._Context.__enter__cCs~~~|j|jdSr#)r!rr")r Ztype_argZ value_argZ traceback_argr r r__exit__szGroupLock._Context.__exit__N)__name__ __module__ __qualname____doc__ __slots__rr%r&r r r rrts  rN)r) r'r(r)r*r+rrrrrrobjectrr r r rrs    r)r*r r,rr r r rs