From ba2dd15280263be3cfbbdecde7df65326e8b27bb Mon Sep 17 00:00:00 2001 From: "Sascha L. Teichmann" Date: Sun, 5 Mar 2017 14:03:02 +0100 Subject: [PATCH] Support unix domain sockets. --- cmd/mtredisalize/main.go | 10 +++++++++- cmd/mtseeder/baselevel.go | 11 ++++++++++- 2 files changed, 19 insertions(+), 2 deletions(-) diff --git a/cmd/mtredisalize/main.go b/cmd/mtredisalize/main.go index 60928ac..a052b25 100644 --- a/cmd/mtredisalize/main.go +++ b/cmd/mtredisalize/main.go @@ -12,6 +12,7 @@ import ( "os" "os/signal" "runtime" + "strings" "time" "bitbucket.org/s_l_teichmann/mtsatellite/common" @@ -117,7 +118,14 @@ func main() { var listener net.Listener - listener, err = net.Listen("tcp", fmt.Sprintf("%s:%d", host, port)) + var proto, address string + if strings.ContainsRune(host, '/') { + proto, address = "unix", host + } else { + proto, address = "tcp", fmt.Sprintf("%s:%d", host, port) + } + + listener, err = net.Listen(proto, address) if err != nil { log.Fatal(err) } diff --git a/cmd/mtseeder/baselevel.go b/cmd/mtseeder/baselevel.go index 7c4c839..1e66535 100644 --- a/cmd/mtseeder/baselevel.go +++ b/cmd/mtseeder/baselevel.go @@ -10,6 +10,7 @@ import ( "os" "path/filepath" "strconv" + "strings" "sync" "bitbucket.org/s_l_teichmann/mtsatellite/common" @@ -59,9 +60,17 @@ func createBaseLevel( jobs := make(chan blockPos) var done sync.WaitGroup + var proto string + if strings.ContainsRune(address, '/') { + proto = "unix" + } else { + proto = "tcp" + } + for i := 0; i < numWorkers; i++ { var client *common.RedisClient - if client, err = common.NewRedisClient("tcp", address); err != nil { + + if client, err = common.NewRedisClient(proto, address); err != nil { return } done.Add(1)