wtx 0.47.7

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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use crate::{
  http::{MsgBufferString, MsgData, Request, StatusCode, u31::U31},
  http2::{
    CommonStream, Http2Buffer, Http2Inner, Http2RecvStatus, Http2SendStatus,
    hpack_static_headers::{HpackStaticRequestHeaders, HpackStaticResponseHeaders},
    misc::{manage_recurrent_receiving_of_overall_stream, process_higher_operation_err},
    stream_receiver::StreamOverallRecvParams,
    stream_state::StreamState,
    window::Windows,
    write_functions::send_msg,
  },
  misc::{Lease, LeaseMut, span::Span},
  stream::StreamWriter,
  sync::Arc,
};
use core::{future::poll_fn, pin::pin, task::Waker};

/// Groups the methods used by clients that connect to servers.
#[derive(Debug)]
pub struct ClientStream<HB, SW> {
  inner: Arc<Http2Inner<HB, SW, true>>,
  linger: bool,
  span: Span,
  stream_id: U31,
  // Used after the initial sending
  windows: Windows,
}

impl<HB, SW> ClientStream<HB, SW> {
  pub(crate) const fn new(
    inner: Arc<Http2Inner<HB, SW, true>>,
    linger: bool,
    span: Span,
    stream_id: U31,
  ) -> Self {
    Self { inner, linger, span, stream_id, windows: Windows::new() }
  }
}

impl<HB, SW> ClientStream<HB, SW>
where
  HB: LeaseMut<Http2Buffer>,
  SW: StreamWriter,
{
  /// See [`CommonStream`].
  #[inline]
  pub const fn common(&mut self) -> CommonStream<'_, HB, SW, true> {
    let Self { inner, linger, span, stream_id, windows: _ } = self;
    CommonStream { inner, linger: *linger, span, stream_id: *stream_id }
  }

  /// Receive response
  ///
  /// High-level operation that awaits for the data necessary to build a response and then closes the
  /// stream.
  ///
  /// Returns [`Option::None`] if the network/stream connection has been closed, either locally
  /// or externally.
  ///
  /// Should be called after [`Self::send_req`] or any other low level methods that send data
  /// are successfully executed. More specifically, should only be called in a half-closed stream
  /// state.
  #[inline]
  pub async fn recv_res(
    &mut self,
  ) -> crate::Result<(Http2RecvStatus<StatusCode, ()>, MsgBufferString)> {
    let Self { inner, linger: _, span, stream_id, windows: _ } = self;
    let _e = span.enter();
    _trace!("Receiving response");
    let mut lock_pin = pin!(inner.hd.lock());
    let rslt = poll_fn(|cx| {
      let mut lock = lock_pin!(cx, inner.hd, lock_pin);
      let hdpm = lock.parts_mut();
      manage_recurrent_receiving_of_overall_stream(
        cx,
        hdpm,
        &inner.is_conn_open,
        *stream_id,
        |_, status_code, _, _| status_code,
      )
    })
    .await;
    if let Err(err) = &rslt {
      process_higher_operation_err(err, inner).await;
    }
    rslt
  }

  /// Send Request
  ///
  /// Sends all data related to a request.
  ///
  /// Returns [`Option::None`] if the network/stream connection has been closed, either locally
  /// or externally.
  ///
  /// Shouldn't be called more than once.
  #[inline]
  pub async fn send_req<MD>(&mut self, req: Request<MD>) -> crate::Result<Http2SendStatus>
  where
    MD: MsgData,
    MD::Body: Lease<[u8]>,
  {
    let Self { inner, linger: _, span, stream_id, windows } = self;
    let _e = span.enter();
    _trace!("Sending request");
    let uri = req.msg_data.uri();
    send_msg::<_, _, true>(
      req.msg_data.body().lease(),
      req.msg_data.headers(),
      inner,
      (
        HpackStaticRequestHeaders {
          authority: uri.authority(),
          method: Some(req.method),
          path: uri.relative_reference_slash(),
          protocol: None,
          scheme: uri.scheme(),
        },
        HpackStaticResponseHeaders::EMPTY,
      ),
      *stream_id,
      |hdpm| {
        if let Some(scrp) = hdpm.hb.scrps.remove(stream_id) {
          *windows = scrp.windows;
        }
        drop(hdpm.hb.sorps.insert(
          *stream_id,
          StreamOverallRecvParams {
            body_len: 0,
            content_length: None,
            has_initial_header: false,
            has_one_or_more_data_frames: false,
            is_stream_open: true,
            msg_buffer: MsgBufferString::default(),
            status_code: StatusCode::Ok,
            stream_state: StreamState::HalfClosedLocal,
            // The possible future invocation of this waker by the reading task won't be a problem
            // because users will manually call `recv_res`.
            waker: Waker::noop().clone(),
            windows: *windows,
          },
        ));
      },
    )
    .await
  }
}