okstd 0.2.0

The standard library that's ok
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
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

// async_runtime.rs
use std::future::Future;

pub use futures::{
    future::{FutureExt, TryFutureExt},
    sink::{Sink, SinkExt},
    stream::{Stream, StreamExt, TryStreamExt},
};

pub trait AsyncRead: futures::io::AsyncRead {}
impl<T: ?Sized + futures::io::AsyncRead> AsyncRead for T {}

pub trait AsyncWrite: futures::io::AsyncWrite {}
impl<T: ?Sized + futures::io::AsyncWrite> AsyncWrite for T {}

type Result<T> = anyhow::Result<T>;

pub trait Runtime: Unpin {
    fn block_on<F: Future>(&self, future: F) -> F::Output;

    fn spawn<F>(&self, future: F)
    where
        F: Future + Send + 'static,
        F::Output: Send + 'static,
    {
        self.block_on(future);
    }
}

impl Runtime for tokio::runtime::Runtime {
    fn block_on<F: Future>(&self, future: F) -> F::Output {
        self.block_on(future)
    }
}
#[cfg(feature = "cros_async")]
impl Runtime for cros_async::Executor {
    fn block_on<F: Future>(&self, future: F) -> F::Output {
        self.run_until(future).unwrap()
    }
}

pub enum Runtimes {
    Tokio(tokio::runtime::Runtime),

    #[cfg(feature = "cros_async")]
    Cros(cros_async::Executor),
}

impl Runtime for Runtimes {
    fn block_on<F: Future>(&self, future: F) -> F::Output {
        match self {
            Runtimes::Tokio(rt) => rt.block_on(future),

            #[cfg(feature = "cros_async")]
            Runtimes::Cros(executor) => executor.block_on(future),
        }
    }
}

impl Runtimes {
    pub fn setup_runtimes() -> Result<Runtimes> {
        let worker_count = num_cpus::get() * 1.5 as usize;
        if let Ok(rt) = tokio::runtime::Builder::new_multi_thread()
            // create a thread pool with core count of the machine
            .worker_threads(worker_count)
            .enable_all()
            .build()
        {
            Ok(Runtimes::Tokio(rt))
        } else {
            Err(anyhow::anyhow!("No supported runtime available"))
        }
    }
}