Use try..finally in contextlib.contextmanager

2015-04-15 10:25:43 +01:00 · 2015-04-15 10:25:43 +01:00 · a971fa9d58
parent ded4128965
commit a971fa9d58
1 changed files with 5 additions and 3 deletions
--- a/synapse/storage/util/id_generators.py
+++ b/synapse/storage/util/id_generators.py
@ -97,7 +97,9 @@ class StreamIdGenerator(object):

        @contextlib.contextmanager
        def manager():
+            try:
                yield next_id
+            finally:
                with self._lock:
                    self._unfinished_ids.remove(next_id)