wtx 0.50.0

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

use crate::{collections::MaybeUninitSlice, stream::StreamCommon};
use core::num::NonZeroUsize;

/// A stream of values sent asynchronously.
pub trait StreamReader: StreamCommon {
  /// Pulls some bytes from this source into the specified buffer, returning how many bytes
  /// were read.
  fn read(
    &mut self,
    bytes: MaybeUninitSlice<'_, u8>,
  ) -> impl Future<Output = crate::Result<Option<NonZeroUsize>>>;

  /// Reads and at the same time discards exactly `len` bytes.
  #[inline]
  fn read_skip(&mut self, len: usize) -> impl Future<Output = crate::Result<()>> {
    async move {
      let mut buffer = [0; 32];
      let mut counter = len;
      for _ in 0..len {
        if counter == 0 {
          break;
        }
        let slice = if let Some(el) = buffer.get_mut(..counter) { el } else { &mut buffer[..] };
        let Some(read) = self.read(slice.into()).await? else {
          return Err(crate::Error::UnexpectedStreamReadEOF);
        };
        counter = counter.wrapping_sub(read.get());
      }
      Ok(())
    }
  }
}

impl<T> StreamReader for &mut T
where
  T: StreamReader,
{
  #[inline]
  async fn read(&mut self, bytes: MaybeUninitSlice<'_, u8>) -> crate::Result<Option<NonZeroUsize>> {
    (**self).read(bytes).await
  }
}

impl StreamReader for () {
  #[inline]
  async fn read(&mut self, _: MaybeUninitSlice<'_, u8>) -> crate::Result<Option<NonZeroUsize>> {
    Ok(None)
  }
}