Add GetServerRole()

ServiceStack · Jul 11, 2015 · b5d3b70 · b5d3b70
1 parent ad2865e
commit b5d3b70
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/src/ServiceStack.Redis/RedisClient.cs b/src/ServiceStack.Redis/RedisClient.cs
@@ -1050,6 +1050,37 @@ public void MergeHyperLogs(string toKey, params string[] fromKeys)
             base.PfMerge(toKey, fromKeys);
         }
 
+        public RedisServerRole GetServerRole()
+        {
+            if (AssertServerVersionNumber() >= 2812)
+            {
+                var text = base.Role();
+                var roleName = text.Children[0].Text;
+                return ToServerRole(roleName); 
+            }
+
+            string role;
+            this.Info.TryGetValue("role", out role);
+            return ToServerRole(role);
+        }
+
+        private static RedisServerRole ToServerRole(string roleName)
+        {
+            if (string.IsNullOrEmpty(roleName))
+                return RedisServerRole.Unknown;
+
+            switch (roleName)
+            {
+                case "master":
+                    return RedisServerRole.Master;
+                case "slave":
+                    return RedisServerRole.Slave;
+                case "sentinel":
+                    return RedisServerRole.Sentinel;
+                default:
+                    return RedisServerRole.Unknown;
+            }
+        }
     }
 
 }
diff --git a/tests/ServiceStack.Redis.Tests/RedisClientConfigTests.cs b/tests/ServiceStack.Redis.Tests/RedisClientConfigTests.cs
@@ -83,6 +83,8 @@ public void Can_get_Role_Info()
         {
             var result = Redis.Role();
             result.PrintDump();
+            Assert.That(result.Children[0].Text, Is.EqualTo("master"));
+            Assert.That(Redis.GetServerRole(), Is.EqualTo(RedisServerRole.Master));
 
             //needs redis-server v3.0
             //var slave = new RedisClient("10.0.0.9:6380");