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

//! Personal collection of async utilities
#![warn(missing_docs)]
#![cfg_attr(feature = "cargo-clippy", allow(clippy::style))]

///Await future in async context.
///
///Because `.await` is retarded.
///
///```rust
///#![feature(async_await)]
///
///async fn do_async() {
///}
///
///async fn my_main() {
///    cute_async::matsu!(do_async());
///}
///
///```
#[macro_export]
macro_rules! matsu {
    ($exp:expr) => {
        ($exp).await
    }
}

///Unreachable optimization hint.
///
///In debug mod it panics using `unreachable!` macro
///But hitting this macro in release mode  would result in UB.
#[macro_export]
macro_rules! unreach {
    () => ({
        #[cfg(not(debug_assertions))]
        unsafe {
            std::hint::unreachable_unchecked();
        }
        #[cfg(debug_assertions)]
        {
            unreachable!()
        }
    })
}

///Gets `Pin` out of value
///
///## Usage:
///
///```rust,no_run
///use cute_async::AsPin;
///
///use core::task;
///use core::pin::Pin;
///use core::future::Future;
///
///pub struct MyFuture<T>(T);
///
///impl<T: AsPin + Unpin + Future> Future for MyFuture<T> {
///    type Output = T::Output;
///
///    fn poll(mut self: Pin<&mut Self>, ctx: &mut task::Context<'_>) -> task::Poll<Self::Output> {
///        let inner = self.0.as_pin();
///        Future::poll(inner, ctx)
///    }
///}
///```

pub trait AsPin: core::ops::Deref {
    ///Gets `Pin` out of self.
    fn as_pin(&mut self) -> core::pin::Pin<&'_ mut Self>;
}

impl<T: core::ops::Deref> AsPin for T {
    #[inline(always)]
    fn as_pin(&mut self) -> core::pin::Pin<&'_ mut Self> {
        unsafe {
            core::pin::Pin::new_unchecked(self)
        }
    }
}

pub mod fut;
pub use fut::*;
pub mod adaptors;
pub use adaptors::*;
#[cfg(feature = "tokio")]
pub mod runtime;