From 4748b2571fc02d5ebbfe59e457f0e8d8ef0eb5f3 Mon Sep 17 00:00:00 2001
From: Carl Lerche <me@carllerche.com>
Date: Wed, 6 May 2020 19:02:07 -0700
Subject: rt: simplify coop implementation (#2498)

Simplifies coop implementation. Prunes unused code, create a `Budget`
type to track the current budget.
---
 tokio/src/sync/batch_semaphore.rs | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'tokio/src/sync/batch_semaphore.rs')

diff --git a/tokio/src/sync/batch_semaphore.rs b/tokio/src/sync/batch_semaphore.rs
index 436737a6..698e908e 100644
--- a/tokio/src/sync/batch_semaphore.rs
+++ b/tokio/src/sync/batch_semaphore.rs
@@ -386,7 +386,11 @@ impl Future for Acquire<'_> {
     type Output = Result<(), AcquireError>;
 
     fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
+        // First, ensure the current task has enough budget to proceed.
+        ready!(crate::coop::poll_proceed(cx));
+
         let (node, semaphore, needed, queued) = self.project();
+
         match semaphore.poll_acquire(cx, needed, node, *queued) {
             Pending => {
                 *queued = true;
-- 
cgit v1.2.3