pacifica-rs 1.0.0

rust implementation of PacificA: Replication in Log-Based Distributed Storage Systems
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

use crate::rpc::message::AppendEntriesRequest;
use crate::rpc::message::AppendEntriesResponse;
use crate::rpc::message::InstallSnapshotRequest;
use crate::rpc::message::InstallSnapshotResponse;
use crate::rpc::message::ReplicaRecoverRequest;
use crate::rpc::message::ReplicaRecoverResponse;
use crate::rpc::message::TransferPrimaryRequest;
use crate::rpc::message::TransferPrimaryResponse;
use std::future::Future;

use crate::error::ConnectError;
use crate::rpc::RpcClientError;
use crate::rpc::RpcOption;
use crate::{ReplicaId, TypeConfig};

pub trait ConnectionClient<C>: Send + Sync
where
    C: TypeConfig,
{
    /// create connect the specified replica
    fn connect(&self, _replica_id: &ReplicaId<C::NodeId>) -> impl Future<Output = Result<(), ConnectError<C>>> + Send {
        async {
            Ok(())
        }
    }

    /// disconnect the specified replica
    fn disconnect(&self, _replica_id: &ReplicaId<C::NodeId>) -> impl Future<Output = bool> + Send {
        async {
            true
        }
    }

    /// Checks for a connection to the specified ['replica_id'] and crate connection if ['create_if_absent'] is true
    fn check_connected(
        &self,
        replica_id: &ReplicaId<C::NodeId>,
        create_if_absent: bool,
    ) -> impl Future<Output = Result<bool, ConnectError<C>>> + Send {
        async move {
            if !self.is_connected(replica_id).await {
                if create_if_absent {
                    self.connect(replica_id).await?;
                    return Ok(true);
                }
                return Ok(false);
            }
            Ok(true)
        }
    }

    ///
    fn is_connected(&self, _replica_id: &ReplicaId<C::NodeId>) -> impl Future<Output = bool> + Send  {
        async {
            true
        }
    }
}

pub trait ReplicaClient<C>: ConnectionClient<C> + Send + Sync
where
    C: TypeConfig,
{
    /// send append entries request to ['target']
    fn append_entries(
        &self,
        target: ReplicaId<C::NodeId>,
        request: AppendEntriesRequest<C>,
        rpc_option: RpcOption,
    ) -> impl Future<Output = Result<AppendEntriesResponse, RpcClientError>> + Send;

    /// send install snapshot request to ['target_id']
    fn install_snapshot(
        &self,
        target_id: ReplicaId<C::NodeId>,
        request: InstallSnapshotRequest<C>,
        rpc_option: RpcOption,
    ) -> impl Future<Output = Result<InstallSnapshotResponse, RpcClientError>> + Send;

    /// send replica recover request to ['primary_id']
    fn replica_recover(
        &self,
        primary_id: ReplicaId<C::NodeId>,
        request: ReplicaRecoverRequest<C>,
        rpc_option: RpcOption,
    ) -> impl Future<Output = Result<ReplicaRecoverResponse, RpcClientError>> + Send;

    /// send transfer primary request to ['secondary_id']
    fn transfer_primary(
        &self,
        secondary_id: ReplicaId<C::NodeId>,
        request: TransferPrimaryRequest<C>,
        rpc_option: RpcOption,
    ) -> impl Future<Output = Result<TransferPrimaryResponse, RpcClientError>> + Send;
}