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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
use std::{
borrow::Borrow,
sync::{mpsc, Arc, RwLock},
};
use crate::{
dispatch::{
dispatcher::{ThreadLocal, ThreadPoolWrapper},
stage::Stage,
},
world::World,
};
use std::borrow::BorrowMut;
pub fn new_async<'a, R>(
world: R,
stages: Vec<Stage<'static>>,
thread_local: ThreadLocal<'a>,
thread_pool: Arc<RwLock<ThreadPoolWrapper>>,
) -> AsyncDispatcher<'a, R> {
AsyncDispatcher {
data: Data::Inner(Inner { world, stages }),
thread_local,
thread_pool,
}
}
/// Like, `Dispatcher` but works asynchronously.
pub struct AsyncDispatcher<'a, R> {
data: Data<R>,
thread_local: ThreadLocal<'a>,
thread_pool: Arc<RwLock<ThreadPoolWrapper>>,
}
impl<'a, R> AsyncDispatcher<'a, R>
where
R: Borrow<World> + Send + Sync + 'static,
{
/// Sets up all the systems which means they are gonna add default values
/// for the resources they need.
pub fn setup(&mut self)
where
R: BorrowMut<World>,
{
let inner = self.data.inner();
let stages = &mut inner.stages;
let world = inner.world.borrow_mut();
for stage in stages {
stage.setup(world);
}
for sys in &mut self.thread_local {
sys.setup(world);
}
}
/// Dispatches the systems asynchronously.
/// Does not execute thread local systems.
///
/// If you want to wait for the systems to finish,
/// call `wait()`.
pub fn dispatch(&mut self) {
let (snd, mut inner) = self.data.sender();
self.thread_pool
.read()
.unwrap()
.as_ref()
.unwrap()
.spawn(move || {
let world: &World = inner.world.borrow();
for stage in &mut inner.stages {
stage.execute(world);
}
let _ = snd.send(inner);
});
}
/// Waits for all the asynchronously dispatched systems to finish
/// and executes thread local systems (if there are any).
pub fn wait(&mut self) {
let world = self.data.inner().world.borrow();
for sys in &mut self.thread_local {
sys.run_now(world);
}
}
/// Waits for all the asynchronously dispatched systems to finish
/// without executing thread local systems.
///
/// See `wait` for executing thread local systems.
pub fn wait_without_tl(&mut self) {
self.data.inner();
}
/// Checks if any of the asynchronously dispatched systems are running.
pub fn running(&mut self) -> bool {
self.data.inner_noblock().is_none()
}
/// Returns the `World`.
///
/// This will wait for the asynchronous systems to finish.
///
/// Renamed to `self.world()`.
#[deprecated(since = "0.8.0", note = "renamed to `world`")]
pub fn res(&mut self) -> &R {
self.world()
}
/// Returns the `World`.
///
/// This will wait for the asynchronous systems to finish.
pub fn world(&mut self) -> &R {
&self.data.inner().world
}
/// Borrows the `World` mutably.
///
/// This will wait for the asynchronous systems to finish.
///
/// Renamed to `self.world_mut()`.
#[deprecated(since = "0.8.0", note = "renamed to `world_mut`")]
pub fn mut_res(&mut self) -> &mut R {
&mut self.data.inner().world
}
/// Borrows the `World` mutably.
///
/// This will wait for the asynchronous systems to finish.
pub fn world_mut(&mut self) -> &mut R {
&mut self.data.inner().world
}
}
enum Data<R> {
Inner(Inner<R>),
Rx(mpsc::Receiver<Inner<R>>),
}
impl<R> Data<R> {
fn inner(&mut self) -> &mut Inner<R> {
*self = match self {
Data::Inner(inner) => return inner,
Data::Rx(rx) => Data::Inner(rx.recv().expect("Sender dropped")),
};
self.inner()
}
fn inner_noblock(&mut self) -> Option<&mut Inner<R>> {
use std::sync::mpsc::TryRecvError;
let new_self;
match *self {
Data::Inner(ref mut inner) => return Some(inner),
Data::Rx(ref mut rx) => {
let inner = rx
.try_recv()
.map(Some)
.or_else(|e| match e {
TryRecvError::Empty => Ok(None),
TryRecvError::Disconnected => Err(e),
})
.expect("Sender dropped");
match inner {
Some(inner) => new_self = Data::Inner(inner),
None => return None,
}
}
}
*self = new_self;
self.inner_noblock()
}
fn sender(&mut self) -> (mpsc::Sender<Inner<R>>, Inner<R>) {
use std::mem::replace;
self.inner();
let (snd, rx) = mpsc::channel();
let inner = replace(&mut *self, Data::Rx(rx));
let inner = match inner {
Data::Inner(inner) => inner,
Data::Rx(_) => unreachable!(),
};
(snd, inner)
}
}
struct Inner<R> {
stages: Vec<Stage<'static>>,
world: R,
}