блокировка блокировки с оператором 'with'

Question

Насколько я знаю, следующий код будет заблокирован, если lock уже получен другим потоком.

Кажется, что неблокирование может быть реализовано lock.acquire(0), но вместо этого я должен использовать блок try-finally вместо with.

lock = threading.Lock() 

def func(): 
with lock: 
    # do something... 

Есть ли способ реализации неблокируемой приобретения замка?

Answer 1

Весь смысл блокировки состоит в том, чтобы определенные разделы вашей программы выполнялись только по одному потоку или процессу за раз. Это достигается путем блокировки любых потоков/процессов, пытающихся получить блокировку, в то время как что-то еще удерживает ее.

Если вы не хотите, чтобы блокировка блокировалась, почему вы используете блокировку в первую очередь? Предположительно, чтобы вы могли сделать что-то еще, пока вы ждете?

Чтобы попытаться приобрести замок l без блокировки, позвоните по телефону l.acquire(blocking=False). Это немедленно вернет False, если замок не был приобретен. Если замок был, он возвращает True, и вы продолжаете удерживать замок до тех пор, пока не назовете его метод release().

Эта форма, однако, не особенно полезна в операторе with. Обычно вам нужен управляемый код (с отступом после with) для запуска только после того, как блокировка была приобретена. не запрашивать, имеет ли он или нет, и предпринять два альтернативных действия.

Answer 2

Есть ли способ реализовать блокировку блокировки блокировки?

Да. Просто поднимите исключение, если замок не может быть получен немедленно. Что-то вроде:

@contextlib.contextmanager 
def non_blocking_lock(lock=threading.Lock()): 
    if not lock.acquire(blocking=False): 
     raise WouldBlockError 
    try: 
     yield lock 
    finally: 
     lock.release() 

Использование:

with non_blocking_lock(): 
    # run with the lock acquired 

Answer 3

Вы можете реализовать свой собственный замок, как объект, который ведет себя, как вы хотите:

class MyLock(object): 
    def __init__(self): 
     self._lock = threading.Lock() 
     self._locked = False 
    def __enter__(self): 
     locked = self._lock.acquire(False) 
     self._locked = locked 
     return locked 
    def __exit__(self, *args): 
     if self._locked: 
      self._lock.release() 
      self._locked = False 

Я тестировал его с этим кодом:

In [5]: ml = MyLock() 
In [6]: with ml as aq: 
    ...:  print "got it?!", aq 
    ...:  with ml as try2: 
    ...:   print "got it second time???", try2 

Выход:

got it?! True 
got it second time??? False 

Вы также можете использовать методы блокировки и разблокировки, которые используют внутренний замок, но вы получаете изображение, соответствующее его потребностям.

Answer 4

@contextmanager 
def nonblocking(lock): 
    locked = lock.acquire(False) 
    try: 
     yield locked 
    finally: 
     if locked: 
      lock.release() 

lock = threading.Lock() 
with nonblocking(lock) as locked: 
    if locked: 
     do_stuff()