MIT-6.824: Lab 3A: leader election

type Raft struct {
	mu        sync.Mutex          // Lock to protect shared access to this peer's state
	peers     []*labrpc.ClientEnd // RPC end points of all peers
	persister *Persister          // Object to hold this peer's persisted state
	me        int                 // this peer's index into peers[]
	dead      int32               // set by Kill()

	// Your data here (3A, 3B, 3C).
	// Look at the paper's Figure 2 for a description of what
	// state a Raft server must maintain.
	State RaftState // state of current Raft Node

	currentTerm int      // latest term server has seen (initialized to 0 on first boot, increases monotonically)
	votedFor    int      // candidateId that received vote in current term (or null if none)
	logs        []*Entry // log entries; each entry contains command for state machine, and term when entry was received by leader (first index is 1)

	electionTimer  *time.Timer // election timer
	heartbeatTimer *time.Timer // heartbeat timer
}

func Make(peers []*labrpc.ClientEnd, me int,
	persister *Persister, applyCh chan ApplyMsg) *Raft {
	rf := &Raft{}
	rf.peers = peers
	rf.persister = persister
	rf.me = me

	// Your initialization code here (3A, 3B, 3C).
	rf.State = StateFollower

	rf.currentTerm = 0
	rf.votedFor = -1
	rf.logs = make([]*Entry, 0)

	rf.electionTimer = time.NewTimer(getElectionTimeout())
	rf.heartbeatTimer = time.NewTimer(getHeartbeatTimeout())

	// initialize from state persisted before a crash
	rf.readPersist(persister.ReadRaftState())

	// start ticker goroutine to start elections
	go rf.ticker()


	return rf
}

type RequestAppendEntriesArgs struct {
	Term     int // leader's term
	LeaderId int // leader's id
}
type RequestAppendEntriesReply struct {

}

// RequestAppendEntries RPC handler.
// expired leader, follower and candidate use this
func (rf *Raft) RequestAppendEntries(args *RequestAppendEntriesArgs, reply *RequestAppendEntriesReply) {

}

type RequestVoteArgs struct {
	Term         int // candidate’s term
	CandidateId  int // candidate requesting vote
	LastLogIndex int // index of candidate’s last log entry (§5.4)
	LastLogTerm  int // term of candidate’s last log entry (§5.4)
}

type RequestVoteReply struct {
	Term        int  // currentTerm, for candidate to update itself
	VoteGranted bool // true means candidate received vote
}

// example RequestVote RPC handler.
func (rf *Raft) RequestVote(args *RequestVoteArgs, reply *RequestVoteReply) {

}

func (rf *Raft) ticker() {
	for rf.killed() == false {

		// Your code here (3A)
		// Check if a leader election should be started.
		select {
		case <-rf.electionTimer.C:
			rf.mu.Lock()
			if rf.State != StateLeader {
				rf.currentTerm += 1
				rf.startElection()
			}
			rf.electionTimer.Reset(getElectionTimeout())
			rf.mu.Unlock()
		case <-rf.heartbeatTimer.C:
			rf.mu.Lock()
			if rf.State == StateLeader {
				rf.startHeartbeat()
			}
			rf.heartbeatTimer.Reset(getHeartbeatTimeout())
			rf.mu.Unlock()
		}
	}
}

#!/bin/bash

# 定义执行次数
runs=100

# 循环执行命令
for (( i=1; i<=$runs; i++ )); do
    echo "Running test $i"
    go test -timeout 30s -run "^TestInitialElection3A$" -count=1 6.5840/raft
    if [ $? -ne 0 ]; then
        echo "Error: Test failed. Exiting..."
        exit 1
    fi

    go test -timeout 30s -run "^TestReElection3A$" -count=1 6.5840/raft
    if [ $? -ne 0 ]; then
        echo "Error: Test failed. Exiting..."
        exit 1
    fi

    go test -timeout 30s -run "^TestManyElections3A$" -count=1 6.5840/raft
    if [ $? -ne 0 ]; then
        echo "Error: Test failed. Exiting..."
        exit 1
    fi
done

echo "All tests passed successfully."

func getElectionTimeout() time.Duration {
	ms := 200 + (rand.Int63() % 200)
	return time.Duration(ms) * time.Millisecond
}

func getHeartbeatTimeout() time.Duration {
	return time.Millisecond * 100
}