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,9 +97,11 @@ class StreamIdGenerator(object):
|
||||||
|
|
||||||
@contextlib.contextmanager
|
@contextlib.contextmanager
|
||||||
def manager():
|
def manager():
|
||||||
yield next_id
|
try:
|
||||||
with self._lock:
|
yield next_id
|
||||||
self._unfinished_ids.remove(next_id)
|
finally:
|
||||||
|
with self._lock:
|
||||||
|
self._unfinished_ids.remove(next_id)
|
||||||
|
|
||||||
return manager()
|
return manager()
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue