async_codec_lite/framed/
write.rs

1use super::inner::{FramedInner, WriteFrame};
2use crate::{codec::Encoder, error::Error};
3pub use bytes::{Bytes, BytesMut};
4use futures_core::Stream;
5use futures_io::AsyncWrite;
6use futures_sink::Sink;
7use pin_project_lite::pin_project;
8use std::{
9    fmt,
10    pin::Pin,
11    task::{Context, Poll},
12};
13
14pin_project! {
15    pub struct FramedWrite<T, E> {
16        #[pin]
17        inner: FramedInner<T, E, WriteFrame>,
18    }
19}
20
21impl<T, E> FramedWrite<T, E>
22where
23    T: AsyncWrite,
24{
25    pub fn new(inner: T, encoder: E) -> FramedWrite<T, E> {
26        FramedWrite {
27            inner: FramedInner {
28                inner,
29                codec: encoder,
30                state: WriteFrame::default(),
31            },
32        }
33    }
34}
35
36impl<T, E> FramedWrite<T, E> {
37    pub fn get_ref(&self) -> &T {
38        &self.inner.inner
39    }
40
41    pub fn get_mut(&mut self) -> &mut T {
42        &mut self.inner.inner
43    }
44
45    pub fn get_pin_mut(self: Pin<&mut Self>) -> Pin<&mut T> {
46        self.project().inner.project().inner
47    }
48
49    pub fn into_inner(self) -> T {
50        self.inner.inner
51    }
52
53    pub fn encoder(&self) -> &E {
54        &self.inner.codec
55    }
56
57    pub fn encoder_mut(&mut self) -> &mut E {
58        &mut self.inner.codec
59    }
60}
61
62impl<T, E> Sink<E::Item> for FramedWrite<T, E>
63where
64    T: AsyncWrite,
65    E: Encoder,
66{
67    type Error = Error<E::Error>;
68
69    fn poll_ready(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
70        self.project().inner.poll_ready(cx)
71    }
72
73    fn start_send(self: Pin<&mut Self>, item: E::Item) -> Result<(), Self::Error> {
74        self.project().inner.start_send(item)
75    }
76
77    fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
78        self.project().inner.poll_flush(cx)
79    }
80
81    fn poll_close(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
82        self.project().inner.poll_close(cx)
83    }
84}
85
86impl<T, D> Stream for FramedWrite<T, D>
87where
88    T: Stream,
89{
90    type Item = T::Item;
91
92    fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
93        self.project().inner.project().inner.poll_next(cx)
94    }
95}
96
97impl<T, U> fmt::Debug for FramedWrite<T, U>
98where
99    T: fmt::Debug,
100    U: fmt::Debug,
101{
102    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
103        f.debug_struct("FramedWrite")
104            .field("inner", &self.get_ref())
105            .field("encoder", &self.encoder())
106            .field("buffer", &self.inner.state.buffer)
107            .finish()
108    }
109}
async_codec_lite/framed/write.rs

async_codec_lite/framed/
write.rs
