Python: построение реентерабельного семафора (объединение RLock и семафора)

Question

Как бы вы объединили threading.RLock с threading.Semaphore? Или такая структура уже существует?

В Python существует примитив для блокировки реентера, threading.RLock(N), который позволяет одному и тому же потоку приобретать блокировку несколько раз, но ни один другой поток не может. Существует также threading.Semaphore(N), что позволяет блокировать блокировку N раз перед блокировкой. Как объединить эти две структуры? Я хочу до N отдельных потоков, чтобы иметь возможность получить блокировку, но я бы хотел, чтобы каждый отдельный замок на потоке был реентерабельным.

Answer 1

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

import threading 
import datetime 
class ReentrantSemaphore(object): 
    '''A counting Semaphore which allows threads to reenter.''' 
    def __init__(self, value = 1): 
    self.local = threading.local() 
    self.sem = threading.Semaphore(value) 

    def acquire(self): 
    if not getattr(self.local, 'lock_level', 0): 
     # We do not yet have the lock, acquire it. 
     start = datetime.datetime.utcnow() 
     self.sem.acquire() 
     end = datetime.datetime.utcnow() 
     if end - start > datetime.timedelta(seconds = 3): 
     logging.info("Took %d Sec to lock."%((end - start).total_seconds())) 
     self.local.lock_time = end 
     self.local.lock_level = 1 
    else: 
     # We already have the lock, just increment it due to the recursive call. 
     self.local.lock_level += 1 

    def release(self): 
    if getattr(self.local, 'lock_level', 0) < 1: 
     raise Exception("Trying to release a released lock.") 

    self.local.lock_level -= 1 
    if self.local.lock_level == 0: 
     self.sem.release() 

    __enter__ = acquire 
    def __exit__(self, t, v, tb): 
    self.release()