mqrstt 0.4.2

Pure rust MQTTv5 client implementation Smol and Tokio
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

use tokio::io::AsyncWriteExt;

#[cfg(feature = "logs")]
use tracing::trace;

use crate::packets::ConnAck;
use crate::packets::{ConnAckReasonCode, Packet};
use crate::{connect_options::ConnectOptions, error::ConnectionError};

pub(crate) trait StreamExt {
    fn connect(&mut self, options: &ConnectOptions) -> impl std::future::Future<Output = Result<ConnAck, ConnectionError>>;
    fn read_packet(&mut self) -> impl std::future::Future<Output = Result<Packet, ConnectionError>>;
    fn write_packet(&mut self, packet: &Packet) -> impl std::future::Future<Output = Result<(), ConnectionError>>;
    fn write_packets(&mut self, packets: &[Packet]) -> impl std::future::Future<Output = Result<(), ConnectionError>>;
    fn flush_packets(&mut self) -> impl std::future::Future<Output = std::io::Result<()>>;
}

impl<S> StreamExt for S
where
    S: tokio::io::AsyncRead + tokio::io::AsyncWrite + Sized + Unpin,
{
    async fn connect(&mut self, options: &ConnectOptions) -> Result<ConnAck, ConnectionError> {
        let connect = options.create_connect_from_options();

        self.write_packet(&connect).await?;

        let packet = Packet::async_read(self).await?;
        if let Packet::ConnAck(con) = packet {
            if con.reason_code == ConnAckReasonCode::Success {
                #[cfg(feature = "logs")]
                trace!("Connected to server");
                Ok(con)
            } else {
                Err(ConnectionError::ConnectionRefused(con.reason_code))
            }
        } else {
            Err(ConnectionError::NotConnAck(packet))
        }
    }

    async fn read_packet(&mut self) -> Result<Packet, ConnectionError> {
        Ok(Packet::async_read(self).await?)
    }

    async fn write_packet(&mut self, packet: &Packet) -> Result<(), ConnectionError> {
        match packet.async_write(self).await {
            Ok(_) => (),
            Err(err) => {
                return match err {
                    crate::packets::error::WriteError::SerializeError(serialize_error) => Err(ConnectionError::SerializationError(serialize_error)),
                    crate::packets::error::WriteError::IoError(error) => Err(ConnectionError::Io(error)),
                };
            }
        }

        #[cfg(feature = "logs")]
        trace!("Sending packet {}", packet);

        self.flush().await?;
        // self.flush_packets().await?;

        Ok(())
    }

    async fn write_packets(&mut self, packets: &[Packet]) -> Result<(), ConnectionError> {
        for packet in packets {
            let _ = packet.async_write(self).await;
            #[cfg(feature = "logs")]
            trace!("Sending packet {}", packet);
        }
        self.flush_packets().await?;
        Ok(())
    }

    fn flush_packets(&mut self) -> impl std::future::Future<Output = std::io::Result<()>> {
        tokio::io::AsyncWriteExt::flush(self)
    }
}