d3c4e4a20a
This introduces a `Pender` struct with enum cases for thread-mode, interrupt-mode and custom callback executors. This avoids calls through function pointers when using only the thread or interrupt executors. Faster, and friendlier to `cargo-call-stack`. `embassy-executor` now has `arch-xxx` Cargo features to select the arch and to enable the builtin executors (thread and interrupt).
101 lines
3.1 KiB
Rust
101 lines
3.1 KiB
Rust
#[cfg(feature = "executor-interrupt")]
|
|
compile_error!("`executor-interrupt` is not supported with `arch-std`.");
|
|
|
|
#[cfg(feature = "executor-thread")]
|
|
pub use thread::*;
|
|
#[cfg(feature = "executor-thread")]
|
|
mod thread {
|
|
use std::marker::PhantomData;
|
|
use std::sync::{Condvar, Mutex};
|
|
|
|
#[cfg(feature = "nightly")]
|
|
pub use embassy_macros::main_std as main;
|
|
|
|
use crate::raw::{Pender, PenderInner};
|
|
use crate::{raw, Spawner};
|
|
|
|
#[derive(Copy, Clone)]
|
|
pub(crate) struct ThreadPender(&'static Signaler);
|
|
|
|
impl ThreadPender {
|
|
#[allow(unused)]
|
|
pub(crate) fn pend(self) {
|
|
self.0.signal()
|
|
}
|
|
}
|
|
|
|
/// Single-threaded std-based executor.
|
|
pub struct Executor {
|
|
inner: raw::Executor,
|
|
not_send: PhantomData<*mut ()>,
|
|
signaler: &'static Signaler,
|
|
}
|
|
|
|
impl Executor {
|
|
/// Create a new Executor.
|
|
pub fn new() -> Self {
|
|
let signaler = &*Box::leak(Box::new(Signaler::new()));
|
|
Self {
|
|
inner: raw::Executor::new(Pender(PenderInner::Thread(ThreadPender(signaler)))),
|
|
not_send: PhantomData,
|
|
signaler,
|
|
}
|
|
}
|
|
|
|
/// Run the executor.
|
|
///
|
|
/// The `init` closure is called with a [`Spawner`] that spawns tasks on
|
|
/// this executor. Use it to spawn the initial task(s). After `init` returns,
|
|
/// the executor starts running the tasks.
|
|
///
|
|
/// To spawn more tasks later, you may keep copies of the [`Spawner`] (it is `Copy`),
|
|
/// for example by passing it as an argument to the initial tasks.
|
|
///
|
|
/// This function requires `&'static mut self`. This means you have to store the
|
|
/// Executor instance in a place where it'll live forever and grants you mutable
|
|
/// access. There's a few ways to do this:
|
|
///
|
|
/// - a [StaticCell](https://docs.rs/static_cell/latest/static_cell/) (safe)
|
|
/// - a `static mut` (unsafe)
|
|
/// - a local variable in a function you know never returns (like `fn main() -> !`), upgrading its lifetime with `transmute`. (unsafe)
|
|
///
|
|
/// This function never returns.
|
|
pub fn run(&'static mut self, init: impl FnOnce(Spawner)) -> ! {
|
|
init(self.inner.spawner());
|
|
|
|
loop {
|
|
unsafe { self.inner.poll() };
|
|
self.signaler.wait()
|
|
}
|
|
}
|
|
}
|
|
|
|
struct Signaler {
|
|
mutex: Mutex<bool>,
|
|
condvar: Condvar,
|
|
}
|
|
|
|
impl Signaler {
|
|
fn new() -> Self {
|
|
Self {
|
|
mutex: Mutex::new(false),
|
|
condvar: Condvar::new(),
|
|
}
|
|
}
|
|
|
|
fn wait(&self) {
|
|
let mut signaled = self.mutex.lock().unwrap();
|
|
while !*signaled {
|
|
signaled = self.condvar.wait(signaled).unwrap();
|
|
}
|
|
*signaled = false;
|
|
}
|
|
|
|
fn signal(&self) {
|
|
let mut signaled = self.mutex.lock().unwrap();
|
|
*signaled = true;
|
|
self.condvar.notify_one();
|
|
}
|
|
}
|
|
}
|