azservicebus 0.25.1

An unofficial AMQP 1.0 rust client for Azure Service Bus
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 fe2o3_amqp_management::{error::InvalidType, response::Response};
use fe2o3_amqp_types::{
    messaging::{message::__private::Deserializable, Body, Message},
    primitives::{Binary, OrderedMap, Uuid},
};
use serde_amqp::Value;

use crate::{
    amqp::management_constants::properties::{LOCK_TOKEN, MESSAGE, MESSAGES},
    primitives::service_bus_received_message::{
        ReceivedMessageLockToken, ServiceBusReceivedMessage,
    },
};

type DeferredMessage = OrderedMap<String, Value>;
type ListOfDeferredMessages = Vec<DeferredMessage>;
type ReceiveBySequenceNumberResponseBody = OrderedMap<String, ListOfDeferredMessages>;

#[derive(Debug)]
pub(crate) struct ReceiveBySequenceNumberResponse {
    pub deferred_messages: Vec<(Uuid, Binary)>,
}

impl ReceiveBySequenceNumberResponse {
    pub fn into_received_messages(
        self,
    ) -> Result<Vec<ServiceBusReceivedMessage>, serde_amqp::Error> {
        let mut received_messages = Vec::with_capacity(self.deferred_messages.len());
        for (lock_token, buf) in self.deferred_messages {
            let raw_amqp_message: Deserializable<Message<Body<Value>>> =
                serde_amqp::from_slice(&buf)?;
            let message = ServiceBusReceivedMessage {
                _is_settled: false,
                raw_amqp_message: raw_amqp_message.0,
                lock_token: ReceivedMessageLockToken::LockToken(lock_token),
            };
            received_messages.push(message);
        }

        Ok(received_messages)
    }
}

impl Response for ReceiveBySequenceNumberResponse {
    const STATUS_CODE: u16 = 200;

    type Body = Option<ReceiveBySequenceNumberResponseBody>;

    type Error = super::ManagementError;

    fn decode_message(
        message: fe2o3_amqp_types::messaging::Message<Self::Body>,
    ) -> Result<Self, Self::Error> {
        let messages = message
            .body
            .ok_or(super::ManagementError::DecodeError(None))?
            .swap_remove(MESSAGES)
            .ok_or_else(|| InvalidType {
                expected: MESSAGES.into(),
                actual: "None".into(),
            })?;

        let deferred_messages = messages
            .into_iter()
            .map(|mut map| {
                let uuid = map.swap_remove(LOCK_TOKEN).map(Uuid::try_from);
                let message = map.swap_remove(MESSAGE).map(Binary::try_from);
                match (uuid, message) {
                    (Some(Ok(uuid)), Some(Ok(message))) => Some((uuid, message)),
                    _ => None,
                }
            })
            .collect::<Option<Vec<_>>>()
            .ok_or(super::ManagementError::DecodeError(None))?;

        Ok(Self { deferred_messages })
    }
}