Matrix-Mirrors/synapse-product
1
0
Fork 0
mirror of https://git.anonymousland.org/anonymousland/synapse-product.git synced 2024-12-22 01:44:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

Doc get_next() context manager usage

Browse Source
This commit is contained in:
Erik Johnston 2016-08-15 13:45:26 +01:00
parent d9664344ec
commit 89e786bd85
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
synapse/storage/_base.py
View File

@ -873,6 +873,10 @@ class SQLBaseStore(object):
txn.call_after(cache_func.invalidate, keys) txn.call_after(cache_func.invalidate, keys)
if isinstance(self.database_engine, PostgresEngine): if isinstance(self.database_engine, PostgresEngine):
# get_next() returns a context manager which is designed to wrap
# the transaction. However, we want to only get an ID when we want
# to use it, here, so we need to call __enter__ manually, and have
# __exit__ called after the transaction finishes.
ctx = self._cache_id_gen.get_next() ctx = self._cache_id_gen.get_next()
stream_id = ctx.__enter__() stream_id = ctx.__enter__()
txn.call_after(ctx.__exit__, None, None, None) txn.call_after(ctx.__exit__, None, None, None)