replication master side

2025-10-05 16:57:06 +08:00 · 2022-11-21 23:36:35 +08:00
parent a7a3da6e49
commit ba7ea942cb
23 changed files with 886 additions and 217 deletions
--- a/database/replication_slave_test.go
+++ b/database/replication_slave_test.go
@@ -0,0 +1,131 @@
+package database
+
+import (
+	"bytes"
+	"github.com/hdt3213/godis/config"
+	"github.com/hdt3213/godis/lib/utils"
+	"github.com/hdt3213/godis/redis/client"
+	"github.com/hdt3213/godis/redis/connection"
+	"github.com/hdt3213/godis/redis/protocol"
+	"github.com/hdt3213/godis/redis/protocol/asserts"
+	"testing"
+	"time"
+)
+
+func TestReplicationSlaveSide(t *testing.T) {
+	mdb := mockServer()
+	masterCli, err := client.MakeClient("127.0.0.1:6379")
+	if err != nil {
+		t.Error(err)
+	}
+	masterCli.Start()
+
+	// sync with master
+	ret := masterCli.Send(utils.ToCmdLine("set", "1", "1"))
+	asserts.AssertStatusReply(t, ret, "OK")
+	conn := connection.NewFakeConn()
+	ret = mdb.Exec(conn, utils.ToCmdLine("SLAVEOF", "127.0.0.1", "6379"))
+	asserts.AssertStatusReply(t, ret, "OK")
+	success := false
+	for i := 0; i < 30; i++ {
+		// wait for sync
+		time.Sleep(time.Second)
+		ret = mdb.Exec(conn, utils.ToCmdLine("GET", "1"))
+		bulkRet, ok := ret.(*protocol.BulkReply)
+		if ok {
+			if bytes.Equal(bulkRet.Arg, []byte("1")) {
+				success = true
+				break
+			}
+		}
+	}
+	if !success {
+		t.Error("sync failed")
+		return
+	}
+
+	// receive aof
+	ret = masterCli.Send(utils.ToCmdLine("set", "1", "2"))
+	asserts.AssertStatusReply(t, ret, "OK")
+	success = false
+	for i := 0; i < 10; i++ {
+		// wait for sync
+		time.Sleep(time.Second)
+		ret = mdb.Exec(conn, utils.ToCmdLine("GET", "1"))
+		bulkRet, ok := ret.(*protocol.BulkReply)
+		if ok {
+			if bytes.Equal(bulkRet.Arg, []byte("2")) {
+				success = true
+				break
+			}
+		}
+	}
+	if !success {
+		t.Error("sync failed")
+		return
+	}
+	err = mdb.slaveStatus.sendAck2Master()
+	if err != nil {
+		t.Error(err)
+		return
+	}
+	time.Sleep(3 * time.Second)
+
+	// test reconnect
+	config.Properties.ReplTimeout = 1
+	_ = mdb.slaveStatus.masterConn.Close()
+	mdb.slaveStatus.lastRecvTime = time.Now().Add(-time.Hour) // mock timeout
+	mdb.slaveCron()
+	time.Sleep(3 * time.Second)
+	ret = masterCli.Send(utils.ToCmdLine("set", "1", "3"))
+	asserts.AssertStatusReply(t, ret, "OK")
+	success = false
+	for i := 0; i < 10; i++ {
+		// wait for sync
+		time.Sleep(time.Second)
+		ret = mdb.Exec(conn, utils.ToCmdLine("GET", "1"))
+		bulkRet, ok := ret.(*protocol.BulkReply)
+		if ok {
+			if bytes.Equal(bulkRet.Arg, []byte("3")) {
+				success = true
+				break
+			}
+		}
+	}
+	if !success {
+		t.Error("sync failed")
+		return
+	}
+
+	// test slave of no one
+	ret = mdb.Exec(conn, utils.ToCmdLine("SLAVEOF", "NO", "ONE"))
+	asserts.AssertStatusReply(t, ret, "OK")
+	ret = masterCli.Send(utils.ToCmdLine("set", "1", "4"))
+	asserts.AssertStatusReply(t, ret, "OK")
+	ret = mdb.Exec(conn, utils.ToCmdLine("GET", "1"))
+	asserts.AssertBulkReply(t, ret, "3")
+	ret = mdb.Exec(conn, utils.ToCmdLine("SLAVEOF", "127.0.0.1", "6379"))
+	asserts.AssertStatusReply(t, ret, "OK")
+	success = false
+	for i := 0; i < 30; i++ {
+		// wait for sync
+		time.Sleep(time.Second)
+		ret = mdb.Exec(conn, utils.ToCmdLine("GET", "1"))
+		bulkRet, ok := ret.(*protocol.BulkReply)
+		if ok {
+			if bytes.Equal(bulkRet.Arg, []byte("4")) {
+				success = true
+				break
+			}
+		}
+	}
+	if !success {
+		t.Error("sync failed")
+		return
+	}
+
+	err = mdb.slaveStatus.close()
+	if err != nil {
+		t.Error("cannot close")
+	}
+}