diff options
| author | Erik Johnston <erik@matrix.org> | 2019-09-05 15:22:24 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-09-05 15:22:24 +0100 |
| commit | 1a6ae333090c5b34ab6a3e0427dfc87ed99d6a9c (patch) | |
| tree | 0c1b096f70c9d7743049765fdc31639001fa84ea /synapse/federation/sender | |
| parent | Add opentracing to all client servlets (#5983) (diff) | |
| parent | use access methods (duh..) (diff) | |
| download | synapse-1a6ae333090c5b34ab6a3e0427dfc87ed99d6a9c.tar.xz | |
Merge pull request #5984 from matrix-org/joriks/opentracing_link_send_to_edu_contexts
Link the send loop with the edus contexts
Diffstat (limited to 'synapse/federation/sender')
| -rw-r--r-- | synapse/federation/sender/transaction_manager.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/synapse/federation/sender/transaction_manager.py b/synapse/federation/sender/transaction_manager.py index 62ca6a3e87..5b6c79c51a 100644 --- a/synapse/federation/sender/transaction_manager.py +++ b/synapse/federation/sender/transaction_manager.py @@ -26,6 +26,7 @@ from synapse.logging.opentracing import ( set_tag, start_active_span_follows_from, tags, + whitelisted_homeserver, ) from synapse.util.metrics import measure_func @@ -59,9 +60,15 @@ class TransactionManager(object): # The span_contexts is a generator so that it won't be evaluated if # opentracing is disabled. (Yay speed!) - span_contexts = ( - extract_text_map(json.loads(edu.get_context())) for edu in pending_edus - ) + span_contexts = [] + keep_destination = whitelisted_homeserver(destination) + + for edu in pending_edus: + context = edu.get_context() + if context: + span_contexts.append(extract_text_map(json.loads(context))) + if keep_destination: + edu.strip_context() with start_active_span_follows_from("send_transaction", span_contexts): |