neon/executor/
tokio.rs

1use std::sync::Arc;
2
3use super::{BoxFuture, Runtime};
4
5impl Runtime for tokio::runtime::Runtime {
6    fn spawn(&self, fut: BoxFuture) {
7        spawn(self.handle(), fut);
8    }
9}
10
11impl Runtime for Arc<tokio::runtime::Runtime> {
12    fn spawn(&self, fut: BoxFuture) {
13        spawn(self.handle(), fut);
14    }
15}
16
17impl Runtime for &'static tokio::runtime::Runtime {
18    fn spawn(&self, fut: BoxFuture) {
19        spawn(self.handle(), fut);
20    }
21}
22
23impl Runtime for tokio::runtime::Handle {
24    fn spawn(&self, fut: BoxFuture) {
25        spawn(self, fut);
26    }
27}
28
29impl Runtime for &'static tokio::runtime::Handle {
30    fn spawn(&self, fut: BoxFuture) {
31        spawn(self, fut);
32    }
33}
34
35fn spawn(handle: &tokio::runtime::Handle, fut: BoxFuture) {
36    #[allow(clippy::let_underscore_future)]
37    let _ = handle.spawn(fut);
38}
39
40#[cfg(feature = "tokio-rt-multi-thread")]
41pub(crate) fn init(cx: &mut crate::context::ModuleContext) -> crate::result::NeonResult<()> {
42    use once_cell::sync::OnceCell;
43    use tokio::runtime::{Builder, Runtime};
44
45    use crate::context::Context;
46
47    static RUNTIME: OnceCell<Runtime> = OnceCell::new();
48
49    super::RUNTIME.get_or_try_init(cx, |cx| {
50        let runtime = RUNTIME
51            .get_or_try_init(|| {
52                #[cfg(feature = "tokio-rt-multi-thread")]
53                let mut builder = Builder::new_multi_thread();
54
55                #[cfg(not(feature = "tokio-rt-multi-thread"))]
56                let mut builder = Builder::new_current_thread();
57
58                builder.enable_all().build()
59            })
60            .or_else(|err| cx.throw_error(err.to_string()))?;
61
62        Ok(Box::new(runtime))
63    })?;
64
65    Ok(())
66}
neon/executor/tokio.rs

neon/executor/
tokio.rs
