use crate::{
types::{LongLongUInt, LongUInt, ShortString, ShortUInt},
BasicProperties, Result,
};
pub type DeliveryResult = Result<Option<Delivery>>;
#[derive(Clone, Debug, PartialEq)]
pub struct Delivery {
pub delivery_tag: LongLongUInt,
pub exchange: ShortString,
pub routing_key: ShortString,
pub redelivered: bool,
pub properties: BasicProperties,
pub data: Vec<u8>,
}
impl Delivery {
pub(crate) fn new(
delivery_tag: LongLongUInt,
exchange: ShortString,
routing_key: ShortString,
redelivered: bool,
) -> Self {
Self {
delivery_tag,
exchange,
routing_key,
redelivered,
properties: BasicProperties::default(),
data: Vec::new(),
}
}
pub(crate) fn receive_content(&mut self, data: Vec<u8>) {
self.data.extend(data);
}
}
#[derive(Clone, Debug, PartialEq)]
pub struct BasicGetMessage {
pub delivery: Delivery,
pub message_count: LongUInt,
}
impl BasicGetMessage {
pub(crate) fn new(
delivery_tag: LongLongUInt,
exchange: ShortString,
routing_key: ShortString,
redelivered: bool,
message_count: LongUInt,
) -> Self {
Self {
delivery: Delivery::new(delivery_tag, exchange, routing_key, redelivered),
message_count,
}
}
}
#[derive(Clone, Debug, PartialEq)]
pub struct BasicReturnMessage {
pub delivery: Delivery,
pub reply_code: ShortUInt,
pub reply_text: ShortString,
}
impl BasicReturnMessage {
pub(crate) fn new(
exchange: ShortString,
routing_key: ShortString,
reply_code: ShortUInt,
reply_text: ShortString,
) -> Self {
Self {
delivery: Delivery::new(0, exchange, routing_key, false),
reply_code,
reply_text,
}
}
}