ruaft/src/persister.rs

use std::convert::TryFrom;

use bytes::Bytes;
use serde::de::DeserializeOwned;
use serde::ser::Serialize;
use serde_derive::{Deserialize, Serialize};

use crate::log_array::LogArray;
use crate::{Peer, RaftState, Term};

/// An object that saves bytes to permanent storage.
///
/// When the methods of this trait returns, data should have been persisted to
/// the storage. These methods should never return failure except panicking.
/// They should not block forever, either.
pub trait Persister: Send + Sync {
    fn read_state(&self) -> Bytes;
    fn save_state(&self, bytes: Bytes);
    fn state_size(&self) -> usize;

    fn save_snapshot_and_state(&self, state: Bytes, snapshot: &[u8]);
}

#[derive(Serialize, Deserialize)]
pub(crate) struct PersistedRaftState<Command> {
    pub current_term: Term,
    pub voted_for: Option<Peer>,
    pub log: LogArray<Command>,
}

impl<Command: Clone, T: AsRef<RaftState<Command>>> From<T>
    for PersistedRaftState<Command>
{
    fn from(raft_state: T) -> Self {
        Self::from(raft_state.as_ref())
    }
}

impl<Command: Clone> From<&RaftState<Command>> for PersistedRaftState<Command> {
    fn from(raft_state: &RaftState<Command>) -> Self {
        Self {
            current_term: raft_state.current_term,
            voted_for: raft_state.voted_for,
            log: raft_state.log.clone(),
        }
    }
}

impl<Command: DeserializeOwned> TryFrom<Bytes> for PersistedRaftState<Command> {
    type Error = bincode::Error;

    fn try_from(value: Bytes) -> Result<Self, Self::Error> {
        bincode::deserialize(value.as_ref())
    }
}

impl<Command: Serialize> From<PersistedRaftState<Command>> for Bytes {
    fn from(value: PersistedRaftState<Command>) -> Bytes {
        bincode::serialize(&value)
            .expect("Serialization should not fail")
            .into()
    }
}