diff --git a/changelog.d/13554.misc b/changelog.d/13554.misc new file mode 100644 index 000000000000..99dbcebec8a1 --- /dev/null +++ b/changelog.d/13554.misc @@ -0,0 +1 @@ +Instrument `FederationStateIdsServlet` (`/state_ids`) for understandable traces in Jaeger. diff --git a/synapse/util/ratelimitutils.py b/synapse/util/ratelimitutils.py index 434b02b97b9d..ce9fa78b0ae1 100644 --- a/synapse/util/ratelimitutils.py +++ b/synapse/util/ratelimitutils.py @@ -222,20 +222,28 @@ def on_both(r: object) -> object: # Ensure that we've properly cleaned up. self.sleeping_requests.discard(request_id) self.ready_request_queue.pop(request_id, None) - wait_span_scope.__exit__(None, None, None) - wait_timer_cm.__exit__(None, None, None) return r + ret_defer.addCallbacks(on_start, on_err) + ret_defer.addBoth(on_both) + # Tracing wait_span_scope = start_active_span("ratelimit wait") wait_span_scope.__enter__() + # Metrics wait_timer_cm = queue_wait_timer.time() wait_timer_cm.__enter__() - ret_defer.addCallbacks(on_start, on_err) - ret_defer.addBoth(on_both) - return make_deferred_yieldable(ret_defer) + # The current logcontext is now one created by `wait_span_scope`. + # We must only `__exit__` it once it has been restored by `make_deferred_yieldable`. + def on_after_logcontext_restored(r: object) -> object: + wait_timer_cm.__exit__(None, None, None) + wait_span_scope.__exit__(None, None, None) + + d = make_deferred_yieldable(ret_defer) + d.addBoth(on_after_logcontext_restored) + return d def _on_exit(self, request_id: object) -> None: logger.debug("Ratelimit(%s) [%s]: Processed req", self.host, id(request_id))