use std::{mem, num::NonZeroU16};
use ntex_bytes::{ByteString, Bytes};
use ntex_router::Path;
use crate::{error::PayloadError, payload::Payload, v3::codec};
#[derive(Debug)]
pub struct Publish {
pkt: codec::Publish,
pkt_size: u32,
topic: Path<ByteString>,
payload: Payload,
}
impl Publish {
#[doc(hidden)]
pub fn new(pkt: codec::Publish, payload: Payload, pkt_size: u32) -> Self {
Self { topic: Path::new(pkt.topic.clone()), pkt, pkt_size, payload }
}
#[inline]
pub fn dup(&self) -> bool {
self.pkt.dup
}
#[inline]
pub fn retain(&self) -> bool {
self.pkt.retain
}
#[inline]
pub fn qos(&self) -> codec::QoS {
self.pkt.qos
}
#[inline]
pub fn publish_topic(&self) -> &str {
&self.pkt.topic
}
#[inline]
pub fn id(&self) -> Option<NonZeroU16> {
self.pkt.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 packet(&self) -> &codec::Publish {
&self.pkt
}
#[inline]
pub fn packet_mut(&mut self) -> &mut codec::Publish {
&mut self.pkt
}
#[inline]
pub fn packet_size(&self) -> u32 {
self.pkt_size
}
#[inline]
pub fn payload_size(&self) -> usize {
self.pkt.payload_size as usize
}
#[inline]
pub async fn read(&self) -> Result<Option<Bytes>, PayloadError> {
self.payload.read().await
}
#[inline]
pub async fn read_all(&self) -> Result<Bytes, PayloadError> {
self.payload.read_all().await
}
pub fn take_payload(&mut self) -> Payload {
mem::take(&mut self.payload)
}
pub(super) fn into_inner(self) -> (codec::Publish, Payload, u32) {
(self.pkt, self.payload, self.pkt_size)
}
}