pulsar_client/session/
raw_create_consumer.rs

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
use pulsar_binary_protocol_spec::{
    client_channel_messages::{
        handler_reply_session_channel_message::HandlerReplySessionCreateConsumerChannelMessage,
        SessionSendHandlerChannelMessage,
    },
    client_responds::SessionCreateConsumerRespondError,
    futures_channel::oneshot::channel,
    SubscribeCommand,
};
use thiserror::Error;

use crate::consumer::AsyncConsumer;

use super::AsyncSession;

#[derive(Error, Debug)]
pub enum RawCreateConsumerError {
    #[error("SessionChannelClosed")]
    SessionChannelClosed,
    #[error("RespondError {0:?}")]
    RespondError(SessionCreateConsumerRespondError),
    #[error("ChannelClosed")]
    ChannelClosed,
}
impl AsyncSession {
    pub async fn raw_create_consumer(
        &self,
        subscribe_command: SubscribeCommand,
    ) -> Result<AsyncConsumer, RawCreateConsumerError> {
        let (sender, receiver) = channel::<HandlerReplySessionCreateConsumerChannelMessage>();

        self.sender
            .send(SessionSendHandlerChannelMessage::CreateConsumer(
                subscribe_command,
                sender,
            ))
            .await
            .map_err(|_| RawCreateConsumerError::SessionChannelClosed)?;

        match receiver.await {
            Ok(Ok((subscribe_command, success_command, s))) => {
                Ok(AsyncConsumer::new(s, subscribe_command, success_command))
            }
            Ok(Err(err)) => Err(RawCreateConsumerError::RespondError(err)),
            Err(_) => Err(RawCreateConsumerError::ChannelClosed),
        }
    }
}