From 6310addfd8e47eecf0c1ed9109e7f69c05760fe8 Mon Sep 17 00:00:00 2001
From: XTao <xutao881001@gmail.com>
Date: Sat, 30 Sep 2017 00:15:10 +0800
Subject: [PATCH] Fix unicode with ConsistentHashingRouter

---
 rb/router.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/rb/router.py b/rb/router.py
index 2d84a60..71fc58a 100644
--- a/rb/router.py
+++ b/rb/router.py
@@ -112,7 +112,11 @@ def __init__(self, cluster):
         assert_gapless_hosts(self.cluster.hosts)
 
     def get_host_for_key(self, key):
-        rv = self._hash.get_node(key)
+        if isinstance(key, unicode):
+            k = key.encode('utf-8')
+        else:
+            k = str(key)
+        rv = self._hash.get_node(k)
         if rv is None:
             raise UnroutableCommand('Did not find a suitable '
                                     'host for the key.')