openraft 0.9.24

Advanced Raft consensus
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use std::sync::Arc;

use maplit::btreeset;

use crate::engine::testing::UTConfig;
use crate::engine::Command;
use crate::engine::Engine;
use crate::raft_state::Accepted;
use crate::raft_state::LogStateReader;
use crate::testing::log_id;
use crate::EffectiveMembership;
use crate::Membership;
use crate::MembershipState;

fn m01() -> Membership<u64, ()> {
    Membership::new(vec![btreeset! {0,1}], None)
}

fn m23() -> Membership<u64, ()> {
    Membership::new(vec![btreeset! {2,3}], None)
}

fn eng() -> Engine<UTConfig> {
    let mut eng = Engine::default();
    eng.state.enable_validation(false); // Disable validation for incomplete state

    eng.state.committed = Some(log_id(1, 1, 1));
    eng.state.membership_state = MembershipState::new(
        Arc::new(EffectiveMembership::new(Some(log_id(1, 1, 1)), m01())),
        Arc::new(EffectiveMembership::new(Some(log_id(2, 1, 3)), m23())),
    );
    eng
}

#[test]
fn test_following_handler_commit_entries_empty() -> anyhow::Result<()> {
    let mut eng = eng();

    eng.following_handler().commit_entries(None);

    assert_eq!(Some(&log_id(1, 1, 1)), eng.state.committed());
    assert_eq!(
        MembershipState::new(
            Arc::new(EffectiveMembership::new(Some(log_id(1, 1, 1)), m01())),
            Arc::new(EffectiveMembership::new(Some(log_id(2, 1, 3)), m23())),
        ),
        eng.state.membership_state
    );
    assert_eq!(0, eng.output.take_commands().len());

    Ok(())
}

#[test]
fn test_following_handler_commit_entries_ge_accepted() -> anyhow::Result<()> {
    let mut eng = eng();
    let l = eng.state.vote_ref().leader_id();
    eng.state.accepted = Accepted::new(*l, Some(log_id(1, 1, 2)));

    eng.following_handler().commit_entries(Some(log_id(2, 1, 3)));

    assert_eq!(Some(&log_id(1, 1, 2)), eng.state.committed());
    assert_eq!(
        MembershipState::new(
            Arc::new(EffectiveMembership::new(Some(log_id(1, 1, 1)), m01())),
            Arc::new(EffectiveMembership::new(Some(log_id(2, 1, 3)), m23())),
        ),
        eng.state.membership_state
    );
    assert_eq!(
        vec![Command::Commit {
            seq: 1,
            already_committed: Some(log_id(1, 1, 1)),
            upto: log_id(1, 1, 2),
        }],
        eng.output.take_commands()
    );

    Ok(())
}

#[test]
fn test_following_handler_commit_entries_le_accepted() -> anyhow::Result<()> {
    let mut eng = eng();
    let l = eng.state.vote_ref().leader_id();
    eng.state.accepted = Accepted::new(*l, Some(log_id(3, 1, 4)));

    eng.following_handler().commit_entries(Some(log_id(2, 1, 3)));

    assert_eq!(Some(&log_id(2, 1, 3)), eng.state.committed());
    assert_eq!(
        MembershipState::new(
            Arc::new(EffectiveMembership::new(Some(log_id(2, 1, 3)), m23())),
            Arc::new(EffectiveMembership::new(Some(log_id(2, 1, 3)), m23()))
        ),
        eng.state.membership_state
    );
    assert_eq!(
        vec![
            //
            Command::Commit {
                seq: 1,
                already_committed: Some(log_id(1, 1, 1)),
                upto: log_id(2, 1, 3)
            },
        ],
        eng.output.take_commands()
    );

    Ok(())
}