Add HostFilter to RedisSentinel

Author: mythz

src/ServiceStack.Redis/RedisSentinel.cs

@@ -49,6 +49,18 @@ public IRedisClientsManager Setup()
             return this.redisManager;
         }
 
+        public Func<string, string> HostFilter { get; set; }
+
+        internal string[] ConfigureHosts(IEnumerable<string> hosts)
+        {
+            if (hosts == null)
+                return new string[0];
+
+            return HostFilter == null
+                ? hosts.ToArray()
+                : hosts.Map(HostFilter).ToArray();
+        }
+
         private void GetValidSentinel()
         {
             while (this.redisManager == null && ShouldRetry())

src/ServiceStack.Redis/RedisSentinelWorker.cs

@@ -82,11 +82,15 @@ private void ConfigureRedisFromSentinel()
 
             if (redisManager == null)
             {
-                redisManager = redisSentinel.RedisManagerFactory.Create(masters, slaves);
+                redisManager = redisSentinel.RedisManagerFactory.Create(
+                    redisSentinel.ConfigureHosts(masters), 
+                    redisSentinel.ConfigureHosts(slaves));
             }
             else
             {
-                ((IRedisFailover)redisManager).FailoverTo(masters, slaves);
+                ((IRedisFailover)redisManager).FailoverTo(
+                    redisSentinel.ConfigureHosts(masters),
+                    redisSentinel.ConfigureHosts(slaves));
             }
         }
 

tests/ServiceStack.Redis.Tests/RedisSentinelTests.cs

@@ -101,13 +101,9 @@ public void Can_specify_Timeout_on_RedisManager()
         [Test]
         public void Can_specify_db_on_RedisSentinel()
         {
-            var host = new[] {"{0}:{1}".Fmt(TestConfig.SentinelHost, TestConfig.RedisSentinelPort)};
-            var sentinel = new RedisSentinel(host, TestConfig.MasterName) {
-                RedisManagerFactory =
-                {
-                    FactoryFn = (writeHosts, readHosts) =>
-                        new PooledRedisClientManager(writeHosts, readHosts, initalDb:1)
-                }
+            var sentinelHosts = new[] {"{0}:{1}".Fmt(TestConfig.SentinelHost, TestConfig.RedisSentinelPort)};
+            var sentinel = new RedisSentinel(sentinelHosts, TestConfig.MasterName) {
+                HostFilter = host => "{0}?db=1".Fmt(host)
             };
 
             using (var clientsManager = sentinel.Setup())