blokada pyton z-oświadczenie oraz timeout

Question

Używam sekwencję Python 3 tak:

lock = threading.Lock() 
res = lock.acquire(timeout=10) 
if res: 
    # do something .... 
    lock.release() 
else: 
    # do something else ... 

wolałbym używać z-oświadczenie zamiast wyraźnego „zdobyć” i „release”, ale ja don wiesz, jak uzyskać efekt limitu czasu.

Answer 1

Można to zrobić bardzo łatwo z menedżerem kontekstu:

import threading 
from contextlib import contextmanager 

@contextmanager 
def acquire_timeout(lock, timeout): 
    result = lock.acquire(timeout=timeout) 
    yield result 
    if result: 
     lock.release() 


# Usage: 
lock = threading.Lock() 

with acquire_timeout(lock, 2) as acquired: 
    if acquired: 
     print('got the lock') 
     # do something .... 
    else: 
     print('timeout: lock not available') 
     # do something else ... 

* UWAGA: To nie będzie działać w Pythonie 2.x, ponieważ nie jest timeout argument Lock.acquire

Answer 2

Lekko ładniejsza wersja:

import threading 
from contextlib import contextmanager 


class TimeoutLock(object): 
    def __init__(self): 
     self._lock = threading.Lock() 

    def acquire(self, blocking=True, timeout=-1): 
     return self._lock.acquire(blocking, timeout) 

    @contextmanager 
    def acquire_timeout(self, timeout): 
     result = self._lock.acquire(timeout=timeout) 
     yield result 
     if result: 
      self._lock.release() 

    def release(self): 
     self._lock.release() 

# Usage: 
lock = TimeoutLock() 

with lock.acquire_timeout(3) as result: 
    if result: 
     print('got the lock') 
     # do something .... 
    else: 
     print('timeout: lock not available') 
     # do something else ... 

wydaje się, że nie może podklasy threading.Lock, więc Zamiast tego musiałem stworzyć klasę otoki.