minikv 2.0.0

Distributed key-value store with Raft consensus, encryption at rest, and pluggable storage backends.
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

use crate::common::raft::{AppendRequest, AppendResponse, VoteRequest, VoteResponse};
use crate::proto::coordinator_internal_client::CoordinatorInternalClient;
use std::time::Duration;
use tonic::transport::{Channel, Endpoint};

pub type PeerClient = CoordinatorInternalClient<Channel>;

const CONNECT_TIMEOUT: Duration = Duration::from_millis(200);
const REQUEST_TIMEOUT: Duration = Duration::from_millis(500);

pub fn normalize_addr(addr: &str) -> String {
    let addr = addr.trim();
    if addr.starts_with("http://") || addr.starts_with("https://") {
        addr.to_string()
    } else {
        format!("http://{}", addr)
    }
}

pub fn lazy_client(addr: &str) -> Result<PeerClient, tonic::Status> {
    let endpoint = Endpoint::from_shared(normalize_addr(addr))
        .map_err(|e| tonic::Status::invalid_argument(e.to_string()))?
        .connect_timeout(CONNECT_TIMEOUT)
        .timeout(REQUEST_TIMEOUT);
    Ok(CoordinatorInternalClient::new(endpoint.connect_lazy()))
}

pub async fn append_entries(
    client: &mut PeerClient,
    req: &AppendRequest,
) -> Result<AppendResponse, tonic::Status> {
    let resp = client
        .append_entries(crate::proto::AppendRequest::from(req))
        .await?
        .into_inner();
    Ok((&resp).into())
}

pub async fn request_vote(
    client: &mut PeerClient,
    req: &VoteRequest,
) -> Result<VoteResponse, tonic::Status> {
    let resp = client
        .request_vote(crate::proto::VoteRequest::from(req))
        .await?
        .into_inner();
    Ok((&resp).into())
}

pub async fn read_index(
    client: &mut PeerClient,
) -> Result<crate::proto::ReadIndexResponse, tonic::Status> {
    Ok(client
        .read_index(crate::proto::ReadIndexRequest {})
        .await?
        .into_inner())
}

pub async fn send_append_entries_rpc(
    peer_addr: &str,
    req: AppendRequest,
) -> Result<AppendResponse, tonic::Status> {
    let mut client = lazy_client(peer_addr)?;
    append_entries(&mut client, &req).await
}

pub async fn send_request_vote_rpc(
    peer_addr: &str,
    req: VoteRequest,
) -> Result<VoteResponse, tonic::Status> {
    let mut client = lazy_client(peer_addr)?;
    request_vote(&mut client, &req).await
}