mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-05 00:42:43 +08:00
bf7f628810
wip: raft does not care about migrating wip: optimize code wip: raft election wip wip: fix raft leader missing log entries wip fix a dead lock batch set slot route wip: raft persist wip refactor cluster suite remove relay rename relay2 refactor: allow customizing client factory test raft refactor re-balance avoid errors caused by inconsistent status on follower nodes during raft commits test raft election
51 lines
1.5 KiB
Go
51 lines
1.5 KiB
Go
package cluster
|
|
|
|
import (
|
|
"github.com/hdt3213/godis/redis/connection"
|
|
"github.com/hdt3213/godis/redis/protocol/asserts"
|
|
"math/rand"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
func TestRollback(t *testing.T) {
|
|
// rollback uncommitted transaction
|
|
testNodeA := testCluster[0]
|
|
conn := new(connection.FakeConn)
|
|
FlushAll(testNodeA, conn, toArgs("FLUSHALL"))
|
|
txID := rand.Int63()
|
|
txIDStr := strconv.FormatInt(txID, 10)
|
|
keys := []string{"a", "{a}1"}
|
|
groupMap := map[string][]string{
|
|
testNodeA.self: keys,
|
|
}
|
|
args := []string{txIDStr, "DEL"}
|
|
args = append(args, keys...)
|
|
testNodeA.db.Exec(conn, toArgs("SET", "a", "a"))
|
|
ret := execPrepare(testNodeA, conn, makeArgs("Prepare", args...))
|
|
asserts.AssertNotError(t, ret)
|
|
requestRollback(testNodeA, conn, txID, groupMap)
|
|
ret = testNodeA.db.Exec(conn, toArgs("GET", "a"))
|
|
asserts.AssertBulkReply(t, ret, "a")
|
|
|
|
// rollback committed transaction
|
|
FlushAll(testNodeA, conn, toArgs("FLUSHALL"))
|
|
testNodeA.db.Exec(conn, toArgs("SET", "a", "a"))
|
|
txID = rand.Int63()
|
|
txIDStr = strconv.FormatInt(txID, 10)
|
|
args = []string{txIDStr, "DEL"}
|
|
args = append(args, keys...)
|
|
ret = execPrepare(testNodeA, conn, makeArgs("Prepare", args...))
|
|
asserts.AssertNotError(t, ret)
|
|
_, err := requestCommit(testNodeA, conn, txID, groupMap)
|
|
if err != nil {
|
|
t.Errorf("del failed %v", err)
|
|
return
|
|
}
|
|
ret = testNodeA.db.Exec(conn, toArgs("GET", "a")) // call db.Exec to skip key router
|
|
asserts.AssertNullBulk(t, ret)
|
|
requestRollback(testNodeA, conn, txID, groupMap)
|
|
ret = testNodeA.db.Exec(conn, toArgs("GET", "a"))
|
|
asserts.AssertBulkReply(t, ret, "a")
|
|
}
|