cdk-common 0.16.0-rc.0

CDK common types and traits
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

//! Thin wrapper for spawn and spawn_local for native and wasm.

use std::future::Future;
#[cfg(not(target_arch = "wasm32"))]
use std::sync::OnceLock;

use tokio::task::JoinHandle;

#[cfg(not(target_arch = "wasm32"))]
static GLOBAL_RUNTIME: OnceLock<tokio::runtime::Runtime> = OnceLock::new();

/// Spawns a new asynchronous task returning nothing
///
/// # Panics
///
/// Panics if the global Tokio runtime cannot be created when no runtime exists on the current thread.
#[cfg(not(target_arch = "wasm32"))]
pub fn spawn<F>(future: F) -> JoinHandle<F::Output>
where
    F: Future + Send + 'static,
    F::Output: Send + 'static,
{
    if let Ok(handle) = tokio::runtime::Handle::try_current() {
        handle.spawn(future)
    } else {
        // No runtime on this thread (FFI/regular sync context):
        // use (or lazily create) a global runtime and spawn on it.
        GLOBAL_RUNTIME
            .get_or_init(|| {
                tokio::runtime::Builder::new_multi_thread()
                    .enable_all()
                    .build()
                    .expect("failed to build global Tokio runtime")
            })
            .spawn(future)
    }
}

/// Spawns a new asynchronous task returning nothing
#[cfg(target_arch = "wasm32")]
pub fn spawn<F>(future: F) -> JoinHandle<F::Output>
where
    F: Future + 'static,
    F::Output: 'static,
{
    tokio::task::spawn_local(future)
}