rumqttc/mqttbytes/v4/
unsubscribe.rs1use super::*;
2use bytes::{Buf, Bytes};
3
4#[derive(Debug, Clone, PartialEq, Eq)]
6pub struct Unsubscribe {
7 pub pkid: u16,
8 pub topics: Vec<String>,
9}
10
11impl Unsubscribe {
12 pub fn new<S: Into<String>>(topic: S) -> Unsubscribe {
13 Unsubscribe {
14 pkid: 0,
15 topics: vec![topic.into()],
16 }
17 }
18
19 fn len(&self) -> usize {
20 2 + self.topics.iter().fold(0, |s, t| s + t.len() + 2)
22 }
23
24 pub fn size(&self) -> usize {
25 let len = self.len();
26 let remaining_len_size = len_len(len);
27
28 1 + remaining_len_size + len
29 }
30
31 pub fn read(fixed_header: FixedHeader, mut bytes: Bytes) -> Result<Self, Error> {
32 let variable_header_index = fixed_header.fixed_header_len;
33 bytes.advance(variable_header_index);
34
35 let pkid = read_u16(&mut bytes)?;
36 let mut payload_bytes = fixed_header.remaining_len - 2;
37 let mut topics = Vec::with_capacity(1);
38
39 while payload_bytes > 0 {
40 let topic_filter = read_mqtt_string(&mut bytes)?;
41 payload_bytes -= topic_filter.len() + 2;
42 topics.push(topic_filter);
43 }
44
45 let unsubscribe = Unsubscribe { pkid, topics };
46 Ok(unsubscribe)
47 }
48
49 pub fn write(&self, payload: &mut BytesMut) -> Result<usize, Error> {
50 let remaining_len = self.len();
51
52 payload.put_u8(0xA2);
53 let remaining_len_bytes = write_remaining_length(payload, remaining_len)?;
54 payload.put_u16(self.pkid);
55
56 for topic in self.topics.iter() {
57 write_mqtt_string(payload, topic.as_str());
58 }
59 Ok(1 + remaining_len_bytes + remaining_len)
60 }
61}