diff options
author | Richard van der Hoff <richard@matrix.org> | 2017-03-15 02:21:07 +0000 |
---|---|---|
committer | Richard van der Hoff <richard@matrix.org> | 2017-03-15 02:21:07 +0000 |
commit | b5d1c68beb24c314006398052070448d67bb4983 (patch) | |
tree | a4a706b434f64b4c42cda2a35360bdbaed7161ed /synapse/util | |
parent | Merge pull request #1978 from matrix-org/rav/refactor_received_pdu (diff) | |
download | synapse-b5d1c68beb24c314006398052070448d67bb4983.tar.xz |
Implement reset_context_after_deferred
to correctly reset the context when we fire off a deferred we aren't going to wait for.
Diffstat (limited to 'synapse/util')
-rw-r--r-- | synapse/util/logcontext.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/synapse/util/logcontext.py b/synapse/util/logcontext.py index 6c83eb213d..d73670f9f2 100644 --- a/synapse/util/logcontext.py +++ b/synapse/util/logcontext.py @@ -308,6 +308,31 @@ def preserve_context_over_deferred(deferred, context=None): return d +def reset_context_after_deferred(deferred): + """If the deferred is incomplete, add a callback which will reset the + context. + + This is useful when you want to fire off a deferred, but don't want to + wait for it to complete. (The deferred will restore the current log context + when it completes, so if you don't do anything, it will leak log context.) + + (If this feels asymmetric, consider it this way: we are effectively forking + a new thread of execution. We are probably currently within a + ``with LoggingContext()`` block, which is supposed to have a single entry + and exit point. But by spawning off another deferred, we are effectively + adding a new exit point.) + + Args: + deferred (defer.Deferred): deferred + """ + def reset_context(result): + LoggingContext.set_current_context(LoggingContext.sentinel) + return result + + if not deferred.called: + deferred.addBoth(reset_context) + + def preserve_fn(f): """Ensures that function is called with correct context and that context is restored after return. Useful for wrapping functions that return a deferred |