wreq-rt 0.2.2-rc.2

Runtime utilities for wreq
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

//! A `Send + Future` wrapper for non-`Send` futures (e.g. compio futures).

use std::{
    pin::Pin,
    task::{Context, Poll},
};

/// A `Send + Future` wrapper for non-`Send` futures (e.g. compio futures).
///
/// Uses `SendWrapper` to assert `Send` at compile time, with runtime checks
/// that panic on cross-thread access.
pub struct SendFuture<F>(send_wrapper::SendWrapper<F>);

impl<F: Future> SendFuture<F> {
    /// Wraps a non-`Send` future in a `SendFuture`.
    pub fn new(fut: F) -> Self {
        Self(send_wrapper::SendWrapper::new(fut))
    }
}

impl<F: Future> Future for SendFuture<F> {
    type Output = F::Output;

    #[allow(unsafe_code)]
    fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
        let this = unsafe { self.get_unchecked_mut() };
        let inner: Pin<&mut F> = unsafe { Pin::new_unchecked(&mut *this.0) };
        inner.poll(cx)
    }
}