augurier
6 months ago
7 changed files with 199 additions and 26 deletions
Split View
Diff Options
-
+1 -0.gitignore
-
+10 -6cmd/main.go
-
+4 -2internal/nodes/init.go
-
+31 -15internal/nodes/node.go
-
+147 -0internal/nodes/node_storage.go
-
+3 -2internal/nodes/server_node.go
-
+3 -1internal/nodes/vote.go
+ 1
- 0
.gitignore
View File
+ 10
- 6
cmd/main.go
View File
+ 4
- 2
internal/nodes/init.go
View File
+ 31
- 15
internal/nodes/node.go
View File
+ 147
- 0
internal/nodes/node_storage.go
View File
| @ -0,0 +1,147 @@ | |||
| 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 | |||
| } | |||