You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.1 KiB
95 lines
2.1 KiB
package nodes
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"net/rpc"
|
|
"simple-kv-store/internal/logprovider"
|
|
"time"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
var log, _ = logprovider.CreateDefaultZapLogger(zap.InfoLevel)
|
|
|
|
func newNode(address string) *Public_node_info {
|
|
return &Public_node_info{
|
|
connect: false,
|
|
address: address,
|
|
}
|
|
}
|
|
|
|
func Init(selfId string, nodeAddr map[string]string, db *leveldb.DB, rstorage *RaftStorage) *Node {
|
|
ns := make(map[string]*Public_node_info)
|
|
for id, addr := range nodeAddr {
|
|
ns[id] = newNode(addr)
|
|
}
|
|
|
|
// 创建节点
|
|
node := &Node{
|
|
selfId: selfId,
|
|
leaderId: "",
|
|
nodes: ns,
|
|
maxLogId: -1, // 后来发现论文中是从1开始的（初始0），但不想改了
|
|
currTerm: 1,
|
|
log: make([]RaftLogEntry, 0),
|
|
commitIndex: -1,
|
|
lastApplied: -1,
|
|
nextIndex: make(map[string]int),
|
|
matchIndex: make(map[string]int),
|
|
db: db,
|
|
storage: rstorage,
|
|
}
|
|
node.initLeaderState()
|
|
return node
|
|
}
|
|
|
|
func (n *Node) initLeaderState() {
|
|
for peerId := range n.nodes {
|
|
n.nextIndex[peerId] = len(n.log) // 发送日志的下一个索引
|
|
n.matchIndex[peerId] = 0 // 复制日志的最新匹配索引
|
|
}
|
|
n.storage.SetTermAndVote(n.currTerm, n.votedFor)
|
|
}
|
|
|
|
func Start(node *Node) {
|
|
node.state = Follower // 所有节点以 Follower 状态启动
|
|
node.resetElectionTimer() // 启动选举超时定时器
|
|
|
|
go func() {
|
|
for {
|
|
switch node.state {
|
|
case Follower:
|
|
// 监听心跳超时
|
|
fmt.Printf("[%s] is a follower, 监听中...\n", node.selfId)
|
|
|
|
case Leader:
|
|
// 发送心跳
|
|
fmt.Printf("[%s] is the leader, 发送心跳...\n", node.selfId)
|
|
node.BroadCastKV(Normal)
|
|
}
|
|
time.Sleep(50 * time.Millisecond)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (node *Node) Rpc(port string) {
|
|
err := rpc.Register(node)
|
|
if err != nil {
|
|
log.Fatal("rpc register failed", zap.Error(err))
|
|
}
|
|
rpc.HandleHTTP()
|
|
|
|
l, e := net.Listen("tcp", port)
|
|
if e != nil {
|
|
log.Fatal("listen error:", zap.Error(e))
|
|
}
|
|
go func() {
|
|
err := http.Serve(l, nil)
|
|
if err != nil {
|
|
log.Fatal("http server error:", zap.Error(err))
|
|
}
|
|
}()
|
|
}
|