etcd-rs 1.0.1

etcd client for rust
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

//! Leases are a mechanism for detecting client liveness. The cluster grants leases with a time-to-live. A lease expires if the etcd cluster does not receive a keepAlive within a given TTL period.
//!
//! # Examples
//!
//! Grant lease and keep lease alive

mod grant;
mod keep_alive;
mod revoke;
mod time_to_live;

pub use grant::{LeaseGrantRequest, LeaseGrantResponse};
pub use keep_alive::{LeaseKeepAliveRequest, LeaseKeepAliveResponse};
pub use revoke::{LeaseRevokeRequest, LeaseRevokeResponse};
pub use time_to_live::{LeaseTimeToLiveRequest, LeaseTimeToLiveResponse};

use async_trait::async_trait;
use tokio::sync::mpsc::Sender;
use tonic::Streaming;

use crate::{Error, Result};

pub type LeaseId = i64;

#[async_trait]
pub trait LeaseOp {
    async fn grant_lease<R>(&self, req: R) -> Result<LeaseGrantResponse>
    where
        R: Into<LeaseGrantRequest> + Send;

    async fn revoke<R>(&self, req: R) -> Result<LeaseRevokeResponse>
    where
        R: Into<LeaseRevokeRequest> + Send;

    async fn keep_alive_for(&self, lease_id: LeaseId) -> Result<LeaseKeepAlive>;

    async fn time_to_live<R>(&self, req: R) -> Result<LeaseTimeToLiveResponse>
    where
        R: Into<LeaseTimeToLiveRequest> + Send;
}

pub struct LeaseKeepAlive {
    id: LeaseId,
    req_tx: Sender<crate::proto::etcdserverpb::LeaseKeepAliveRequest>,
    resp_rx: Streaming<crate::proto::etcdserverpb::LeaseKeepAliveResponse>,
}

impl LeaseKeepAlive {
    pub(crate) fn new(
        id: LeaseId,
        req_tx: Sender<crate::proto::etcdserverpb::LeaseKeepAliveRequest>,
        resp_rx: Streaming<crate::proto::etcdserverpb::LeaseKeepAliveResponse>,
    ) -> Self {
        Self {
            id,
            req_tx,
            resp_rx,
        }
    }

    #[inline]
    pub fn lease_id(&mut self) -> LeaseId {
        self.id
    }

    pub async fn keep_alive(&mut self) -> Result<Option<LeaseKeepAliveResponse>> {
        let req = LeaseKeepAliveRequest::new(self.lease_id());

        self.req_tx
            .send(req.into())
            .await
            .map_err(|_| Error::ChannelClosed)?;

        Ok(match self.resp_rx.message().await? {
            Some(resp) => Some(resp.into()),
            None => None,
        })
    }
}