This commit is contained in:
Toshit Chawda 2024-09-26 19:21:09 -07:00
parent b602b38c47
commit 5e4461d61d
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D

View file

@ -477,12 +477,12 @@ impl Stream for MuxStreamIo {
} }
} }
impl Sink<&[u8]> for MuxStreamIo { impl Sink<BytesMut> for MuxStreamIo {
type Error = std::io::Error; type Error = std::io::Error;
fn poll_ready(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> { fn poll_ready(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
self.project().tx.poll_ready(cx) self.project().tx.poll_ready(cx)
} }
fn start_send(self: Pin<&mut Self>, item: &[u8]) -> Result<(), Self::Error> { fn start_send(self: Pin<&mut Self>, item: BytesMut) -> Result<(), Self::Error> {
self.project().tx.start_send(item) self.project().tx.start_send(item)
} }
fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> { fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
@ -731,7 +731,7 @@ impl AsyncWrite for MuxStreamAsyncWrite {
let mut this = self.as_mut().project(); let mut this = self.as_mut().project();
ready!(this.tx.as_mut().poll_ready(cx))?; ready!(this.tx.as_mut().poll_ready(cx))?;
match this.tx.as_mut().start_send(buf) { match this.tx.as_mut().start_send(buf.into()) {
Ok(()) => { Ok(()) => {
let mut cx = Context::from_waker(noop_waker_ref()); let mut cx = Context::from_waker(noop_waker_ref());
let cx = &mut cx; let cx = &mut cx;