multiplexed-connection 0.4.0

Creates connection with multiple data channels over single async data stream.
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

use std::pin::Pin;
use std::task::{Poll, Context};
use futures::ready;

use tokio::io::AsyncWrite;

use connection_utils::Channel;
use super::RpcChannel;

impl AsyncWrite for RpcChannel {
    fn poll_write(
        mut self: Pin<&mut Self>,
        cx: &mut Context<'_>,
        buf: &[u8],
    ) -> Poll<Result<usize, std::io::Error>> {
        if self.write_async_future.is_none() {
            let fut = self.rpc_client.as_ref().send_channel_data(self.id(), buf.to_vec());
            self.write_async_future = Some(Box::pin(fut));
        }

        return self.poll_async_write_future(cx);
    }

    fn poll_flush(self: std::pin::Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), std::io::Error>> {
        if self.write_async_future.is_some() {
            ready!(self.poll_async_write_future(cx))?;
        }

        return Poll::Ready(Ok(()));
    }

    fn poll_shutdown(self: std::pin::Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), std::io::Error>> {
        if self.write_async_future.is_some() {
            ready!(self.poll_async_write_future(cx))?;
        }

        return Poll::Ready(Ok(()));
    }
}