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.
148 lines
3.1 KiB
148 lines
3.1 KiB
package nodes
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// RaftStorage 结构，持久化 currentTerm、votedFor 和 logEntries
|
|
type RaftStorage struct {
|
|
mu sync.Mutex
|
|
filePath string
|
|
CurrentTerm int `json:"current_term"`
|
|
VotedFor string `json:"voted_for"`
|
|
LogEntries []RaftLogEntry `json:"log_entries"`
|
|
}
|
|
|
|
// NewRaftStorage 创建 Raft 存储
|
|
func NewRaftStorage(filePath string) *RaftStorage {
|
|
storage := &RaftStorage{
|
|
filePath: filePath,
|
|
}
|
|
storage.loadData() // 载入已有数据
|
|
return storage
|
|
}
|
|
|
|
// loadData 读取 JSON 文件数据
|
|
func (rs *RaftStorage) loadData() {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
|
|
file, err := os.Open(rs.filePath)
|
|
if err != nil {
|
|
log.Info("文件未创建：" + rs.filePath)
|
|
rs.saveData() // 文件不存在时创建默认数据
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
err = json.NewDecoder(file).Decode(rs)
|
|
if err != nil {
|
|
log.Error("读取文件失败：" + rs.filePath)
|
|
}
|
|
}
|
|
|
|
// 持久化数据到 JSON(必须持有锁，不能直接外部调用)
|
|
func (rs *RaftStorage) saveData() {
|
|
// 获取文件所在的目录
|
|
dir := filepath.Dir(rs.filePath)
|
|
|
|
// 确保目录存在
|
|
if err := os.MkdirAll(dir, 0755); err != nil {
|
|
log.Error("创建存储目录失败", zap.Error(err))
|
|
return
|
|
}
|
|
|
|
file, err := os.Create(rs.filePath)
|
|
if err != nil {
|
|
log.Error("持久化节点出错", zap.Error(err))
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
err = json.NewEncoder(file).Encode(rs)
|
|
if err != nil {
|
|
log.Error("持久化写入失败")
|
|
}
|
|
}
|
|
|
|
// SetCurrentTerm 设置当前 term，并清空 votedFor（符合 Raft 规范）
|
|
func (rs *RaftStorage) SetCurrentTerm(term int) {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
if term > rs.CurrentTerm {
|
|
rs.CurrentTerm = term
|
|
rs.VotedFor = "" // 新任期清空投票
|
|
rs.saveData()
|
|
}
|
|
}
|
|
|
|
// GetCurrentTerm 获取当前 term
|
|
func (rs *RaftStorage) GetCurrentTerm() int {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
return rs.CurrentTerm
|
|
}
|
|
|
|
// SetVotedFor 记录投票给谁
|
|
func (rs *RaftStorage) SetVotedFor(candidate string) {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
rs.VotedFor = candidate
|
|
rs.saveData()
|
|
}
|
|
|
|
// GetVotedFor 获取投票对象
|
|
func (rs *RaftStorage) GetVotedFor() string {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
return rs.VotedFor
|
|
}
|
|
|
|
// 同时设置
|
|
func (rs *RaftStorage) SetTermAndVote(term int, candidate string) {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
rs.VotedFor = candidate
|
|
rs.CurrentTerm = term
|
|
rs.saveData()
|
|
}
|
|
|
|
// append日志
|
|
func (rs *RaftStorage) AppendLog(rlogE RaftLogEntry) {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
|
|
rs.LogEntries = append(rs.LogEntries, rlogE)
|
|
rs.saveData()
|
|
}
|
|
|
|
// 更改日志
|
|
func (rs *RaftStorage) WriteLog(rlogEs []RaftLogEntry) {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
|
|
rs.LogEntries = rlogEs
|
|
rs.saveData()
|
|
}
|
|
|
|
// 获取所有日志
|
|
func (rs *RaftStorage) GetLogEntries() []RaftLogEntry {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
return rs.LogEntries
|
|
}
|
|
|
|
// GetLastLogIndex 获取最新日志的 index
|
|
func (rs *RaftStorage) GetLastLogIndex() int {
|
|
rs.mu.Lock()
|
|
defer rs.mu.Unlock()
|
|
if len(rs.LogEntries) == 0 {
|
|
return 0
|
|
}
|
|
return len(rs.LogEntries)-1
|
|
}
|