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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
use std::convert::TryFrom;
use std::num::NonZeroU16;
use bytes::Bytes;
use bytestring::ByteString;
use ntex::router::Path;
use serde::de::DeserializeOwned;
use serde_json::Error as JsonError;
use crate::v3::codec;
pub struct Publish {
publish: codec::Publish,
topic: Path<ByteString>,
query: Option<ByteString>,
}
impl Publish {
pub(crate) fn new(publish: codec::Publish) -> Self {
let (topic, query) = if let Some(pos) = publish.topic.find('?') {
(
ByteString::try_from(publish.topic.get_ref().slice(0..pos)).unwrap(),
Some(
ByteString::try_from(
publish.topic.get_ref().slice(pos + 1..publish.topic.len()),
)
.unwrap(),
),
)
} else {
(publish.topic.clone(), None)
};
let topic = Path::new(topic);
Self { publish, topic, query }
}
#[inline]
pub fn dup(&self) -> bool {
self.publish.dup
}
#[inline]
pub fn retain(&self) -> bool {
self.publish.retain
}
#[inline]
pub fn qos(&self) -> codec::QoS {
self.publish.qos
}
#[inline]
pub fn publish_topic(&self) -> &str {
&self.publish.topic
}
#[inline]
pub fn id(&self) -> Option<NonZeroU16> {
self.publish.packet_id
}
#[inline]
pub fn topic(&self) -> &Path<ByteString> {
&self.topic
}
#[inline]
pub fn topic_mut(&mut self) -> &mut Path<ByteString> {
&mut self.topic
}
#[inline]
pub fn query(&self) -> &str {
self.query.as_ref().map(|s| s.as_ref()).unwrap_or("")
}
#[inline]
pub fn packet(&self) -> &codec::Publish {
&self.publish
}
#[inline]
pub fn packet_mut(&mut self) -> &mut codec::Publish {
&mut self.publish
}
#[inline]
pub fn payload(&self) -> &Bytes {
&self.publish.payload
}
pub fn take_payload(&self) -> Bytes {
self.publish.payload.clone()
}
pub fn json<T: DeserializeOwned>(&mut self) -> Result<T, JsonError> {
serde_json::from_slice(&self.publish.payload)
}
pub(super) fn into_inner(self) -> codec::Publish {
self.publish
}
}
impl std::fmt::Debug for Publish {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
self.publish.fmt(f)
}
}