moirai-executor 0.8.0

Hybrid executor implementation for Moirai concurrency library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

//! Thread-local bound on nested inline polls.

use std::cell::Cell;

const ASYNC_INLINE_POLL_DEPTH_LIMIT: usize = 1;

thread_local! {
    static ASYNC_INLINE_POLL_DEPTH: Cell<usize> = const { Cell::new(0) };
}

pub(super) struct InlinePollDepthGuard {
    previous: usize,
}

impl InlinePollDepthGuard {
    pub(super) fn try_enter() -> Option<Self> {
        ASYNC_INLINE_POLL_DEPTH.with(|depth| {
            let previous = depth.get();
            (previous < ASYNC_INLINE_POLL_DEPTH_LIMIT).then(|| {
                depth.set(previous + 1);
                Self { previous }
            })
        })
    }
}

impl Drop for InlinePollDepthGuard {
    fn drop(&mut self) {
        ASYNC_INLINE_POLL_DEPTH.with(|depth| depth.set(self.previous));
    }
}