Use redisDB number too.

BotBotMe · Apr 22, 2014 · e638673 · e638673 · bartTC · Apr 22, 2014
1 parent c31b353
commit e638673
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/common/queue.go b/common/queue.go
@@ -5,6 +5,8 @@ import (
 	"net/url"
 	"os"
 	"time"
+	"strconv"
+	"strings"
 
 	"github.com/golang/glog"
 	"github.com/monnand/goredis"
@@ -94,7 +96,13 @@ func NewRedisQueue() Queue {
 	if err != nil {
 		glog.Fatal("Could not read Redis string", err)
 	}
-	redisQueue := goredis.Client{Addr: redisUrl.Host}
+
+	redisDb, err := strconv.Atoi(strings.TrimLeft(redisUrl.Path, "/"))
+	if err != nil {
+		glog.Fatal("Could not read Redis path", err)
+	}
+
+	redisQueue := goredis.Client{Addr: redisUrl.Host, Db: redisDb}
 	s := RedisQueue{queue: &redisQueue}
 	s.waitForRedis()
 	return &s