augurier
6 months ago
10 changed files with 236 additions and 84 deletions
Split View
Diff Options
-
+1 -0.gitignore
-
+22 -2cmd/main.go
-
+2 -0go.mod
-
+18 -0go.sum
-
+9 -6internal/nodes/init.go
-
+6 -35internal/nodes/node.go
-
+12 -9internal/nodes/server_node.go
-
+45 -0test/common.go
-
+110 -0test/restart_follower_test.go
-
+11 -32test/server_client_test.go
+ 1
- 0
.gitignore
View File
+ 22
- 2
cmd/main.go
View File
+ 2
- 0
go.mod
View File
+ 18
- 0
go.sum
View File
+ 9
- 6
internal/nodes/init.go
View File
+ 6
- 35
internal/nodes/node.go
View File
+ 12
- 9
internal/nodes/server_node.go
View File
+ 45
- 0
test/common.go
View File
| @ -0,0 +1,45 @@ | |||
| package test | |||
| import ( | |||
| "fmt" | |||
| "os" | |||
| "os/exec" | |||
| "strconv" | |||
| "strings" | |||
| ) | |||
| func ExecuteNodeI(i int, isLeader bool, isNewDb bool, clusters []string) *exec.Cmd { | |||
| tmpClusters := append(clusters[:i], clusters[i+1:]...) | |||
| port := fmt.Sprintf(":%d", uint16(9090)+uint16(i)) | |||
| var isleader string | |||
| if isLeader { | |||
| isleader = "true" | |||
| } else { | |||
| isleader = "false" | |||
| } | |||
| var isnewdb string | |||
| if isNewDb { | |||
| isnewdb = "true" | |||
| } else { | |||
| isnewdb = "false" | |||
| } | |||
| cmd := exec.Command( | |||
| "../main", | |||
| "-id", strconv.Itoa(i + 1), | |||
| "-port", port, | |||
| "-cluster", strings.Join(tmpClusters, ","), | |||
| "-isleader=" + isleader, | |||
| "-isNewDb=" + isnewdb, | |||
| ) | |||
| cmd.Stdout = os.Stdout | |||
| cmd.Stderr = os.Stderr | |||
| // 执行命令 | |||
| err := cmd.Start() | |||
| if err != nil { | |||
| fmt.Println("启动进程出错:", err) | |||
| return nil | |||
| } | |||
| return cmd | |||
| } | |||
+ 110
- 0
test/restart_follower_test.go
View File
| @ -0,0 +1,110 @@ | |||
| package test | |||
| import ( | |||
| "fmt" | |||
| "os/exec" | |||
| "simple-kv-store/internal/client" | |||
| "simple-kv-store/internal/nodes" | |||
| "strconv" | |||
| "syscall" | |||
| "testing" | |||
| "time" | |||
| ) | |||
| func TestFollowerRestart(t *testing.T) { | |||
| // 登记结点信息 | |||
| n := 3 | |||
| var clusters []string | |||
| for i := 0; i < n; i++ { | |||
| port := fmt.Sprintf("%d", uint16(9090)+uint16(i)) | |||
| addr := "127.0.0.1:" + port | |||
| clusters = append(clusters, addr) | |||
| } | |||
| // 结点启动 | |||
| var cmds []*exec.Cmd | |||
| for i := 0; i < n; i++ { | |||
| var cmd *exec.Cmd | |||
| if i == 0 { | |||
| cmd = ExecuteNodeI(i, true, true, clusters) | |||
| } else { | |||
| cmd = ExecuteNodeI(i, false, true, clusters) | |||
| } | |||
| if cmd == nil { | |||
| return | |||
| } else { | |||
| cmds = append(cmds, cmd) | |||
| } | |||
| } | |||
| time.Sleep(time.Second) // 等待启动完毕 | |||
| // client启动, 连接leader | |||
| cWrite := clientPkg.Client{Address: clusters[0], ServerId: "1"} | |||
| // 写入 | |||
| var s clientPkg.Status | |||
| for i := 0; i < 5; i++ { | |||
| key := strconv.Itoa(i) | |||
| s := cWrite.Write(nodes.LogEntry{Key: key, Value: "hello"}) | |||
| if s != clientPkg.Ok { | |||
| t.Errorf("write test fail") | |||
| } | |||
| } | |||
| time.Sleep(time.Second) // 等待写入完毕 | |||
| // 模拟最后一个结点崩溃 | |||
| err := cmds[n - 1].Process.Signal(syscall.SIGTERM) | |||
| if err != nil { | |||
| fmt.Println("Error sending signal:", err) | |||
| return | |||
| } | |||
| // 继续写入 | |||
| for i := 5; i < 10; i++ { | |||
| key := strconv.Itoa(i) | |||
| s := cWrite.Write(nodes.LogEntry{Key: key, Value: "hello"}) | |||
| if s != clientPkg.Ok { | |||
| t.Errorf("write test fail") | |||
| } | |||
| } | |||
| // 恢复结点 | |||
| cmd := ExecuteNodeI(n - 1, false, false, clusters) | |||
| if cmd == nil { | |||
| t.Errorf("recover test1 fail") | |||
| return | |||
| } else { | |||
| cmds[n - 1] = cmd | |||
| } | |||
| time.Sleep(time.Second) // 等待启动完毕 | |||
| // client启动, 连接节点n-1(去读它的数据) | |||
| cRead := clientPkg.Client{Address: clusters[n - 1], ServerId: "n"} | |||
| // 读崩溃前写入数据 | |||
| for i := 0; i < 5; i++ { | |||
| key := strconv.Itoa(i) | |||
| var value string | |||
| s = cRead.Read(key, &value) | |||
| if s != clientPkg.Ok { | |||
| t.Errorf("Read test1 fail") | |||
| } | |||
| } | |||
| // 读未写入数据 | |||
| for i := 5; i < 15; i++ { | |||
| key := strconv.Itoa(i) | |||
| var value string | |||
| s = cRead.Read(key, &value) | |||
| if s != clientPkg.NotFound { | |||
| t.Errorf("Read test2 fail") | |||
| } | |||
| } | |||
| // 通知进程结束 | |||
| for _, cmd := range cmds { | |||
| err := cmd.Process.Signal(syscall.SIGTERM) | |||
| if err != nil { | |||
| fmt.Println("Error sending signal:", err) | |||
| return | |||
| } | |||
| } | |||
| } | |||