From 2438b436716ed21432b809a21024f54caebab887 Mon Sep 17 00:00:00 2001
From: Motoyuki Kimura <moymoymox@gmail.com>
Date: Sat, 20 Apr 2024 19:35:11 +0900
Subject: [PATCH] sync: instrument `Semaphore` for task dumps (#6499)

---
 tokio/src/sync/batch_semaphore.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tokio/src/sync/batch_semaphore.rs b/tokio/src/sync/batch_semaphore.rs
index d7eb1d6b7..def5cbc9f 100644
--- a/tokio/src/sync/batch_semaphore.rs
+++ b/tokio/src/sync/batch_semaphore.rs
@@ -575,6 +575,8 @@ impl Future for Acquire<'_> {
     type Output = Result<(), AcquireError>;
 
     fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
+        ready!(crate::trace::trace_leaf(cx));
+
         #[cfg(all(tokio_unstable, feature = "tracing"))]
         let _resource_span = self.node.ctx.resource_span.clone().entered();
         #[cfg(all(tokio_unstable, feature = "tracing"))]