forked from MirrorHub/synapse
Use try..finally in contextlib.contextmanager
This commit is contained in:
parent
ded4128965
commit
a971fa9d58
1 changed files with 5 additions and 3 deletions
|
@ -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)
|
||||
|
||||
|
|
Loading…
Reference in a new issue