mutant-client 0.6.2

WebSocket client library for MutAnt distributed mutable key value storage over Autonomi network
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

#[macro_export]
macro_rules! direct_request {
    ($self:expr, $key:ident, $($req:tt)*) => {{
        let key = PendingRequestKey::$key;
        let req = Request::$key(mutant_protocol::$($req)*);

        if $self.pending_requests.lock().unwrap().contains_key(&key) {
            return Err(ClientError::InternalError(
                "Another list_keys request is already pending".to_string(),
            ));
        } else {
            let (sender, receiver) = oneshot::channel();
            let pending_sender = PendingSender::$key(sender);

            $self
                .pending_requests
                .lock()
                .unwrap()
                .insert(key.clone(), pending_sender);

            match $self.send_request(req).await {
                Ok(_) => {
                    debug!("{} request sent, waiting for response...", stringify!($key));
                    match receiver.await {
                        Ok(result) => result,
                        Err(_) => {
                            $self.pending_requests.lock().unwrap().remove(&key);
                            error!("{} response channel canceled", stringify!($key));
                            Err(ClientError::InternalError(
                                "{} response channel canceled".to_string(),
                            ))
                        }
                    }
                }
                Err(e) => {
                    $self.pending_requests.lock().unwrap().remove(&key);
                    error!("Failed to send {} request: {:?}", stringify!($key), e);
                    Err(e)
                }
            }
        }
    }};
}

#[macro_export]
macro_rules! long_request {
    ($self:expr, $key:ident, $($req:tt)*) => {{
        let key = PendingRequestKey::TaskCreation;
        let req = Request::$key(mutant_protocol::$($req)*);
        if $self.pending_requests.lock().unwrap().contains_key(&key) {
            return Err(ClientError::InternalError(
                "Another put/get request is already pending".to_string(),
            ));
        }

        let (completion_tx, completion_rx) = oneshot::channel();
        let (progress_tx, progress_rx) = mpsc::unbounded_channel();

        let (task_creation_tx, task_creation_rx) = oneshot::channel();
        $self.pending_requests.lock().unwrap().insert(
            key.clone(),
            PendingSender::TaskCreation(
                task_creation_tx,
                (completion_tx, progress_tx),
                TaskType::$key,
            ),
        );

        let start_task = async move {
            match $self.send_request(req).await {
                Ok(_) => {
                    debug!("{} request sent, waiting for TaskCreated response...", stringify!($key));
                    let task_id = task_creation_rx.await.map_err(|_| {
                        ClientError::InternalError("TaskCreated channel canceled".to_string())
                    })??;

                    info!("Task created with ID: {}", task_id);

                    completion_rx.await.map_err(|_| {
                        error!("Completion channel canceled");
                        ClientError::InternalError("Completion channel canceled".to_string())
                    })?
                }
                Err(e) => {
                    error!("Failed to send {} request: {:?}", stringify!($key), e);
                    $self.pending_requests.lock().unwrap().remove(&key);
                    Err(e)
                }
            }
        };

        Ok((start_task, progress_rx))

    }}
}