Skip to content

Instantly share code, notes, and snippets.

@jelmervdl

jelmervdl/readwritelock.py

Created March 22, 2023 11:51
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save jelmervdl/d933f45b4af9404d9a3d4f525276a005 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save jelmervdl/d933f45b4af9404d9a3d4f525276a005 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. jelmervdl created this gist Mar 22, 2023.
    30 changes: 30 additions & 0 deletions readwritelock.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,30 @@
    class ReadWriteLock:
    def __init__(self):
    self.readers = 0
    self.lock = threading.Condition(threading.Lock())

    @contextmanager
    def lock_for_reading(self):
    self.lock.acquire()
    try:
    self.readers += 1
    finally:
    self.lock.release()

    try:
    yield
    finally:
    self.lock.acquire()
    try:
    self.readers -= 1
    finally:
    self.lock.release()

    @contextmanager
    def lock_for_writing(self):
    self.lock.acquire()
    self.lock.wait_for(lambda: self.readers == 0)
    try:
    yield
    finally:
    self.lock.release()