kcl-async 0.1.0

Async KCL MultiLang Daemon library
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

use super::{
    message::input::Message as MessageIn, message::output::Message as MessageOut,
    transport::Transport,
};

#[derive(Debug, thiserror::Error)]
pub enum CheckpointError<TransportError> {
    #[error(transparent)]
    TransportError(TransportError),

    #[error("failed to checkpoint: {reason}")]
    Failed { reason: String },

    #[error("invalid state: {}", message.id())]
    InvalidState { message: MessageIn },
}

pub struct Checkpointer<'a, T>(&'a mut T);

impl<'a, T> Checkpointer<'a, T>
where
    T: Transport,
{
    pub(crate) fn new(transport: &'a mut T) -> Self {
        Self(transport)
    }

    pub async fn checkpoint(
        &mut self,
        sequence_number: Option<String>,
        sub_sequence_number: Option<u64>,
    ) -> Result<(), CheckpointError<T::Error>> {
        let request = MessageOut::Checkpoint(super::message::output::CheckpointMessage {
            sequence_number,
            sub_sequence_number,
        });

        self.0
            .write_message(&request)
            .await
            .map_err(CheckpointError::TransportError)?;

        let response = self
            .0
            .read_message()
            .await
            .map_err(CheckpointError::TransportError)?;

        if let MessageIn::Checkpoint(msg) = response {
            if let Some(error) = msg.error {
                Err(CheckpointError::Failed { reason: error })
            } else {
                Ok(())
            }
        } else {
            Err(CheckpointError::InvalidState { message: response })
        }
    }
}