1
0
mirror of https://github.com/redis/go-redis.git synced 2025-07-28 06:42:00 +03:00

Better redis process abstraction

This commit is contained in:
Dimitrij Denissenko
2015-01-15 17:23:22 +00:00
parent f6bca78168
commit e16db84e4a
2 changed files with 90 additions and 67 deletions

View File

@ -1,11 +1,6 @@
package redis_test
import (
"io/ioutil"
"os"
"os/exec"
"path/filepath"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"gopkg.in/redis.v2"
@ -16,69 +11,23 @@ var _ = Describe("Sentinel", func() {
const masterName = "mymaster"
const masterPort = "8123"
const sentinelPort = "8124"
const sentinelConf = `
port ` + sentinelPort + `
sentinel monitor ` + masterName + ` 127.0.0.1 ` + masterPort + ` 1
sentinel down-after-milliseconds ` + masterName + ` 400
sentinel failover-timeout ` + masterName + ` 800
sentinel parallel-syncs ` + masterName + ` 1
`
var runCmd = func(name string, args ...string) *os.Process {
cmd := exec.Command(name, args...)
if false {
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
}
err := cmd.Start()
Expect(err).NotTo(HaveOccurred())
return cmd.Process
}
var connect = func(port string) *redis.Client {
client := redis.NewTCPClient(&redis.Options{
Addr: ":" + port,
})
Eventually(func() error {
return client.Ping().Err()
}, "1s", "100ms").ShouldNot(HaveOccurred())
return client
}
var startMaster = func() (*redis.Client, *os.Process) {
proc := runCmd("redis-server", "--port", masterPort)
return connect(masterPort), proc
}
var startSlave = func(port string) (*redis.Client, *os.Process) {
proc := runCmd("redis-server", "--port", port, "--slaveof", "127.0.0.1", masterPort)
return connect(port), proc
}
var startSentinel = func() *os.Process {
dir, err := ioutil.TempDir("", "sentinel")
Expect(err).NotTo(HaveOccurred())
fname := filepath.Join(dir, "sentinel.conf")
err = ioutil.WriteFile(fname, []byte(sentinelConf), 0664)
Expect(err).NotTo(HaveOccurred())
proc := runCmd("redis-server", fname, "--sentinel")
client := connect(sentinelPort)
client.Close()
return proc
}
It("should facilitate failover", func() {
master, mproc := startMaster()
defer mproc.Kill()
slave1, sproc1 := startSlave("8125")
defer sproc1.Kill()
slave2, sproc2 := startSlave("8126")
defer sproc2.Kill()
sntproc := startSentinel()
defer sntproc.Kill()
master, err := startRedis(masterPort)
Expect(err).NotTo(HaveOccurred())
defer master.Close()
sentinel, err := startSentinel(sentinelPort, masterName, masterPort)
Expect(err).NotTo(HaveOccurred())
defer sentinel.Close()
slave1, err := startRedis("8125", "--slaveof", "127.0.0.1", masterPort)
Expect(err).NotTo(HaveOccurred())
defer slave1.Close()
slave2, err := startRedis("8126", "--slaveof", "127.0.0.1", masterPort)
Expect(err).NotTo(HaveOccurred())
defer slave2.Close()
client := redis.NewFailoverClient(&redis.FailoverOptions{
MasterName: masterName,
@ -87,7 +36,7 @@ sentinel parallel-syncs ` + masterName + ` 1
defer client.Close()
// Set value on master, verify
err := client.Set("foo", "master").Err()
err = client.Set("foo", "master").Err()
Expect(err).NotTo(HaveOccurred())
val, err := master.Get("foo").Result()