wtx 0.52.1

A collection of different transport implementations and related tools focused primarily on web technologies.
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use crate::{
  executor::{
    Executor, ExecutorTy, Runtime, TcpListener, TcpStream, resolve_addrs, tcp_listener_std,
  },
  net::{TcpParams, ToSocketAddrs},
};
use core::{
  net::SocketAddr,
  pin::Pin,
  task::{Context, Poll, ready},
};
use tokio::runtime::{Builder, LocalOptions};

/// Uses the structures originated from the `tokio` project.
#[derive(Clone, Debug, Default)]
pub struct TokioExecutor {}

impl Executor for TokioExecutor {
  const TY: ExecutorTy = ExecutorTy::Tokio;

  type LocalRuntime = tokio::runtime::LocalRuntime;
  type SpawnFuture<T> = TokioSpawnFutureFuture<T>;
  type SpawnLocalFuture<T> = TokioSpawnFutureFuture<T>;
  type TcpListener = tokio::net::TcpListener;
  type TcpStream = tokio::net::TcpStream;

  #[inline]
  fn spawn<F>(&self, future: F) -> Self::SpawnFuture<F::Output>
  where
    F: Future + Send + 'static,
    F::Output: Send + 'static,
  {
    TokioSpawnFutureFuture(tokio::task::spawn(future))
  }

  #[inline]
  fn spawn_local<F>(&self, future: F, _: &Self::LocalRuntime) -> Self::SpawnLocalFuture<F::Output>
  where
    F: Future + 'static,
    F::Output: 'static,
  {
    TokioSpawnFutureFuture(tokio::task::spawn_local(future))
  }
}

impl TcpListener for tokio::net::TcpListener {
  type TcpStream = tokio::net::TcpStream;

  #[inline]
  async fn bind<A>(addr: A, tcp_params: TcpParams) -> crate::Result<Self>
  where
    A: ToSocketAddrs,
  {
    let tcp_listener = tcp_listener_std(addr, &TokioExecutor::default(), tcp_params).await?;
    tcp_listener.set_nonblocking(true)?;
    Ok(tokio::net::TcpListener::from_std(tcp_listener)?)
  }

  #[inline]
  async fn accept(&self, tcp_params: TcpParams) -> crate::Result<(Self::TcpStream, SocketAddr)> {
    let rslt = (*self).accept().await?;
    rslt.0.set_nodelay(tcp_params.tcp_nodelay())?;
    Ok(rslt)
  }
}

impl Runtime for tokio::runtime::LocalRuntime {
  #[inline]
  fn new() -> crate::Result<Self> {
    Ok(Builder::new_current_thread().enable_all().build_local(LocalOptions::default())?)
  }

  #[inline]
  fn block_on<F>(&self, future: F) -> F::Output
  where
    F: Future,
  {
    (*self).block_on(future)
  }
}

impl TcpStream for tokio::net::TcpStream {
  type Executor = TokioExecutor;

  #[inline]
  async fn connect<A>(addr: A, tcp_params: TcpParams) -> crate::Result<Self>
  where
    A: ToSocketAddrs,
  {
    let stream = resolve_addrs(addr, &TokioExecutor::default(), async |socket_addr| {
      Ok(tokio::net::TcpStream::connect(socket_addr).await?)
    })
    .await?;
    stream.set_nodelay(tcp_params.tcp_nodelay())?;
    Ok(stream)
  }

  #[inline]
  fn peer_addr(&self) -> crate::Result<SocketAddr> {
    Ok((*self).peer_addr()?)
  }
}

/// Returned by [`TokioExecutor::spawn`].
#[derive(Debug)]
pub struct TokioSpawnFutureFuture<T>(tokio::task::JoinHandle<T>);

impl<T> Future for TokioSpawnFutureFuture<T> {
  type Output = crate::Result<T>;

  #[inline]
  fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
    Poll::Ready(Ok(ready!(Pin::new(&mut self.0).poll(cx))?))
  }
}