augurier
6 months ago
9 changed files with 374 additions and 130 deletions
Unified View
Diff Options
-
+3 -3cmd/main.go
-
+28 -28internal/client/client_node.go
-
+103 -65internal/nodes/init.go
-
+28 -8internal/nodes/node.go
-
+19 -6internal/nodes/server_node.go
-
+186 -0internal/nodes/vote.go
-
+2 -10test/common.go
-
+3 -3test/restart_follower_test.go
-
+2 -7test/server_client_test.go
+ 3
- 3
cmd/main.go
View File
+ 28
- 28
internal/client/client_node.go
View File
+ 103
- 65
internal/nodes/init.go
View File
+ 28
- 8
internal/nodes/node.go
View File
+ 19
- 6
internal/nodes/server_node.go
View File
+ 186
- 0
internal/nodes/vote.go
View File
| @ -0,0 +1,186 @@ | |||||
| package nodes | |||||
| import ( | |||||
| "net/rpc" | |||||
| "strconv" | |||||
| "sync" | |||||
| "time" | |||||
| "go.uber.org/zap" | |||||
| ) | |||||
| type RequestVoteArgs struct { | |||||
| Term int // 候选人的当前任期 | |||||
| CandidateId string // 候选人 ID | |||||
| LastLogIndex int // 候选人最后一条日志的索引 | |||||
| LastLogTerm int // 候选人最后一条日志的任期 | |||||
| } | |||||
| type RequestVoteReply struct { | |||||
| Term int // 当前节点的最新任期 | |||||
| VoteGranted bool // 是否同意投票 | |||||
| } | |||||
| func (n *Node) startElection() { | |||||
| n.mu.Lock() | |||||
| defer n.mu.Unlock() | |||||
| // 1. 增加当前任期，转换为 Candidate | |||||
| n.currTerm++ | |||||
| n.state = Candidate | |||||
| n.votedFor = n.selfId // 自己投自己 | |||||
| log.Sugar().Infof("[%s] 开始选举，当前任期: %d", n.selfId, n.currTerm) | |||||
| // 2. 重新设置选举超时，防止重复选举 | |||||
| n.resetElectionTimer() | |||||
| // 3. 构造 RequestVote 请求 | |||||
| var lastLogIndex int | |||||
| var lastLogTerm int | |||||
| if len(n.log) == 0 { | |||||
| lastLogIndex = 0 | |||||
| lastLogTerm = 0 // 论文中定义，空日志时 Term 设为 0 | |||||
| } else { | |||||
| lastLogIndex = len(n.log) - 1 | |||||
| lastLogTerm = n.log[lastLogIndex].Term | |||||
| } | |||||
| args := RequestVoteArgs{ | |||||
| Term: n.currTerm, | |||||
| CandidateId: n.selfId, | |||||
| LastLogIndex: lastLogIndex, | |||||
| LastLogTerm: lastLogTerm, | |||||
| } | |||||
| // 4. 并行向其他节点发送请求投票 | |||||
| var mu sync.Mutex | |||||
| cond := sync.NewCond(&mu) | |||||
| totalNodes := len(n.nodes) | |||||
| grantedVotes := 1 // 自己的票 | |||||
| for peerId := range n.nodes { | |||||
| go func(peerId string) { | |||||
| reply := RequestVoteReply{} | |||||
| if n.sendRequestVote(peerId, &args, &reply) { | |||||
| mu.Lock() | |||||
| if reply.Term > n.currTerm { | |||||
| // 发现更高任期，回退为 Follower | |||||
| log.Sugar().Infof("[%s] 发现更高的 Term (%d)，回退为 Follower", n.selfId, reply.Term) | |||||
| n.currTerm = reply.Term | |||||
| n.state = Follower | |||||
| n.votedFor = "" | |||||
| n.resetElectionTimer() | |||||
| mu.Unlock() | |||||
| return | |||||
| } | |||||
| if reply.VoteGranted { | |||||
| grantedVotes++ | |||||
| } | |||||
| if grantedVotes > totalNodes/2 { | |||||
| n.state = Leader | |||||
| log.Sugar().Infof("[%s] 当选 Leader!", n.selfId) | |||||
| n.initLeaderState() | |||||
| cond.Broadcast() | |||||
| } | |||||
| mu.Unlock() | |||||
| } | |||||
| }(peerId) | |||||
| } | |||||
| // 5. 等待选举结果 | |||||
| timeout := time.After(300 * time.Millisecond) | |||||
| for { | |||||
| mu.Lock() | |||||
| if n.state != Candidate { // 选举成功或回退，不再等待 | |||||
| mu.Unlock() | |||||
| return | |||||
| } | |||||
| select { | |||||
| case <-timeout: | |||||
| log.Sugar().Infof("[%s] 选举超时，重新发起选举", n.selfId) | |||||
| mu.Unlock() | |||||
| return | |||||
| default: | |||||
| cond.Wait() | |||||
| } | |||||
| mu.Unlock() | |||||
| } | |||||
| } | |||||
| func (node *Node) sendRequestVote(peerId string, args *RequestVoteArgs, reply *RequestVoteReply) bool { | |||||
| log.Sugar().Infof("Sending RequestVote to %s at %s", peerId, node.nodes[peerId].address) | |||||
| client, err := rpc.DialHTTP("tcp", node.nodes[peerId].address) | |||||
| if err != nil { | |||||
| log.Error("dialing: ", zap.Error(err)) | |||||
| return false | |||||
| } | |||||
| defer func(client *rpc.Client) { | |||||
| err := client.Close() | |||||
| if err != nil { | |||||
| log.Error("client close err: ", zap.Error(err)) | |||||
| } | |||||
| }(client) | |||||
| callErr := client.Call("Node.RequestVote", args, reply) // RPC | |||||
| if callErr != nil { | |||||
| log.Error("dialing node_"+peerId+"fail: ", zap.Error(callErr)) | |||||
| } | |||||
| return callErr == nil | |||||
| } | |||||
| func (n *Node) RequestVote(args *RequestVoteArgs, reply *RequestVoteReply) error { | |||||
| n.mu.Lock() | |||||
| log.Info(n.selfId) | |||||
| defer n.mu.Unlock() | |||||
| // 1. 如果候选人的任期小于当前任期，则拒绝投票 | |||||
| if args.Term < n.currTerm { | |||||
| reply.Term = n.currTerm | |||||
| reply.VoteGranted = false | |||||
| return nil | |||||
| } | |||||
| // 2. 如果请求的 Term 更高，则更新当前 Term 并回退为 Follower | |||||
| if args.Term > n.currTerm { | |||||
| n.currTerm = args.Term | |||||
| n.state = Follower | |||||
| n.votedFor = "" | |||||
| n.resetElectionTimer() // 重新设置选举超时 | |||||
| } | |||||
| // 3. 检查是否已经投过票，且是否投给了同一个候选人 | |||||
| if n.votedFor == "" || n.votedFor == args.CandidateId { | |||||
| // 4. 检查日志是否足够新 | |||||
| var lastLogIndex int | |||||
| var lastLogTerm int | |||||
| if len(n.log) == 0 { | |||||
| lastLogIndex = -1 | |||||
| lastLogTerm = 0 | |||||
| } else { | |||||
| lastLogIndex = len(n.log) - 1 | |||||
| lastLogTerm = n.log[lastLogIndex].Term | |||||
| } | |||||
| if args.LastLogTerm > lastLogTerm || | |||||
| (args.LastLogTerm == lastLogTerm && args.LastLogIndex >= lastLogIndex) { | |||||
| // 5. 投票给候选人 | |||||
| n.votedFor = args.CandidateId | |||||
| log.Info("term" + strconv.Itoa(n.currTerm) + ", " + n.selfId + "投票给" + n.votedFor) | |||||
| reply.VoteGranted = true | |||||
| n.resetElectionTimer() | |||||
| } else { | |||||
| reply.VoteGranted = false | |||||
| } | |||||
| } else { | |||||
| reply.VoteGranted = false | |||||
| } | |||||
| reply.Term = n.currTerm | |||||
| return nil | |||||
| } | |||||